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

fix flaky send queue test (#3668)

Browse source
This commit is contained in:
Marten Seemann 2023-01-16 02:35:15 -08:00 • committed by GitHub
parent a2e52e31a8
commit 5141739d57
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
send_queue_test.go
View file

@ -78,8 +78,8 @@ var _ = Describe("Send Queue", func() {
written := make(chan struct{}, 100)
// now start sending out packets. This should free up queue space.
c.EXPECT().Write(gomock.Any()).DoAndReturn(func(b []byte) error {
<-write
written <- struct{}{}
<-write
return nil
}).AnyTimes()
// allow the first packet to be sent immediately
@ -96,10 +96,15 @@ var _ = Describe("Send Queue", func() {
<-written
// now fill up the send queue
for i := 0; i < sendQueueCapacity+1; i++ {
for i := 0; i < sendQueueCapacity; i++ {
Expect(q.WouldBlock()).To(BeFalse())
q.Send(getPacket([]byte("foobar")))
}
// One more packet is queued when it's picked up by Run and written to the connection.
// In this test, it's blocked on write channel in the mocked Write call.
<-written
Eventually(q.WouldBlock()).Should(BeFalse())
q.Send(getPacket([]byte("foobar")))
Expect(q.WouldBlock()).To(BeTrue())
Consistently(q.Available()).ShouldNot(Receive())