mirrors/uquic
1
0
Fork 0
mirror of https://github.com/refraction-networking/uquic.git synced 2025-04-04 12:47:36 +03:00
Code Issues Projects Releases Packages Wiki Activity

add a multiplexer for the logging.ConnectionTracer

Browse source
This commit is contained in:
Marten Seemann 2020-07-06 13:57:18 +07:00
parent 2f5cba0856
commit 2d4eca1dd6
6 changed files with 597 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

159
logging/multiplex_test.go Normal file
View file

@ -0,0 +1,159 @@
package logging
import (
"net"
"time"
"github.com/lucas-clemente/quic-go/internal/wire"
. "github.com/onsi/ginkgo"
)
var _ = Describe("Tracing", func() {
var (
tracer ConnectionTracer
tr1 *MockConnectionTracer
tr2 *MockConnectionTracer
)
BeforeEach(func() {
tr1 = NewMockConnectionTracer(mockCtrl)
tr2 = NewMockConnectionTracer(mockCtrl)
tracer = NewConnectionMultiplexer(tr1, tr2)
})
It("trace the ConnectionStarted event", func() {
local := &net.UDPAddr{IP: net.IPv4(1, 2, 3, 4)}
remote := &net.UDPAddr{IP: net.IPv4(4, 3, 2, 1)}
tr1.EXPECT().StartedConnection(local, remote, VersionNumber(1234), ConnectionID{1, 2, 3, 4}, ConnectionID{4, 3, 2, 1})
tr2.EXPECT().StartedConnection(local, remote, VersionNumber(1234), ConnectionID{1, 2, 3, 4}, ConnectionID{4, 3, 2, 1})
tracer.StartedConnection(local, remote, VersionNumber(1234), ConnectionID{1, 2, 3, 4}, ConnectionID{4, 3, 2, 1})
})
It("traces the ClosedConnection event", func() {
reason := NewTimeoutCloseReason(TimeoutReasonIdle)
tr1.EXPECT().ClosedConnection(reason)
tr2.EXPECT().ClosedConnection(reason)
tracer.ClosedConnection(reason)
})
It("traces the SentTransportParameters event", func() {
tp := &wire.TransportParameters{InitialMaxData: 1337}
tr1.EXPECT().SentTransportParameters(tp)
tr2.EXPECT().SentTransportParameters(tp)
tracer.SentTransportParameters(tp)
})
It("traces the ReceivedTransportParameters event", func() {
tp := &wire.TransportParameters{InitialMaxData: 1337}
tr1.EXPECT().ReceivedTransportParameters(tp)
tr2.EXPECT().ReceivedTransportParameters(tp)
tracer.ReceivedTransportParameters(tp)
})
It("traces the SentPacket event", func() {
hdr := &ExtendedHeader{Header: Header{DestConnectionID: ConnectionID{1, 2, 3}}}
ack := &AckFrame{AckRanges: []AckRange{{Smallest: 1, Largest: 10}}}
ping := &PingFrame{}
tr1.EXPECT().SentPacket(hdr, ByteCount(1337), ack, []Frame{ping})
tr2.EXPECT().SentPacket(hdr, ByteCount(1337), ack, []Frame{ping})
tracer.SentPacket(hdr, 1337, ack, []Frame{ping})
})
It("traces the ReceivedVersionNegotiationPacket event", func() {
hdr := &Header{DestConnectionID: ConnectionID{1, 2, 3}}
tr1.EXPECT().ReceivedVersionNegotiationPacket(hdr, []VersionNumber{1337})
tr2.EXPECT().ReceivedVersionNegotiationPacket(hdr, []VersionNumber{1337})
tracer.ReceivedVersionNegotiationPacket(hdr, []VersionNumber{1337})
})
It("traces the ReceivedRetry event", func() {
hdr := &Header{DestConnectionID: ConnectionID{1, 2, 3}}
tr1.EXPECT().ReceivedRetry(hdr)
tr2.EXPECT().ReceivedRetry(hdr)
tracer.ReceivedRetry(hdr)
})
It("traces the ReceivedPacket event", func() {
hdr := &ExtendedHeader{Header: Header{DestConnectionID: ConnectionID{1, 2, 3}}}
ping := &PingFrame{}
tr1.EXPECT().ReceivedPacket(hdr, ByteCount(1337), []Frame{ping})
tr2.EXPECT().ReceivedPacket(hdr, ByteCount(1337), []Frame{ping})
tracer.ReceivedPacket(hdr, 1337, []Frame{ping})
})
It("traces the BufferedPacket event", func() {
tr1.EXPECT().BufferedPacket(PacketTypeHandshake)
tr2.EXPECT().BufferedPacket(PacketTypeHandshake)
tracer.BufferedPacket(PacketTypeHandshake)
})
It("traces the DroppedPacket event", func() {
tr1.EXPECT().DroppedPacket(PacketTypeInitial, ByteCount(1337), PacketDropHeaderParseError)
tr2.EXPECT().DroppedPacket(PacketTypeInitial, ByteCount(1337), PacketDropHeaderParseError)
tracer.DroppedPacket(PacketTypeInitial, 1337, PacketDropHeaderParseError)
})
It("traces the UpdatedMetrics event", func() {
rttStats := &RTTStats{}
rttStats.UpdateRTT(time.Second, 0, time.Now())
tr1.EXPECT().UpdatedMetrics(rttStats, ByteCount(1337), ByteCount(42), 13)
tr2.EXPECT().UpdatedMetrics(rttStats, ByteCount(1337), ByteCount(42), 13)
tracer.UpdatedMetrics(rttStats, 1337, 42, 13)
})
It("traces the LostPacket event", func() {
tr1.EXPECT().LostPacket(EncryptionHandshake, PacketNumber(42), PacketLossReorderingThreshold)
tr2.EXPECT().LostPacket(EncryptionHandshake, PacketNumber(42), PacketLossReorderingThreshold)
tracer.LostPacket(EncryptionHandshake, 42, PacketLossReorderingThreshold)
})
It("traces the UpdatedPTOCount event", func() {
tr1.EXPECT().UpdatedPTOCount(uint32(88))
tr2.EXPECT().UpdatedPTOCount(uint32(88))
tracer.UpdatedPTOCount(88)
})
It("traces the UpdatedKeyFromTLS event", func() {
tr1.EXPECT().UpdatedKeyFromTLS(EncryptionHandshake, PerspectiveClient)
tr2.EXPECT().UpdatedKeyFromTLS(EncryptionHandshake, PerspectiveClient)
tracer.UpdatedKeyFromTLS(EncryptionHandshake, PerspectiveClient)
})
It("traces the UpdatedKey event", func() {
tr1.EXPECT().UpdatedKey(KeyPhase(42), true)
tr2.EXPECT().UpdatedKey(KeyPhase(42), true)
tracer.UpdatedKey(KeyPhase(42), true)
})
It("traces the DroppedEncryptionLevel event", func() {
tr1.EXPECT().DroppedEncryptionLevel(EncryptionHandshake)
tr2.EXPECT().DroppedEncryptionLevel(EncryptionHandshake)
tracer.DroppedEncryptionLevel(EncryptionHandshake)
})
It("traces the SetLossTimer event", func() {
now := time.Now()
tr1.EXPECT().SetLossTimer(TimerTypePTO, EncryptionHandshake, now)
tr2.EXPECT().SetLossTimer(TimerTypePTO, EncryptionHandshake, now)
tracer.SetLossTimer(TimerTypePTO, EncryptionHandshake, now)
})
It("traces the LossTimerExpired event", func() {
tr1.EXPECT().LossTimerExpired(TimerTypePTO, EncryptionHandshake)
tr2.EXPECT().LossTimerExpired(TimerTypePTO, EncryptionHandshake)
tracer.LossTimerExpired(TimerTypePTO, EncryptionHandshake)
})
It("traces the LossTimerCanceled event", func() {
tr1.EXPECT().LossTimerCanceled()
tr2.EXPECT().LossTimerCanceled()
tracer.LossTimerCanceled()
})
It("traces the Close event", func() {
tr1.EXPECT().Close()
tr2.EXPECT().Close()
tracer.Close()
})
})