Add new key tests

crypto/key_test.go

@@ -0,0 +1,38 @@
+package crypto
+
+import (
+	"github.com/stretchr/testify/assert"
+	// "encoding/base64"
+	"regexp"
+	"testing"
+)
+
+const name = "richard.stallman"
+const domain = "gnu.org"
+const passphrase = "I love GNU"
+
+var rsaKey, ecKey string
+
+func TestGenerateRsaKey(t *testing.T) {
+	var pmCrypto = PmCrypto{}
+	var err error
+	rsaKey, err = pmCrypto.generateKey(name, domain, passphrase, "RSA", 1024, nil, nil, nil, nil)
+	if err != nil {
+		t.Fatal("Cannot encrypt token:", err)
+	}
+
+	rTest := regexp.MustCompile("(?s)^-----BEGIN PGP PRIVATE KEY BLOCK-----.*-----END PGP PRIVATE KEY BLOCK-----$")
+	assert.Regexp(t, rTest, rsaKey)
+}
+
+func TestGenerateECKey(t *testing.T) {
+	var pmCrypto = PmCrypto{}
+	var err error
+	ecKey, err = pmCrypto.generateKey(name, domain, passphrase, "x25519", 1024, nil, nil, nil, nil)
+	if err != nil {
+		t.Fatal("Cannot encrypt token:", err)
+	}
+
+	rTest := regexp.MustCompile("(?s)^-----BEGIN PGP PRIVATE KEY BLOCK-----.*-----END PGP PRIVATE KEY BLOCK-----$")
+	assert.Regexp(t, rTest, ecKey)
+}