package pki

import (
	"github.com/harmony-one/harmony/crypto"
	"reflect"
	"testing"
	"time"
)

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")
	}
	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")
	}

}