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

fix flaky stream tests

Browse source 
fixes #549
This commit is contained in:
Marten Seemann 2017-04-18 10:15:54 +07:00
parent 4b100409c4
commit 2a1b3da971
No known key found for this signature in database
GPG key ID: 3603F40B121FCDEA
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
stream_test.go
View file

@ -662,37 +662,43 @@ var _ = Describe("Stream", func() {
Context("writing", func() { Context("writing", func() {
It("writes and gets all data at once", func(done Done) { It("writes and gets all data at once", func(done Done) {
var writeReturned bool
go func() { go func() {
n, err := str.Write([]byte("foobar")) n, err := str.Write([]byte("foobar"))
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
Expect(n).To(Equal(6)) Expect(n).To(Equal(6))
close(done) writeReturned = true
}() }()
Eventually(func() []byte { Eventually(func() []byte {
str.mutex.Lock() str.mutex.Lock()
defer str.mutex.Unlock() defer str.mutex.Unlock()
return str.dataForWriting return str.dataForWriting
}).Should(Equal([]byte("foobar"))) }).Should(Equal([]byte("foobar")))
Consistently(func() bool { return writeReturned }).Should(BeFalse())
Expect(onDataCalled).To(BeTrue()) Expect(onDataCalled).To(BeTrue())
Expect(str.lenOfDataForWriting()).To(Equal(protocol.ByteCount(6))) Expect(str.lenOfDataForWriting()).To(Equal(protocol.ByteCount(6)))
data := str.getDataForWriting(1000) data := str.getDataForWriting(1000)
Expect(data).To(Equal([]byte("foobar"))) Expect(data).To(Equal([]byte("foobar")))
Expect(str.writeOffset).To(Equal(protocol.ByteCount(6))) Expect(str.writeOffset).To(Equal(protocol.ByteCount(6)))
Expect(str.dataForWriting).To(BeNil()) Expect(str.dataForWriting).To(BeNil())
Eventually(func() bool { return writeReturned }).Should(BeTrue())
close(done)
}) })
It("writes and gets data in two turns", func(done Done) { It("writes and gets data in two turns", func(done Done) {
var writeReturned bool
go func() { go func() {
n, err := str.Write([]byte("foobar")) n, err := str.Write([]byte("foobar"))
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
Expect(n).To(Equal(6)) Expect(n).To(Equal(6))
close(done) writeReturned = true
}() }()
Eventually(func() []byte { Eventually(func() []byte {
str.mutex.Lock() str.mutex.Lock()
defer str.mutex.Unlock() defer str.mutex.Unlock()
return str.dataForWriting return str.dataForWriting
}).Should(Equal([]byte("foobar"))) }).Should(Equal([]byte("foobar")))
Consistently(func() bool { return writeReturned }).Should(BeFalse())
Expect(str.lenOfDataForWriting()).To(Equal(protocol.ByteCount(6))) Expect(str.lenOfDataForWriting()).To(Equal(protocol.ByteCount(6)))
data := str.getDataForWriting(3) data := str.getDataForWriting(3)
Expect(data).To(Equal([]byte("foo"))) Expect(data).To(Equal([]byte("foo")))
@ -704,6 +710,8 @@ var _ = Describe("Stream", func() {
Expect(str.writeOffset).To(Equal(protocol.ByteCount(6))) Expect(str.writeOffset).To(Equal(protocol.ByteCount(6)))
Expect(str.dataForWriting).To(BeNil()) Expect(str.dataForWriting).To(BeNil())
Expect(str.lenOfDataForWriting()).To(Equal(protocol.ByteCount(0))) Expect(str.lenOfDataForWriting()).To(Equal(protocol.ByteCount(0)))
Eventually(func() bool { return writeReturned }).Should(BeTrue())
close(done)
}) })
It("getDataForWriting returns nil if no data is available", func() { It("getDataForWriting returns nil if no data is available", func() {