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

use new gomock feature to generate type-safe methods in mocks (#4057)

Browse Source
This commit is contained in:
Marten Seemann
2023-10-21 12:55:33 +07:00
committed by GitHub
parent b344940f06
commit a263164d9f
53 changed files with 7933 additions and 754 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
send_queue_test.go
View File

@@ -31,7 +31,7 @@ var _ = Describe("Send Queue", func() {
q.Send(p, 10, protocol.ECT1) // make sure the packet size is passed through to the conn
written := make(chan struct{})
c.EXPECT().Write([]byte("foobar"), uint16(10), protocol.ECT1).Do(func([]byte, uint16, protocol.ECN) { close(written) })
c.EXPECT().Write([]byte("foobar"), uint16(10), protocol.ECT1).Do(func([]byte, uint16, protocol.ECN) error { close(written); return nil })
done := make(chan struct{})
go func() {
defer GinkgoRecover()
@@ -149,7 +149,7 @@ var _ = Describe("Send Queue", func() {
It("blocks Close() until the packet has been sent out", func() {
written := make(chan []byte)
c.EXPECT().Write(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(p []byte, _ uint16, _ protocol.ECN) { written <- p })
c.EXPECT().Write(gomock.Any(), gomock.Any(), gomock.Any()).Do(func(p []byte, _ uint16, _ protocol.ECN) error { written <- p; return nil })
done := make(chan struct{})
go func() {
defer GinkgoRecover()