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

use a mock connection in the closedSession tests

Browse source
This commit is contained in:
Marten Seemann 2020-01-11 19:14:21 +07:00
parent 15301a7f6e
commit adfc015a9f
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
closed_session_test.go
View file

@ -4,6 +4,7 @@ import (
"errors"
"time"
"github.com/golang/mock/gomock"
"github.com/lucas-clemente/quic-go/internal/protocol"
"github.com/lucas-clemente/quic-go/internal/utils"
@ -14,11 +15,11 @@ import (
var _ = Describe("Closed local session", func() {
var (
sess packetHandler
mconn *mockConnection
mconn *MockConnection
)
BeforeEach(func() {
mconn = newMockConnection()
mconn = NewMockConnection(mockCtrl)
sess = newClosedLocalSession(mconn, []byte("close"), protocol.PerspectiveClient, utils.DefaultLogger)
})
@ -33,12 +34,14 @@ var _ = Describe("Closed local session", func() {
})
It("repeats the packet containing the CONNECTION_CLOSE frame", func() {
written := make(chan []byte)
mconn.EXPECT().Write(gomock.Any()).Do(func(p []byte) { written <- p }).AnyTimes()
for i := 1; i <= 20; i++ {
sess.handlePacket(&receivedPacket{})
if i == 1 || i == 2 || i == 4 || i == 8 || i == 16 {
Eventually(mconn.written).Should(Receive(Equal([]byte("close")))) // receive the CONNECTION_CLOSE
Eventually(written).Should(Receive(Equal([]byte("close")))) // receive the CONNECTION_CLOSE
} else {
Consistently(mconn.written, 10*time.Millisecond).Should(HaveLen(0))
Consistently(written, 10*time.Millisecond).Should(HaveLen(0))
}
}
// stop the session