wireguard-operator/client_linux_integration_test.go at master · balboah/wireguard-operator · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
// +build integration

package operator

import (
	"encoding/base64"
	"testing"

	"github.com/vishvananda/netlink"
)

func TestWgClient(t *testing.T) {
	wg0, err := NewWgLink("wgo-test")
	if err != nil {
		t.Fatal(err)
	}
	defer func() { wg0.Close() }()

	addr4, err := netlink.ParseAddr("192.168.2.1/24")
	if err != nil {
		t.Fatal(err)
	}
	addr6, err := netlink.ParseAddr("fd01::/64")
	if err != nil {
		t.Fatal(err)
	}
	if err := wg0.AddrAdd(addr4, addr6); err != nil {
		t.Fatal(err)
	}

	c, err := NewWgClient(wg0, 1234, "")
	if err != nil {
		t.Fatal(err)
	}
	if p := c.PublicKey(); len(p) != 32 {
		t.Log(p)
		t.Error("did not generate key")
	}
	if c, err = NewWgClient(wg0, 1234, "QPaixVanELV/q/fTgSP3xtP2iJX2Alr8uXyWlU5NzEw="); err != nil {
		t.Fatal(err)
	}
	pub := base64.StdEncoding.EncodeToString(c.PublicKey())
	if pub != "VTqwfWmU9DKd48UrZ93BM7/9Rpn86Ang89WbUvSKwx8=" {
		t.Log(pub)
		t.Error("public key is invalid")
	}
}