monorepo

Co-authored-by: Malte Poll <mp@edgeless.systems>
Co-authored-by: katexochen <katexochen@users.noreply.github.com>
Co-authored-by: Daniel Weiße <dw@edgeless.systems>
Co-authored-by: Thomas Tendyck <tt@edgeless.systems>
Co-authored-by: Benedict Schlueter <bs@edgeless.systems>
Co-authored-by: leongross <leon.gross@rub.de>
Co-authored-by: Moritz Eckert <m1gh7ym0@gmail.com>

cli/vpn/configurer_test.go

@@ -0,0 +1,100 @@
+package vpn
+
+import (
+	"fmt"
+	"net"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"github.com/vishvananda/netlink"
+	"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
+)
+
+type stubNetworkLink struct {
+	link netlink.Link
+	addr string
+	up   bool
+}
+
+func newStubNetworkLink() *stubNetworkLink {
+	return &stubNetworkLink{}
+}
+
+func (s *stubNetworkLink) LinkAdd(link netlink.Link) error {
+	s.link = link
+	return nil
+}
+
+func (s *stubNetworkLink) LinkByName(name string) (netlink.Link, error) {
+	if name != s.link.Attrs().Name {
+		return nil, fmt.Errorf("could not find interface with name %v", name)
+	}
+	return s.link, nil
+}
+
+func (s *stubNetworkLink) ParseAddr(addr string) (*netlink.Addr, error) {
+	return netlink.ParseAddr(addr)
+}
+
+func (s *stubNetworkLink) AddrAdd(link netlink.Link, addr *netlink.Addr) error {
+	if link.Attrs().Name != s.link.Attrs().Name {
+		return fmt.Errorf("could not find interface with name %v", link.Attrs().Name)
+	}
+	s.addr = addr.IP.String()
+	return nil
+}
+
+func (s *stubNetworkLink) LinkSetUp(link netlink.Link) error {
+	if link.Attrs().Name != s.link.Attrs().Name {
+		return fmt.Errorf("could not find interface with name %v", link.Attrs().Name)
+	}
+	s.up = true
+	return nil
+}
+
+type stubVPN struct {
+	name   string
+	config wgtypes.Config
+}
+
+func newStubVPN() *stubVPN {
+	return &stubVPN{}
+}
+
+func (s *stubVPN) ConfigureDevice(name string, cfg wgtypes.Config) error {
+	s.name = name
+	s.config = cfg
+	return nil
+}
+
+func TestVPNClient(t *testing.T) {
+	assert := assert.New(t)
+	require := require.New(t)
+
+	link := newStubNetworkLink()
+	vpn := newStubVPN()
+	client, err := New(link, vpn)
+	require.NoError(err)
+	coordinatorPubKey, err := wgtypes.GenerateKey()
+	require.NoError(err)
+	clientPrivKey, err := wgtypes.GenerateKey()
+	require.NoError(err)
+	clientVpnIp := "192.0.2.1"
+	coordinatorPubIp := "192.0.2.2"
+	assert.NoError(client.Configure(clientVpnIp, coordinatorPubKey.String(), coordinatorPubIp, clientPrivKey.String()))
+
+	// assert expected interface
+	assert.Equal(interfaceName, link.link.Attrs().Name)
+	assert.NotNil(link.addr)
+	assert.True(link.up)
+
+	// assert vpn config
+	config := client.vpn.(*stubVPN).config
+	assert.Equal(wireguardPort, *config.ListenPort)
+	assert.Equal(clientPrivKey, *config.PrivateKey)
+	assert.Less(0, len(config.Peers))
+	assert.Equal(coordinatorPubKey, config.Peers[0].PublicKey)
+	assert.Equal(net.JoinHostPort(coordinatorPubIp, "51820"), config.Peers[0].Endpoint.String())
+	assert.Equal("10.118.0.1/32", config.Peers[0].AllowedIPs[0].String())
+}