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

use a mock connection in the send queue tests

Browse Source
This commit is contained in:
Marten Seemann
2020-01-11 19:14:38 +07:00
parent adfc015a9f
commit 13b2c7403d
2 changed files with 13 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
send_queue_test.go
View File

@@ -1,16 +1,17 @@
package quic
import (
"github.com/golang/mock/gomock"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)
var _ = Describe("Send Queue", func() {
var q *sendQueue
var c *mockConnection
var c *MockConnection
BeforeEach(func() {
c = newMockConnection()
c = NewMockConnection(mockCtrl)
q = newSendQueue(c)
})
@@ -25,8 +26,11 @@ var _ = Describe("Send Queue", func() {
}
It("sends a packet", func() {
q.Send(getPacket([]byte("foobar")))
p := getPacket([]byte("foobar"))
q.Send(p)
written := make(chan struct{})
c.EXPECT().Write(p.raw).Do(func([]byte) { close(written) })
done := make(chan struct{})
go func() {
defer GinkgoRecover()
@@ -34,7 +38,7 @@ var _ = Describe("Send Queue", func() {
close(done)
}()
Eventually(c.written).Should(Receive(Equal([]byte("foobar"))))
Eventually(written).Should(BeClosed())
q.Close()
Eventually(done).Should(BeClosed())
})
@@ -42,6 +46,9 @@ var _ = Describe("Send Queue", func() {
It("blocks sending when too many packets are queued", func() {
q.Send(getPacket([]byte("foobar")))
written := make(chan []byte, 2)
c.EXPECT().Write(gomock.Any()).Do(func(p []byte) { written <- p }).Times(2)
sent := make(chan struct{})
go func() {
defer GinkgoRecover()
@@ -58,8 +65,8 @@ var _ = Describe("Send Queue", func() {
close(done)
}()
Eventually(c.written).Should(Receive(Equal([]byte("foobar"))))
Eventually(c.written).Should(Receive(Equal([]byte("raboof"))))
Eventually(written).Should(Receive(Equal([]byte("foobar"))))
Eventually(written).Should(Receive(Equal([]byte("raboof"))))
q.Close()
Eventually(done).Should(BeClosed())
})