84 lines
2.2 KiB
Go
84 lines
2.2 KiB
Go
|
package frames
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"os"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/lucas-clemente/quic-go/protocol"
|
||
|
|
"github.com/lucas-clemente/quic-go/utils"
|
||
|
|
|
||
|
|
. "github.com/onsi/ginkgo"
|
||
|
|
. "github.com/onsi/gomega"
|
||
|
|
)
|
||
|
|
|
||
|
|
var _ = Describe("Frame logging", func() {
|
||
|
|
var (
|
||
|
|
buf bytes.Buffer
|
||
|
|
)
|
||
|
|
|
||
|
|
BeforeEach(func() {
|
||
|
|
buf.Reset()
|
||
|
|
utils.SetLogLevel(utils.LogLevelDebug)
|
||
|
|
utils.SetLogWriter(&buf)
|
||
|
|
})
|
||
|
|
|
||
|
|
AfterSuite(func() {
|
||
|
|
utils.SetLogLevel(utils.LogLevelNothing)
|
||
|
|
utils.SetLogWriter(os.Stdout)
|
||
|
|
})
|
||
|
|
|
||
|
|
It("doesn't log when debug is disabled", func() {
|
||
|
|
utils.SetLogLevel(utils.LogLevelInfo)
|
||
|
|
LogFrame(&RstStreamFrame{}, true)
|
||
|
|
Expect(buf.Len()).To(BeZero())
|
||
|
|
})
|
||
|
|
|
||
|
|
It("logs sent frames", func() {
|
||
|
|
LogFrame(&RstStreamFrame{}, true)
|
||
|
|
Expect(string(buf.Bytes())).To(Equal("\t-> &frames.RstStreamFrame{StreamID:0x0, ErrorCode:0x0, ByteOffset:0x0}\n"))
|
||
|
|
})
|
||
|
|
|
||
|
|
It("logs received frames", func() {
|
||
|
|
LogFrame(&RstStreamFrame{}, false)
|
||
|
|
Expect(string(buf.Bytes())).To(Equal("\t<- &frames.RstStreamFrame{StreamID:0x0, ErrorCode:0x0, ByteOffset:0x0}\n"))
|
||
|
|
})
|
||
|
|
|
||
|
|
It("logs stream frames", func() {
|
||
|
|
frame := &StreamFrame{
|
||
|
|
StreamID: 42,
|
||
|
|
Offset: 0x1337,
|
||
|
|
Data: bytes.Repeat([]byte{'f'}, 0x100),
|
||
|
|
}
|
||
|
|
LogFrame(frame, false)
|
||
|
|
Expect(string(buf.Bytes())).To(Equal("\t<- &frames.StreamFrame{StreamID: 42, FinBit: false, Offset: 0x1337, Data length: 0x100, Offset + Data length: 0x1437}\n"))
|
||
|
|
})
|
||
|
|
|
||
|
|
It("logs ACK frames", func() {
|
||
|
|
frame := &AckFrame{
|
||
|
|
LargestAcked: 0x1337,
|
||
|
|
LowestAcked: 0x42,
|
||
|
|
DelayTime: 1 * time.Millisecond,
|
||
|
|
}
|
||
|
|
LogFrame(frame, false)
|
||
|
|
Expect(string(buf.Bytes())).To(Equal("\t<- &frames.AckFrame{LargestAcked: 0x1337, LowestAcked: 0x42, AckRanges: []frames.AckRange(nil), DelayTime: 1ms}\n"))
|
||
|
|
})
|
||
|
|
|
||
|
|
It("logs incoming StopWaiting frames", func() {
|
||
|
|
frame := &StopWaitingFrame{
|
||
|
|
LeastUnacked: 0x1337,
|
||
|
|
}
|
||
|
|
LogFrame(frame, false)
|
||
|
|
Expect(string(buf.Bytes())).To(Equal("\t<- &frames.StopWaitingFrame{LeastUnacked: 0x1337}\n"))
|
||
|
|
})
|
||
|
|
|
||
|
|
It("logs outgoing StopWaiting frames", func() {
|
||
|
|
frame := &StopWaitingFrame{
|
||
|
|
LeastUnacked: 0x1337,
|
||
|
|
PacketNumberLen: protocol.PacketNumberLen4,
|
||
|
|
}
|
||
|
|
LogFrame(frame, true)
|
||
|
|
Expect(string(buf.Bytes())).To(Equal("\t-> &frames.StopWaitingFrame{LeastUnacked: 0x1337, PacketNumberLen: 0x4}\n"))
|
||
|
|
})
|
||
|
|
})
|