Function TestDialQuicWithSkipPortReuse

connection/quic_connection_test.go:717–780

TestDialQuicWithSkipPortReuse tests that DialQuic works correctly with the WithSkipPortReuse option.

(t *testing.T)

// TestDialQuicWithSkipPortReuse tests that DialQuic works correctly with the WithSkipPortReuse option.
func TestDialQuicWithSkipPortReuse(t *testing.T) {
	t.Parallel()

	ctx, cancel := context.WithCancel(t.Context())
	defer cancel()

	// Start a mock QUIC server (similar to TestQUICServer)
	udpListener, err := net.ListenUDP("udp", &net.UDPAddr{IP: net.IPv4(127, 0, 0, 1), Port: 0})
	require.NoError(t, err)
	defer func() { _ = udpListener.Close() }()

	serverAddr := netip.MustParseAddrPort(udpListener.LocalAddr().String())

	quicTransport := &quic.Transport{Conn: udpListener, ConnectionIDLength: 16}
	quicListener, err := quicTransport.Listen(testTLSServerConfig, testQUICConfig)
	require.NoError(t, err)

	serverDone := make(chan struct{})
	go func() {
		// Accept one connection
		session, err := quicListener.Accept(ctx)
		if err != nil {
			close(serverDone)
			return
		}
		// Keep session open until context is cancelled
		<-ctx.Done()
		_ = session.CloseWithError(0, "test done")
		close(serverDone)
	}()

	// Test DialQuic with WithSkipPortReuse option
	tlsClientConfig := &tls.Config{
		// nolint: gosec
		InsecureSkipVerify: true,
		NextProtos:         []string{"argotunnel"},
	}

	log := zerolog.New(io.Discard)
	dialCtx, dialCancel := context.WithTimeout(t.Context(), 5*time.Second)
	defer dialCancel()

	// Dial with skipPortReuse option - should use a random ephemeral port
	conn, err := DialQuic(
		dialCtx,
		testQUICConfig,
		tlsClientConfig,
		serverAddr,
		nil, // connect on a random port
		0,
		&log,
		dialopts.DialOpts{SkipPortReuse: true},
	)
	require.NoError(t, err)
	require.NotNil(t, conn)

	// Verify we can get connection state
	_ = conn.ConnectionState()