gr1ffon/quic-go
1
0
Fork 0
forked from quic-go/quic-go
Code Pull Requests Activity

logging: migrate tests away from Ginkgo (#4641)

Browse Source
This commit is contained in:
Marten Seemann
2024-09-11 15:49:02 +08:00
committed by GitHub
parent e52027f496
commit 72be861aa2
5 changed files with 494 additions and 348 deletions
Download Patch File Download Diff File Expand all files Collapse all files

91
logging/tracer_test.go Normal file
View File

@@ -0,0 +1,91 @@
package logging_test
import (
"net"
"testing"
mocklogging "github.com/quic-go/quic-go/internal/mocks/logging"
"github.com/quic-go/quic-go/internal/protocol"
. "github.com/quic-go/quic-go/logging"
"github.com/stretchr/testify/require"
"go.uber.org/mock/gomock"
)
func TestNilTracerWhenEmpty(t *testing.T) {
require.Nil(t, NewMultiplexedTracer())
}
func TestSingleTracer(t *testing.T) {
tr := &Tracer{}
tracer := NewMultiplexedTracer(tr)
require.Equal(t, tr, tracer)
}
func TestTracerPacketSent(t *testing.T) {
ctrl := gomock.NewController(t)
t1, tr1 := mocklogging.NewMockTracer(ctrl)
t2, tr2 := mocklogging.NewMockTracer(ctrl)
tracer := NewMultiplexedTracer(t1, t2, &Tracer{})
remote := &net.UDPAddr{IP: net.IPv4(4, 3, 2, 1)}
hdr := &Header{DestConnectionID: protocol.ParseConnectionID([]byte{1, 2, 3})}
f := &MaxDataFrame{MaximumData: 1337}
tr1.EXPECT().SentPacket(remote, hdr, ByteCount(1024), []Frame{f})
tr2.EXPECT().SentPacket(remote, hdr, ByteCount(1024), []Frame{f})
tracer.SentPacket(remote, hdr, 1024, []Frame{f})
}
func TestTracerVersionNegotiationSent(t *testing.T) {
ctrl := gomock.NewController(t)
t1, tr1 := mocklogging.NewMockTracer(ctrl)
t2, tr2 := mocklogging.NewMockTracer(ctrl)
tracer := NewMultiplexedTracer(t1, t2, &Tracer{})
remote := &net.UDPAddr{IP: net.IPv4(4, 3, 2, 1)}
src := ArbitraryLenConnectionID{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13}
dest := ArbitraryLenConnectionID{1, 2, 3, 4}
versions := []Version{1, 2, 3}
tr1.EXPECT().SentVersionNegotiationPacket(remote, dest, src, versions)
tr2.EXPECT().SentVersionNegotiationPacket(remote, dest, src, versions)
tracer.SentVersionNegotiationPacket(remote, dest, src, versions)
}
func TestTracerPacketDropped(t *testing.T) {
ctrl := gomock.NewController(t)
t1, tr1 := mocklogging.NewMockTracer(ctrl)
t2, tr2 := mocklogging.NewMockTracer(ctrl)
tracer := NewMultiplexedTracer(t1, t2, &Tracer{})
remote := &net.UDPAddr{IP: net.IPv4(4, 3, 2, 1)}
tr1.EXPECT().DroppedPacket(remote, PacketTypeRetry, ByteCount(1024), PacketDropDuplicate)
tr2.EXPECT().DroppedPacket(remote, PacketTypeRetry, ByteCount(1024), PacketDropDuplicate)
tracer.DroppedPacket(remote, PacketTypeRetry, 1024, PacketDropDuplicate)
}
func TestTracerDebug(t *testing.T) {
ctrl := gomock.NewController(t)
t1, tr1 := mocklogging.NewMockTracer(ctrl)
t2, tr2 := mocklogging.NewMockTracer(ctrl)
tracer := NewMultiplexedTracer(t1, t2, &Tracer{})
tr1.EXPECT().Debug("foo", "bar")
tr2.EXPECT().Debug("foo", "bar")
tracer.Debug("foo", "bar")
}
func TestTracerClose(t *testing.T) {
ctrl := gomock.NewController(t)
t1, tr1 := mocklogging.NewMockTracer(ctrl)
t2, tr2 := mocklogging.NewMockTracer(ctrl)
tracer := NewMultiplexedTracer(t1, t2, &Tracer{})
tr1.EXPECT().Close()
tr2.EXPECT().Close()
tracer.Close()
}