forked from quic-go/quic-go
migrate the closed connection tests away from Ginkgo (#4848)
This commit is contained in:
@@ -2,25 +2,33 @@ package quic
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"net"
|
"net"
|
||||||
|
"testing"
|
||||||
|
|
||||||
"github.com/quic-go/quic-go/internal/utils"
|
"github.com/quic-go/quic-go/internal/utils"
|
||||||
|
|
||||||
. "github.com/onsi/ginkgo/v2"
|
"github.com/stretchr/testify/require"
|
||||||
. "github.com/onsi/gomega"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
var _ = Describe("Closed local connection", func() {
|
func TestClosedLocalConnection(t *testing.T) {
|
||||||
It("repeats the packet containing the CONNECTION_CLOSE frame", func() {
|
|
||||||
written := make(chan net.Addr, 1)
|
written := make(chan net.Addr, 1)
|
||||||
conn := newClosedLocalConn(func(addr net.Addr, _ packetInfo) { written <- addr }, utils.DefaultLogger)
|
conn := newClosedLocalConn(func(addr net.Addr, _ packetInfo) { written <- addr }, utils.DefaultLogger)
|
||||||
addr := &net.UDPAddr{IP: net.IPv4(127, 1, 2, 3), Port: 1337}
|
addr := &net.UDPAddr{IP: net.IPv4(127, 1, 2, 3), Port: 1337}
|
||||||
for i := 1; i <= 20; i++ {
|
for i := 1; i <= 20; i++ {
|
||||||
conn.handlePacket(receivedPacket{remoteAddr: addr})
|
conn.handlePacket(receivedPacket{remoteAddr: addr})
|
||||||
if i == 1 || i == 2 || i == 4 || i == 8 || i == 16 {
|
if i == 1 || i == 2 || i == 4 || i == 8 || i == 16 {
|
||||||
Expect(written).To(Receive(Equal(addr))) // receive the CONNECTION_CLOSE
|
select {
|
||||||
|
case gotAddr := <-written:
|
||||||
|
require.Equal(t, addr, gotAddr) // receive the CONNECTION_CLOSE
|
||||||
|
default:
|
||||||
|
t.Fatal("expected to receive address")
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
Expect(written).ToNot(Receive())
|
select {
|
||||||
|
case gotAddr := <-written:
|
||||||
|
t.Fatalf("unexpected address received: %v", gotAddr)
|
||||||
|
default:
|
||||||
|
// Nothing received, which is expected
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
})
|
}
|
||||||
})
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user