mirror of
https://github.com/refraction-networking/uquic.git
synced 2025-04-03 20:27:35 +03:00
47 lines
1 KiB
Go
47 lines
1 KiB
Go
package quic
|
|
|
|
import (
|
|
"net"
|
|
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("Connection (for sending packets)", func() {
|
|
var (
|
|
c sendConn
|
|
packetConn *MockPacketConn
|
|
addr net.Addr
|
|
)
|
|
|
|
BeforeEach(func() {
|
|
addr = &net.UDPAddr{IP: net.IPv4(192, 168, 100, 200), Port: 1337}
|
|
packetConn = NewMockPacketConn(mockCtrl)
|
|
rawConn, err := wrapConn(packetConn)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
c = newSendConn(rawConn, addr, nil)
|
|
})
|
|
|
|
It("writes", func() {
|
|
packetConn.EXPECT().WriteTo([]byte("foobar"), addr)
|
|
Expect(c.Write([]byte("foobar"), 6)).To(Succeed())
|
|
})
|
|
|
|
It("gets the remote address", func() {
|
|
Expect(c.RemoteAddr().String()).To(Equal("192.168.100.200:1337"))
|
|
})
|
|
|
|
It("gets the local address", func() {
|
|
addr := &net.UDPAddr{
|
|
IP: net.IPv4(192, 168, 0, 1),
|
|
Port: 1234,
|
|
}
|
|
packetConn.EXPECT().LocalAddr().Return(addr)
|
|
Expect(c.LocalAddr()).To(Equal(addr))
|
|
})
|
|
|
|
It("closes", func() {
|
|
packetConn.EXPECT().Close()
|
|
Expect(c.Close()).To(Succeed())
|
|
})
|
|
})
|