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

small improvements of Stream tests

Browse Source
This commit is contained in:
Marten Seemann
2016-05-26 09:18:07 +07:00
parent be34c9e3f4
commit 3a08041ab3
1 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
stream_test.go
View File

@@ -348,8 +348,8 @@ var _ = Describe("Stream", func() {
updated = str.connectionFlowController.UpdateSendWindow(1)
Expect(updated).To(BeTrue())
n, err := str.Write([]byte{0xDE, 0xCA, 0xFB, 0xAD})
Expect(n).To(Equal(4))
Expect(err).ToNot(HaveOccurred())
Expect(n).To(Equal(4))
})
It("returns true when the flow control window was updated", func() {
@@ -379,9 +379,10 @@ var _ = Describe("Stream", func() {
str.UpdateSendFlowControlWindow(3)
}()
n, err := str.Write([]byte{0x13, 0x37})
Expect(err).ToNot(HaveOccurred())
Expect(b).To(BeTrue())
Expect(n).To(Equal(2))
Expect(err).ToNot(HaveOccurred())
Expect(str.writeOffset).To(Equal(protocol.ByteCount(3)))
})
It("does not write too much data after receiving a window update", func() {
@@ -397,6 +398,7 @@ var _ = Describe("Stream", func() {
n, err := str.Write([]byte{0x13, 0x37})
Expect(b).To(BeTrue())
Expect(n).To(Equal(2))
Expect(str.writeOffset).To(Equal(protocol.ByteCount(2)))
Expect(err).ToNot(HaveOccurred())
})
@@ -410,6 +412,7 @@ var _ = Describe("Stream", func() {
_, err := str.Write([]byte{0x42})
Expect(err).ToNot(HaveOccurred())
Expect(str.writeOffset).To(Equal(protocol.ByteCount(1)))
var sendWindowUpdated bool
go func() {
@@ -423,6 +426,7 @@ var _ = Describe("Stream", func() {
Expect(b).To(BeTrue())
Expect(sendWindowUpdated).To(BeTrue())
Expect(n).To(Equal(2))
Expect(str.writeOffset).To(Equal(protocol.ByteCount(3)))
Expect(err).ToNot(HaveOccurred())
})
@@ -438,10 +442,11 @@ var _ = Describe("Stream", func() {
}()
n, err := str.Write([]byte{0xDE, 0xCA, 0xFB, 0xAD})
Expect(err).ToNot(HaveOccurred())
Expect(handler.frames).To(HaveLen(2))
Expect(b).To(BeTrue())
Expect(n).To(Equal(4))
Expect(err).ToNot(HaveOccurred())
Expect(str.writeOffset).To(Equal(protocol.ByteCount(4)))
})
It("writes after a flow control window update", func() {
@@ -458,9 +463,10 @@ var _ = Describe("Stream", func() {
str.UpdateSendFlowControlWindow(3)
}()
n, err := str.Write([]byte{0xDE, 0xAD})
Expect(err).ToNot(HaveOccurred())
Expect(b).To(BeTrue())
Expect(n).To(Equal(2))
Expect(err).ToNot(HaveOccurred())
Expect(str.writeOffset).To(Equal(protocol.ByteCount(3)))
})
It("immediately returns on remote errors", func() {
@@ -477,8 +483,8 @@ var _ = Describe("Stream", func() {
}()
_, err := str.Write([]byte{0xDE, 0xCA, 0xFB, 0xAD})
Expect(b).To(BeTrue())
Expect(err).To(MatchError(testErr))
Expect(b).To(BeTrue())
})
It("works with large flow control windows", func() {