Refactored into namespaced packages.

keys/keys_test.go

@@ -0,0 +1,39 @@
+package keys
+
+import (
+	"github.com/stretchr/testify/assert"
+	"regexp"
+	"testing"
+)
+
+const testSK = "f43a0435f69529f310bbd1d6263d2fbf0977f54bfe2310cc37ae5904b83bb167"
+const testPK = "cfa87f35acbde29ba1ab3ee42de527b2cad33ac487e80cf2d6405ea0042c8fef"
+
+var Hex64Pattern = regexp.MustCompile("^[a-f0-9]{64}$")
+
+func TestGeneratePrivateKey(t *testing.T) {
+	sk, err := GeneratePrivateKey()
+
+	assert.NoError(t, err)
+	if !Hex64Pattern.MatchString(sk) {
+		t.Errorf("invalid private key format: %s", sk)
+	}
+}
+
+func TestGenerateUniquePrivateKeys(t *testing.T) {
+	sk1, _ := GeneratePrivateKey()
+	sk2, _ := GeneratePrivateKey()
+	assert.NotEqual(t, sk1, sk2)
+}
+
+func TestGetPublicKey(t *testing.T) {
+	pk, err := GetPublicKey(testSK)
+
+	assert.NoError(t, err)
+	assert.Equal(t, testPK, pk)
+}
+
+func TestGetPublicKeyInvalidPrivateKey(t *testing.T) {
+	_, err := GetPublicKey("abc123")
+	assert.ErrorContains(t, err, "private key must be 64 lowercase hex characters")
+}