From a30d6403f723765b8f9b7609e7eb3ade0f5434a0 Mon Sep 17 00:00:00 2001
From: Niall Sheridan <nsheridan@gmail.com>
Date: Sat, 10 Sep 2016 17:40:23 +0100
Subject: Make client a top-level package for consistency

---
 client/keys_test.go | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 client/keys_test.go

(limited to 'client/keys_test.go')

diff --git a/client/keys_test.go b/client/keys_test.go
new file mode 100644
index 0000000..9e930d5
--- /dev/null
+++ b/client/keys_test.go
@@ -0,0 +1,28 @@
+package client
+
+import (
+	"reflect"
+	"testing"
+)
+
+func TestGenerateKeys(t *testing.T) {
+	var tests = []struct {
+		key  string
+		size int
+		want string
+	}{
+		{"ecdsa", 256, "*ecdsa.PrivateKey"},
+		{"rsa", 1024, "*rsa.PrivateKey"},
+		{"ed25519", 256, "*ed25519.PrivateKey"},
+	}
+
+	for _, tst := range tests {
+		k, _, err := GenerateKey(tst.key, tst.size)
+		if err != nil {
+			t.Error(err)
+		}
+		if reflect.TypeOf(k).String() != tst.want {
+			t.Errorf("Wrong key type returned. Got %s, wanted %s", reflect.TypeOf(k).String(), tst.want)
+		}
+	}
+}
-- 
cgit v1.2.3