package owner

import (
	"crypto/ecdsa"
	"crypto/elliptic"
	"encoding/hex"
	"testing"

	"github.com/mr-tron/base58"
	"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
	"github.com/stretchr/testify/require"
)

func TestNEO3WalletFromPublicKey(t *testing.T) {
	rawPub, _ := hex.DecodeString("0369b7b6c49fb937f3de52af189b91069767679c2739798d85f2ed69c079940680")
	x, y := elliptic.UnmarshalCompressed(elliptic.P256(), rawPub)
	require.True(t, x != nil && y != nil)

	expected := "35ee628f21922d7308f1bd71f03a0d8ba89c4e7372fca1442c"
	actual := PublicKeyToIDBytes(&ecdsa.PublicKey{Curve: elliptic.P256(), X: x, Y: y})
	require.Equal(t, expected, hex.EncodeToString(actual))
}

func TestPublicKeyToBytes(t *testing.T) {
	p, err := keys.NewPrivateKey()
	require.NoError(t, err)

	expected, err := base58.Decode(p.PublicKey().Address())
	require.NoError(t, err)

	actual := PublicKeyToIDBytes((*ecdsa.PublicKey)(p.PublicKey()))
	require.Equal(t, expected, actual)
	require.Equal(t, NEO3WalletSize, len(actual))
}

func TestScriptHashToIDBytes(t *testing.T) {
	p, err := keys.NewPrivateKey()
	require.NoError(t, err)

	expected, err := base58.Decode(p.PublicKey().Address())
	require.NoError(t, err)

	actual := ScriptHashToIDBytes(p.GetScriptHash())
	require.Equal(t, expected, actual)
	require.Equal(t, NEO3WalletSize, len(actual))
}