conn_udp_test.go

// SPDX-FileCopyrightText: 2023 Steffen Vogel <post@steffenvogel.de>
// SPDX-License-Identifier: Apache-2.0

package rosenpass

import (
	"net"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestUDPConn(t *testing.T) {
	require := require.New(t)

	spk, _, err := generateStaticKeyPair()
	require.NoError(err)

	ep := net.UDPAddr{
		IP:   net.IPv6loopback,
		Port: 1234,
	}

	p := &peer{
		spkt:     spk,
		endpoint: (*UDPEndpoint)(&ep),
	}

	c, err := NewUDPConn([]*net.UDPAddr{
		{
			Port: 1234,
		},
	})
	require.NoError(err)

	sid, err := generateSessionID()
	require.NoError(err)

	recvFncs, err := c.Open()
	require.NoError(err)

	pls := make(chan payload)
	for _, recvFnc := range recvFncs {
		go func(recvFnc ReceiveFunc) {
			pl, _, err := recvFnc(spk)
			require.NoError(err)

			pls <- pl
		}(recvFnc)
	}

	pl := &emptyData{
		sid: sid,
	}

	err = c.Send(pl, spk, p.endpoint)
	require.NoError(err)

	edr := <-pls
	require.Equal(edr, pl)

	err = c.Close()
	require.NoError(err)
}