package generator

import (
	"testing"

	"github.com/ethereum/go-ethereum/accounts/keystore"
	"github.com/status-im/status-go/extkeys"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func generateTestKey(t *testing.T) *extkeys.ExtendedKey {
	mnemonic := extkeys.NewMnemonic()
	mnemonicPhrase, err := mnemonic.MnemonicPhrase(extkeys.EntropyStrength128, extkeys.EnglishLanguage)
	require.NoError(t, err)

	masterExtendedKey, err := extkeys.NewMaster(mnemonic.MnemonicSeed(mnemonicPhrase, ""))
	require.NoError(t, err)

	return masterExtendedKey
}

func TestValidateKeystoreExtendedKey(t *testing.T) {
	extendedKey1 := generateTestKey(t)
	extendedKey2 := generateTestKey(t)

	// new keystore file format
	key := &keystore.Key{
		PrivateKey:  extendedKey1.ToECDSA(),
		ExtendedKey: extendedKey1,
	}
	assert.NoError(t, ValidateKeystoreExtendedKey(key))

	// old keystore file format where the extended key was
	// from another derivation path and not the same of the private key
	oldKey := &keystore.Key{
		PrivateKey:  extendedKey1.ToECDSA(),
		ExtendedKey: extendedKey2,
	}
	assert.Error(t, ValidateKeystoreExtendedKey(oldKey))

	// normal key where we don't have an extended key
	normalKey := &keystore.Key{
		PrivateKey:  extendedKey1.ToECDSA(),
		ExtendedKey: nil,
	}
	assert.NoError(t, ValidateKeystoreExtendedKey(normalKey))
}