From 7f1c9d1cc2131c49f6b77e90f050caf331df6179 Mon Sep 17 00:00:00 2001
From: "Jason A. Donenfeld" <Jason@zx2c4.com>
Date: Mon, 14 May 2018 14:47:31 +0200
Subject: [PATCH] Fix noise test

---
 noise_test.go | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

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")