Add test.

This commit is contained in:
Jay
2025-10-27 17:40:46 -04:00
parent 7d7e69aa24
commit 223c9faec0

View File

@@ -2,21 +2,24 @@ package roots
import ( import (
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
"regexp"
"testing" "testing"
) )
var hexPattern = regexp.MustCompile("^[a-f0-9]{64}$")
func TestGeneratePrivateKey(t *testing.T) { func TestGeneratePrivateKey(t *testing.T) {
sk, err := GeneratePrivateKey() sk, err := GeneratePrivateKey()
assert.NoError(t, err) assert.NoError(t, err)
if !hexPattern.MatchString(sk) { if !Hex64Pattern.MatchString(sk) {
t.Errorf("invalid private key format: %s", 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) { func TestGetPublicKey(t *testing.T) {
pk, err := GetPublicKey(testSK) pk, err := GetPublicKey(testSK)