Fixed KDF tests

1 files changed, 6 insertions, 3 deletions

diff --git a/src/kdf_test.go b/src/kdf_test.go
index 0cce81d..a89dacc 100644
--- a/src/kdf_test.go
+++ b/src/kdf_test.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"encoding/hex"
+	"golang.org/x/crypto/blake2s"
 	"testing"
 )
 
@@ -44,10 +45,12 @@ func TestKDF(t *testing.T) {
 		},
 	}
 
+	var t0, t1, t2 [blake2s.Size]byte
+
 	for _, test := range tests {
 		key, _ := hex.DecodeString(test.key)
 		input, _ := hex.DecodeString(test.input)
-		t0, t1, t2 := KDF3(key, input)
+		KDF3(&t0, &t1, &t2, key, input)
 		t0s := hex.EncodeToString(t0[:])
 		t1s := hex.EncodeToString(t1[:])
 		t2s := hex.EncodeToString(t2[:])
@@ -59,7 +62,7 @@ func TestKDF(t *testing.T) {
 	for _, test := range tests {
 		key, _ := hex.DecodeString(test.key)
 		input, _ := hex.DecodeString(test.input)
-		t0, t1 := KDF2(key, input)
+		KDF2(&t0, &t1, key, input)
 		t0s := hex.EncodeToString(t0[:])
 		t1s := hex.EncodeToString(t1[:])
 		assertEquals(t, t0s, test.t0)
@@ -69,7 +72,7 @@ func TestKDF(t *testing.T) {
 	for _, test := range tests {
 		key, _ := hex.DecodeString(test.key)
 		input, _ := hex.DecodeString(test.input)
-		t0 := KDF1(key, input)
+		KDF1(&t0, key, input)
 		t0s := hex.EncodeToString(t0[:])
 		assertEquals(t, t0s, test.t0)
 	}