add more test case for crypto/pki

crypto/pki/utils_test.go

@@ -7,19 +7,48 @@ import (
 	"time"
 )
 
-func TestGetPublicKeyFromPrivateKey(test *testing.T) {
+func TestGetAddressFromPublicKey(test *testing.T) {
 	suite := crypto.Ed25519Curve
 	t := time.Now().UnixNano()
 	scalar := suite.Scalar().SetInt64(t)
 	pubKey := GetPublicKeyFromScalar(scalar)
+	addr1 := GetAddressFromPublicKey(pubKey)
+	addr2 := GetAddressFromPrivateKey(scalar)
+	if !reflect.DeepEqual(addr1, addr2) {
+		test.Error("two public address should be equal")
+	}
+}
+
+func TestGetPublicKeyFromPrivateKey(test *testing.T) {
+	suite := crypto.Ed25519Curve
+	t := time.Now().UnixNano()
+	scalar := suite.Scalar().SetInt64(t)
+	pubKey1 := GetPublicKeyFromScalar(scalar)
 	bytes := [32]byte{}
 	tmp, err := scalar.MarshalBinary()
 	copy(bytes[:], tmp)
 	if err != nil {
 		test.Error("unable to marshal private key to binary")
 	}
-	pubKey1 := GetPublicKeyFromPrivateKey(bytes)
-	if !reflect.DeepEqual(pubKey, pubKey1) {
+	pubKey2 := GetPublicKeyFromPrivateKey(bytes)
+	if !reflect.DeepEqual(pubKey1, pubKey2) {
 		test.Error("two public keys should be equal")
 	}
 }
+
+func TestGetPrivateKeyFromInt(test *testing.T) {
+	t := int(time.Now().UnixNano())
+	priKey1 := GetPrivateKeyFromInt(t)
+	priKeyScalar := GetPrivateKeyScalarFromInt(t)
+	tmp, err := priKeyScalar.MarshalBinary()
+	priKey2 := [32]byte{}
+	copy(priKey2[:], tmp)
+
+	if err != nil {
+		test.Error("unable to marshal private key to binary")
+	}
+	if !reflect.DeepEqual(priKey1, priKey2) {
+		test.Error("two private keys should be equal")
+	}
+
+}