91 lines
3.0 KiB
Go
91 lines
3.0 KiB
Go
|
package quic
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/lucas-clemente/quic-go/internal/protocol"
|
||
|
|
"github.com/lucas-clemente/quic-go/internal/wire"
|
||
|
|
|
||
|
|
. "github.com/onsi/ginkgo"
|
||
|
|
. "github.com/onsi/gomega"
|
||
|
|
)
|
||
|
|
|
||
|
|
var _ = Describe("Window Update Queue", func() {
|
||
|
|
var (
|
||
|
|
q *windowUpdateQueue
|
||
|
|
streamGetter *MockStreamGetter
|
||
|
|
queuedFrames []wire.Frame
|
||
|
|
cryptoStream *MockCryptoStream
|
||
|
|
)
|
||
|
|
|
||
|
|
BeforeEach(func() {
|
||
|
|
streamGetter = NewMockStreamGetter(mockCtrl)
|
||
|
|
cryptoStream = NewMockCryptoStream(mockCtrl)
|
||
|
|
cryptoStream.EXPECT().StreamID().Return(protocol.StreamID(0)).AnyTimes()
|
||
|
|
queuedFrames = queuedFrames[:0]
|
||
|
|
q = newWindowUpdateQueue(streamGetter, cryptoStream, func(f wire.Frame) {
|
||
|
|
queuedFrames = append(queuedFrames, f)
|
||
|
|
})
|
||
|
|
})
|
||
|
|
|
||
|
|
It("adds stream offsets and gets MAX_STREAM_DATA frames", func() {
|
||
|
|
stream1 := NewMockStreamI(mockCtrl)
|
||
|
|
stream1.EXPECT().getWindowUpdate().Return(protocol.ByteCount(10))
|
||
|
|
stream3 := NewMockStreamI(mockCtrl)
|
||
|
|
stream3.EXPECT().getWindowUpdate().Return(protocol.ByteCount(30))
|
||
|
|
streamGetter.EXPECT().GetOrOpenReceiveStream(protocol.StreamID(3)).Return(stream3, nil)
|
||
|
|
streamGetter.EXPECT().GetOrOpenReceiveStream(protocol.StreamID(1)).Return(stream1, nil)
|
||
|
|
q.Add(3)
|
||
|
|
q.Add(1)
|
||
|
|
q.QueueAll()
|
||
|
|
Expect(queuedFrames).To(ContainElement(&wire.MaxStreamDataFrame{StreamID: 1, ByteOffset: 10}))
|
||
|
|
Expect(queuedFrames).To(ContainElement(&wire.MaxStreamDataFrame{StreamID: 3, ByteOffset: 30}))
|
||
|
|
})
|
||
|
|
|
||
|
|
It("deletes the entry after getting the MAX_STREAM_DATA frame", func() {
|
||
|
|
stream10 := NewMockStreamI(mockCtrl)
|
||
|
|
stream10.EXPECT().getWindowUpdate().Return(protocol.ByteCount(100))
|
||
|
|
streamGetter.EXPECT().GetOrOpenReceiveStream(protocol.StreamID(10)).Return(stream10, nil)
|
||
|
|
q.Add(10)
|
||
|
|
q.QueueAll()
|
||
|
|
Expect(queuedFrames).To(HaveLen(1))
|
||
|
|
q.QueueAll()
|
||
|
|
Expect(queuedFrames).To(HaveLen(1))
|
||
|
|
})
|
||
|
|
|
||
|
|
It("doesn't queue a MAX_STREAM_DATA for a closed stream", func() {
|
||
|
|
streamGetter.EXPECT().GetOrOpenReceiveStream(protocol.StreamID(12)).Return(nil, nil)
|
||
|
|
q.Add(12)
|
||
|
|
q.QueueAll()
|
||
|
|
Expect(queuedFrames).To(BeEmpty())
|
||
|
|
})
|
||
|
|
|
||
|
|
It("doesn't queue a MAX_STREAM_DATA if the flow controller returns an offset of 0", func() {
|
||
|
|
stream5 := NewMockStreamI(mockCtrl)
|
||
|
|
stream5.EXPECT().getWindowUpdate().Return(protocol.ByteCount(0))
|
||
|
|
streamGetter.EXPECT().GetOrOpenReceiveStream(protocol.StreamID(5)).Return(stream5, nil)
|
||
|
|
q.Add(5)
|
||
|
|
q.QueueAll()
|
||
|
|
Expect(queuedFrames).To(BeEmpty())
|
||
|
|
})
|
||
|
|
|
||
|
|
It("adds MAX_STREAM_DATA frames for the crypto stream", func() {
|
||
|
|
cryptoStream.EXPECT().getWindowUpdate().Return(protocol.ByteCount(42))
|
||
|
|
q.Add(0)
|
||
|
|
q.QueueAll()
|
||
|
|
Expect(queuedFrames).To(Equal([]wire.Frame{
|
||
|
|
&wire.MaxStreamDataFrame{StreamID: 0, ByteOffset: 42},
|
||
|
|
}))
|
||
|
|
})
|
||
|
|
|
||
|
|
It("deduplicates", func() {
|
||
|
|
stream10 := NewMockStreamI(mockCtrl)
|
||
|
|
stream10.EXPECT().getWindowUpdate().Return(protocol.ByteCount(200))
|
||
|
|
streamGetter.EXPECT().GetOrOpenReceiveStream(protocol.StreamID(10)).Return(stream10, nil)
|
||
|
|
q.Add(10)
|
||
|
|
q.Add(10)
|
||
|
|
q.QueueAll()
|
||
|
|
Expect(queuedFrames).To(Equal([]wire.Frame{
|
||
|
|
&wire.MaxStreamDataFrame{StreamID: 10, ByteOffset: 200},
|
||
|
|
}))
|
||
|
|
})
|
||
|
|
})
|