mirror of
https://github.com/refraction-networking/uquic.git
synced 2025-04-03 20:27:35 +03:00
70 lines
3.1 KiB
Go
70 lines
3.1 KiB
Go
package quic
|
|
|
|
import (
|
|
"net"
|
|
|
|
"github.com/golang/mock/gomock"
|
|
mocklogging "github.com/lucas-clemente/quic-go/internal/mocks/logging"
|
|
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
type testConn struct {
|
|
counter int
|
|
net.PacketConn
|
|
}
|
|
|
|
var _ = Describe("Multiplexer", func() {
|
|
It("adds a new packet conn ", func() {
|
|
conn := NewMockPacketConn(mockCtrl)
|
|
conn.EXPECT().ReadFrom(gomock.Any()).Do(func([]byte) { <-(make(chan struct{})) }).MaxTimes(1)
|
|
conn.EXPECT().LocalAddr().Return(&net.UDPAddr{IP: net.IPv4(1, 2, 3, 4), Port: 1234})
|
|
_, err := getMultiplexer().AddConn(conn, 8, nil, nil)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
})
|
|
|
|
It("recognizes when the same connection is added twice", func() {
|
|
pconn := NewMockPacketConn(mockCtrl)
|
|
pconn.EXPECT().LocalAddr().Return(&net.UDPAddr{IP: net.IPv4(1, 2, 3, 4), Port: 4321}).Times(2)
|
|
pconn.EXPECT().ReadFrom(gomock.Any()).Do(func([]byte) { <-(make(chan struct{})) }).MaxTimes(1)
|
|
conn := testConn{PacketConn: pconn}
|
|
tracer := mocklogging.NewMockTracer(mockCtrl)
|
|
_, err := getMultiplexer().AddConn(conn, 8, []byte("foobar"), tracer)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
conn.counter++
|
|
_, err = getMultiplexer().AddConn(conn, 8, []byte("foobar"), tracer)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(getMultiplexer().(*connMultiplexer).conns).To(HaveLen(1))
|
|
})
|
|
|
|
It("errors when adding an existing conn with a different connection ID length", func() {
|
|
conn := NewMockPacketConn(mockCtrl)
|
|
conn.EXPECT().ReadFrom(gomock.Any()).Do(func([]byte) { <-(make(chan struct{})) }).MaxTimes(1)
|
|
conn.EXPECT().LocalAddr().Return(&net.UDPAddr{IP: net.IPv4(1, 2, 3, 4), Port: 1234}).Times(2)
|
|
_, err := getMultiplexer().AddConn(conn, 5, nil, nil)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
_, err = getMultiplexer().AddConn(conn, 6, nil, nil)
|
|
Expect(err).To(MatchError("cannot use 6 byte connection IDs on a connection that is already using 5 byte connction IDs"))
|
|
})
|
|
|
|
It("errors when adding an existing conn with a different stateless rest key", func() {
|
|
conn := NewMockPacketConn(mockCtrl)
|
|
conn.EXPECT().ReadFrom(gomock.Any()).Do(func([]byte) { <-(make(chan struct{})) }).MaxTimes(1)
|
|
conn.EXPECT().LocalAddr().Return(&net.UDPAddr{IP: net.IPv4(1, 2, 3, 4), Port: 1234}).Times(2)
|
|
_, err := getMultiplexer().AddConn(conn, 7, []byte("foobar"), nil)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
_, err = getMultiplexer().AddConn(conn, 7, []byte("raboof"), nil)
|
|
Expect(err).To(MatchError("cannot use different stateless reset keys on the same packet conn"))
|
|
})
|
|
|
|
It("errors when adding an existing conn with different tracers", func() {
|
|
conn := NewMockPacketConn(mockCtrl)
|
|
conn.EXPECT().ReadFrom(gomock.Any()).Do(func([]byte) { <-(make(chan struct{})) }).MaxTimes(1)
|
|
conn.EXPECT().LocalAddr().Return(&net.UDPAddr{IP: net.IPv4(1, 2, 3, 4), Port: 1234}).Times(2)
|
|
_, err := getMultiplexer().AddConn(conn, 7, nil, mocklogging.NewMockTracer(mockCtrl))
|
|
Expect(err).ToNot(HaveOccurred())
|
|
_, err = getMultiplexer().AddConn(conn, 7, nil, mocklogging.NewMockTracer(mockCtrl))
|
|
Expect(err).To(MatchError("cannot use different tracers on the same packet conn"))
|
|
})
|
|
})
|