gluetun/internal/amneziawg/constructor_test.go

package amneziawg

import (
	"net/netip"
	"testing"

	"github.com/qdm12/gluetun/internal/wireguard"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"golang.zx2c4.com/wireguard/device"
)

func Test_New(t *testing.T) {
	t.Parallel()

	const validKeyString = "oMNSf/zJ0pt1ciy+qIRk8Rlyfs9accwuRLnKd85Yl1Q="
	logger := NewMockLogger(nil)
	netLinker := NewMockNetLinker(nil)

	testCases := map[string]struct {
		settings  Settings
		amneziawg *Amneziawg
		err       error
	}{
		"bad_settings": {
			settings: Settings{
				Wireguard: wireguard.Settings{
					PrivateKey: "",
				},
			},
			err: wireguard.ErrPrivateKeyMissing,
		},
		"minimal valid settings": {
			settings: Settings{
				Wireguard: wireguard.Settings{
					PrivateKey: validKeyString,
					PublicKey:  validKeyString,
					Endpoint:   netip.AddrPortFrom(netip.AddrFrom4([4]byte{1, 2, 3, 4}), 0),
					Addresses: []netip.Prefix{
						netip.PrefixFrom(netip.AddrFrom4([4]byte{5, 6, 7, 8}), 32),
					},
					FirewallMark: 100,
				},
			},
			amneziawg: &Amneziawg{
				logger:  logger,
				netlink: netLinker,
				settings: Settings{
					Wireguard: wireguard.Settings{
						InterfaceName: "wg0",
						PrivateKey:    validKeyString,
						PublicKey:     validKeyString,
						Endpoint:      netip.AddrPortFrom(netip.AddrFrom4([4]byte{1, 2, 3, 4}), 51820),
						Addresses: []netip.Prefix{
							netip.PrefixFrom(netip.AddrFrom4([4]byte{5, 6, 7, 8}), 32),
						},
						AllowedIPs: []netip.Prefix{
							netip.MustParsePrefix("0.0.0.0/0"),
						},
						FirewallMark:   100,
						MTU:            device.DefaultMTU,
						IPv6:           ptrTo(false),
						Implementation: "auto",
					},
				},
			},
		},
	}

	for name, testCase := range testCases {
		t.Run(name, func(t *testing.T) {
			t.Parallel()

			wireguard, err := New(testCase.settings, netLinker, logger)

			if testCase.err != nil {
				require.Error(t, err)
				assert.Equal(t, testCase.err.Error(), err.Error())
			} else {
				assert.NoError(t, err)
			}

			assert.Equal(t, testCase.amneziawg, wireguard)
		})
	}
}