forked from quic-go/quic-go
Merge pull request #1019 from lucas-clemente/fix-concurrent-streams-test
fix concurrent streams integration test
This commit is contained in:
@@ -6,7 +6,6 @@ import (
|
|||||||
"io"
|
"io"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/lucas-clemente/quic-go/integrationtests/tools/testserver"
|
"github.com/lucas-clemente/quic-go/integrationtests/tools/testserver"
|
||||||
|
|
||||||
@@ -34,12 +33,15 @@ var _ = Describe("Stream tests", func() {
|
|||||||
var wg sync.WaitGroup
|
var wg sync.WaitGroup
|
||||||
wg.Add(numStreams)
|
wg.Add(numStreams)
|
||||||
for i := 0; i < numStreams; i++ {
|
for i := 0; i < numStreams; i++ {
|
||||||
str, err := sess.OpenStream()
|
str, err := sess.OpenStreamSync()
|
||||||
Expect(err).ToNot(HaveOccurred())
|
Expect(err).ToNot(HaveOccurred())
|
||||||
data := testserver.GeneratePRData(25 * i)
|
data := testserver.GeneratePRData(25 * i)
|
||||||
_, err = str.Write(data)
|
go func() {
|
||||||
|
defer GinkgoRecover()
|
||||||
|
_, err := str.Write(data)
|
||||||
Expect(err).ToNot(HaveOccurred())
|
Expect(err).ToNot(HaveOccurred())
|
||||||
Expect(str.Close()).To(Succeed())
|
Expect(str.Close()).To(Succeed())
|
||||||
|
}()
|
||||||
go func() {
|
go func() {
|
||||||
defer GinkgoRecover()
|
defer GinkgoRecover()
|
||||||
defer wg.Done()
|
defer wg.Done()
|
||||||
@@ -47,7 +49,6 @@ var _ = Describe("Stream tests", func() {
|
|||||||
Expect(err).ToNot(HaveOccurred())
|
Expect(err).ToNot(HaveOccurred())
|
||||||
Expect(dataRead).To(Equal(data))
|
Expect(dataRead).To(Equal(data))
|
||||||
}()
|
}()
|
||||||
time.Sleep(5 * time.Millisecond)
|
|
||||||
}
|
}
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user