From 0af43a29b7cabb6710cd1cb335785ff60dbf758f Mon Sep 17 00:00:00 2001
From: Niall Sheridan <nsheridan@gmail.com>
Date: Sat, 3 Sep 2016 19:14:13 +0100
Subject: Move signing & agent logic out of the main package

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

(limited to 'cmd/cashier/client/keys_test.go')

diff --git a/cmd/cashier/client/keys_test.go b/cmd/cashier/client/keys_test.go
new file mode 100644
index 0000000..9e930d5
--- /dev/null
+++ b/cmd/cashier/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