From: Jason A. Donenfeld <Jason@zx2c4.com>
Date: Mon, 14 May 2018 12:47:31 +0000 (+0200)
Subject: Fix noise test
X-Git-Tag: 0.0.20180514~13
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=7f1c9d1cc2131c49f6b77e90f050caf331df6179;p=thirdparty%2Fwireguard-go.git

Fix noise test
---

diff --git a/noise_test.go b/noise_test.go
index 8e1bd89..b165c0f 100644
--- a/noise_test.go
+++ b/noise_test.go
@@ -102,17 +102,19 @@ func TestNoiseHandshake(t *testing.T) {
 
 	t.Log("deriving keys")
 
-	key1 := peer1.BeginSymmetricSession()
-	key2 := peer2.BeginSymmetricSession()
-
-	if key1 == nil {
-		t.Fatal("failed to dervice keypair for peer 1")
+	err = peer1.BeginSymmetricSession()
+	if err != nil {
+		t.Fatal("failed to derive keypair for peer 1", err)
 	}
 
-	if key2 == nil {
-		t.Fatal("failed to dervice keypair for peer 2")
+	err = peer2.BeginSymmetricSession()
+	if err != nil {
+		t.Fatal("failed to derive keypair for peer 2", err)
 	}
 
+	key1 := peer1.keypairs.next
+	key2 := peer2.keypairs.current
+
 	// encrypting / decryption test
 
 	t.Log("test key pairs")