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

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" "errors"
"time" "time"
"github.com/golang/mock/gomock"
"github.com/lucas-clemente/quic-go/internal/protocol" "github.com/lucas-clemente/quic-go/internal/protocol"
"github.com/lucas-clemente/quic-go/internal/utils" "github.com/lucas-clemente/quic-go/internal/utils"
@ -14,11 +15,11 @@ import (
var _ = Describe("Closed local session", func() { var _ = Describe("Closed local session", func() {
var ( var (
sess packetHandler sess packetHandler
mconn *mockConnection mconn *MockConnection
) )
BeforeEach(func() { BeforeEach(func() {
mconn = newMockConnection() mconn = NewMockConnection(mockCtrl)
sess = newClosedLocalSession(mconn, []byte("close"), protocol.PerspectiveClient, utils.DefaultLogger) 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() { 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++ { for i := 1; i <= 20; i++ {
sess.handlePacket(&receivedPacket{}) sess.handlePacket(&receivedPacket{})
if i == 1 || i == 2 || i == 4 || i == 8 || i == 16 { 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 { } else {
Consistently(mconn.written, 10*time.Millisecond).Should(HaveLen(0)) Consistently(written, 10*time.Millisecond).Should(HaveLen(0))
} }
} }
// stop the session // stop the session