aboutsummaryrefslogtreecommitdiff
path: root/device/cookie_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'device/cookie_test.go')
-rw-r--r--device/cookie_test.go6
1 files changed, 4 insertions, 2 deletions
diff --git a/device/cookie_test.go b/device/cookie_test.go
index 79a6a86..ef01d46 100644
--- a/device/cookie_test.go
+++ b/device/cookie_test.go
@@ -7,6 +7,8 @@ package device
import (
"testing"
+
+ "golang.zx2c4.com/wireguard/wgcfg"
)
func TestCookieMAC1(t *testing.T) {
@@ -18,11 +20,11 @@ func TestCookieMAC1(t *testing.T) {
checker CookieChecker
)
- sk, err := newPrivateKey()
+ sk, err := wgcfg.NewPrivateKey()
if err != nil {
t.Fatal(err)
}
- pk := sk.publicKey()
+ pk := sk.Public()
generator.Init(pk)
checker.Init(pk)