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

use flow controller for window update handling, remove windowUpdateMgr

Browse source
This commit is contained in:
Lucas Clemente 2016-07-11 14:54:32 +02:00
parent e303a7f578
commit f04d14ac3e
8 changed files with 72 additions and 269 deletions
Download patch file Download diff file Expand all files Collapse all files

42
stream_test.go
View file

@ -14,17 +14,9 @@ import (
)
type mockStreamHandler struct {
receiveFlowControlWindowCalled bool
receiveFlowControlWindowCalledForStream protocol.StreamID
scheduledSending bool
}
func (m *mockStreamHandler) updateReceiveFlowControlWindow(streamID protocol.StreamID, byteOffset protocol.ByteCount) error {
m.receiveFlowControlWindowCalled = true
m.receiveFlowControlWindowCalledForStream = streamID
return nil
}
func (m *mockStreamHandler) scheduleSending() { m.scheduledSending = true }
type mockFlowControlHandler struct {
@ -416,40 +408,6 @@ var _ = Describe("Stream", func() {
Expect(str.flowControlManager.(*mockFlowControlHandler).highestReceivedForStream).To(Equal(str.streamID))
Expect(str.flowControlManager.(*mockFlowControlHandler).highestReceived).To(Equal(protocol.ByteCount(2 + 6)))
})
It("updates the flow control window", func() {
str.flowControlManager.(*mockFlowControlHandler).triggerStreamWindowUpdate = true
frame := frames.StreamFrame{
Offset: 0,
Data: []byte("foobar"),
}
err := str.AddStreamFrame(&frame)
Expect(err).ToNot(HaveOccurred())
b := make([]byte, 6)
n, err := str.Read(b)
Expect(err).ToNot(HaveOccurred())
Expect(n).To(Equal(6))
Expect(handler.receiveFlowControlWindowCalled).To(BeTrue())
Expect(handler.receiveFlowControlWindowCalledForStream).To(Equal(str.streamID))
})
It("updates the connection level flow control window", func() {
str.flowControlManager.(*mockFlowControlHandler).triggerConnectionWindowUpdate = true
frame := frames.StreamFrame{
Offset: 0,
Data: []byte("foobar"),
}
err := str.AddStreamFrame(&frame)
Expect(err).ToNot(HaveOccurred())
b := make([]byte, 6)
n, err := str.Read(b)
Expect(err).ToNot(HaveOccurred())
Expect(n).To(Equal(6))
Expect(handler.receiveFlowControlWindowCalled).To(BeTrue())
Expect(handler.receiveFlowControlWindowCalledForStream).To(Equal(protocol.StreamID(0)))
})
// TODO: think about flow control violation
})
Context("closing", func() {