TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2024-11-21 23:29:38 +00:00
Code Issues Projects Releases Packages Wiki Activity

wallet: fix scrypt parameters in wallet regeneration tests

Browse source 
Default NEP2 scrypt parameters should be overriden where needed,
otherwise our regeneration tests won't generate proper wallets.

Signed-off-by: Anna Shaleva <shaleva.ann@nspcc.ru>
This commit is contained in:
Anna Shaleva 2024-03-01 00:05:32 +03:00
parent 685d095fa4
commit 9525bc7785
1 changed files with 61 additions and 44 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

105
pkg/wallet/regenerate_test.go
View file

@ -29,6 +29,10 @@ var (
} }
passwords = []string{"one", "two", "three", "four"} passwords = []string{"one", "two", "three", "four"}
// naiveScrypt is naive scrypt parameters used for testing wallets to save
// time on accounts decryption.
naiveScrypt = keys.ScryptParams{N: 2, R: 1, P: 1}
) )
func getKeys(t *testing.T) []*keys.PublicKey { func getKeys(t *testing.T) []*keys.PublicKey {
@ -42,10 +46,14 @@ func getKeys(t *testing.T) []*keys.PublicKey {
return pubs return pubs
} }
func getAccount(t *testing.T, wif, pass string) *Account { func getNEP2Account(t *testing.T, wif, pass string) *Account {
return getAccountWithScrypt(t, wif, pass, keys.NEP2ScryptParams()) return getAccountWithScrypt(t, wif, pass, keys.NEP2ScryptParams())
} }
func getTestingAccount(t *testing.T, wif, pass string) *Account {
return getAccountWithScrypt(t, wif, pass, naiveScrypt)
}
func getAccountWithScrypt(t *testing.T, wif, pass string, scrypt keys.ScryptParams) *Account { func getAccountWithScrypt(t *testing.T, wif, pass string, scrypt keys.ScryptParams) *Account {
acc, err := NewAccountFromWIF(wif) acc, err := NewAccountFromWIF(wif)
require.NoError(t, err) require.NoError(t, err)
@ -59,24 +67,24 @@ func TestRegenerateSoloWallet(t *testing.T) {
} }
walletPath := filepath.Join(dockerWalletDir, "wallet1_solo.json") walletPath := filepath.Join(dockerWalletDir, "wallet1_solo.json")
wif := privnetWIFs[0] wif := privnetWIFs[0]
acc1 := getAccount(t, wif, "one") acc1 := getNEP2Account(t, wif, "one")
acc2 := getAccount(t, wif, "one") acc2 := getNEP2Account(t, wif, "one")
require.NoError(t, acc2.ConvertMultisig(3, getKeys(t))) require.NoError(t, acc2.ConvertMultisig(3, getKeys(t)))
acc3 := getAccount(t, wif, "one") acc3 := getNEP2Account(t, wif, "one")
require.NoError(t, acc3.ConvertMultisig(1, keys.PublicKeys{getKeys(t)[0]})) require.NoError(t, acc3.ConvertMultisig(1, keys.PublicKeys{getKeys(t)[0]}))
createWallet(t, walletPath, acc1, acc2, acc3) createNEP2Wallet(t, walletPath, acc1, acc2, acc3)
} }
func regenerateWallets(t *testing.T, dir string) { func regenerateWallets(t *testing.T, dir string) {
pubs := getKeys(t) pubs := getKeys(t)
for i := range privnetWIFs { for i := range privnetWIFs {
acc1 := getAccount(t, privnetWIFs[i], passwords[i]) acc1 := getNEP2Account(t, privnetWIFs[i], passwords[i])
acc2 := getAccount(t, privnetWIFs[i], passwords[i]) acc2 := getNEP2Account(t, privnetWIFs[i], passwords[i])
require.NoError(t, acc2.ConvertMultisig(3, pubs)) require.NoError(t, acc2.ConvertMultisig(3, pubs))
createWallet(t, filepath.Join(dir, fmt.Sprintf("wallet%d.json", i+1)), acc1, acc2) createNEP2Wallet(t, filepath.Join(dir, fmt.Sprintf("wallet%d.json", i+1)), acc1, acc2)
} }
} }
@ -99,17 +107,17 @@ func TestRegenerateWalletTestdata(t *testing.T) {
} }
const walletDir = "./testdata/" const walletDir = "./testdata/"
acc1 := getAccount(t, privnetWIFs[0], "one") acc1 := getTestingAccount(t, privnetWIFs[0], "one")
acc2 := getAccount(t, privnetWIFs[0], "one") acc2 := getTestingAccount(t, privnetWIFs[0], "one")
pubs := getKeys(t) pubs := getKeys(t)
require.NoError(t, acc2.ConvertMultisig(3, pubs)) require.NoError(t, acc2.ConvertMultisig(3, pubs))
acc3 := getAccount(t, privnetWIFs[1], "two") acc3 := getTestingAccount(t, privnetWIFs[1], "two")
acc3.Default = true acc3.Default = true
createWallet(t, filepath.Join(walletDir, "wallet1.json"), acc1, acc2) createTestingWallet(t, filepath.Join(walletDir, "wallet1.json"), acc1, acc2)
createWallet(t, filepath.Join(walletDir, "wallet2.json"), acc1, acc2, acc3) createTestingWallet(t, filepath.Join(walletDir, "wallet2.json"), acc1, acc2, acc3)
} }
func TestRegenerateNotaryWallets(t *testing.T) { func TestRegenerateNotaryWallets(t *testing.T) {
@ -123,15 +131,13 @@ func TestRegenerateNotaryWallets(t *testing.T) {
acc4WIF = "L1ioz93TNt6Nu1aoMpZQ4zgdtgC8ZvJMC6pyHFkrovdR3SFwbn6n" acc4WIF = "L1ioz93TNt6Nu1aoMpZQ4zgdtgC8ZvJMC6pyHFkrovdR3SFwbn6n"
) )
var walletDir = filepath.Join("..", "services", "notary", "testdata") var walletDir = filepath.Join("..", "services", "notary", "testdata")
acc1 := getAccountWithScrypt(t, acc1WIF, "one", naiveScrypt)
acc2 := getAccountWithScrypt(t, acc2WIF, "one", naiveScrypt)
acc3 := getAccountWithScrypt(t, acc3WIF, "four", naiveScrypt)
createTestingWallet(t, filepath.Join(walletDir, "notary1.json"), acc1, acc2, acc3)
scryptParams := keys.ScryptParams{N: 2, R: 1, P: 1} acc4 := getAccountWithScrypt(t, acc4WIF, "two", naiveScrypt)
acc1 := getAccountWithScrypt(t, acc1WIF, "one", scryptParams) createTestingWallet(t, filepath.Join(walletDir, "notary2.json"), acc4)
acc2 := getAccountWithScrypt(t, acc2WIF, "one", scryptParams)
acc3 := getAccountWithScrypt(t, acc3WIF, "four", scryptParams)
createWallet(t, filepath.Join(walletDir, "notary1.json"), acc1, acc2, acc3)
acc4 := getAccountWithScrypt(t, acc4WIF, "two", scryptParams)
createWallet(t, filepath.Join(walletDir, "notary2.json"), acc4)
} }
func TestRegenerateOracleWallets(t *testing.T) { func TestRegenerateOracleWallets(t *testing.T) {
@ -144,11 +150,11 @@ func TestRegenerateOracleWallets(t *testing.T) {
acc2WIF = "KyA8z2MyLCSjJFG3F4SUp85CZ4WJm4qgWihFJZFEDYGEyw8oGcEP" acc2WIF = "KyA8z2MyLCSjJFG3F4SUp85CZ4WJm4qgWihFJZFEDYGEyw8oGcEP"
) )
acc1 := getAccount(t, acc1WIF, "one") acc1 := getTestingAccount(t, acc1WIF, "one")
createWallet(t, filepath.Join(walletDir, "oracle1.json"), acc1) createTestingWallet(t, filepath.Join(walletDir, "oracle1.json"), acc1)
acc2 := getAccount(t, acc2WIF, "two") acc2 := getTestingAccount(t, acc2WIF, "two")
createWallet(t, filepath.Join(walletDir, "oracle2.json"), acc2) createTestingWallet(t, filepath.Join(walletDir, "oracle2.json"), acc2)
} }
func TestRegenerateExamplesWallet(t *testing.T) { func TestRegenerateExamplesWallet(t *testing.T) {
@ -160,9 +166,9 @@ func TestRegenerateExamplesWallet(t *testing.T) {
acc1WIF = "L46dn46AMZY7NQGZHemAdgcMabKon85eme45hgQkAUQBiRacY8MB" acc1WIF = "L46dn46AMZY7NQGZHemAdgcMabKon85eme45hgQkAUQBiRacY8MB"
) )
acc1 := getAccount(t, acc1WIF, "qwerty") acc1 := getNEP2Account(t, acc1WIF, "qwerty")
acc1.Label = "my_account" acc1.Label = "my_account"
createWallet(t, walletPath, acc1) createNEP2Wallet(t, walletPath, acc1)
} }
func TestRegenerateCLITestwallet(t *testing.T) { func TestRegenerateCLITestwallet(t *testing.T) {
@ -174,9 +180,9 @@ func TestRegenerateCLITestwallet(t *testing.T) {
accWIF = "L23LrQNWELytYLvb5c6dXBDdF2DNPL9RRNWPqppv3roxacSnn8CN" accWIF = "L23LrQNWELytYLvb5c6dXBDdF2DNPL9RRNWPqppv3roxacSnn8CN"
) )
acc := getAccountWithScrypt(t, accWIF, "testpass", keys.ScryptParams{N: 2, R: 1, P: 1}) acc := getAccountWithScrypt(t, accWIF, "testpass", naiveScrypt)
acc.Label = "kek" acc.Label = "kek"
createWallet(t, walletPath, acc) createTestingWallet(t, walletPath, acc)
} }
func TestRegenerateCLITestwalletMulti(t *testing.T) { func TestRegenerateCLITestwalletMulti(t *testing.T) {
@ -193,14 +199,13 @@ func TestRegenerateCLITestwalletMulti(t *testing.T) {
pass3 = "three" pass3 = "three"
) )
scryptParams := keys.ScryptParams{N: 2, R: 1, P: 1} acc1 := getAccountWithScrypt(t, accWIF1, pass1, naiveScrypt)
acc1 := getAccountWithScrypt(t, accWIF1, pass1, scryptParams)
acc1.Label = pass1 acc1.Label = pass1
acc2 := getAccountWithScrypt(t, accWIF2, pass2, scryptParams) acc2 := getAccountWithScrypt(t, accWIF2, pass2, naiveScrypt)
acc2.Label = pass2 acc2.Label = pass2
acc3 := getAccountWithScrypt(t, accWIF3, pass3, scryptParams) acc3 := getAccountWithScrypt(t, accWIF3, pass3, naiveScrypt)
acc3.Label = pass3 acc3.Label = pass3
createWallet(t, walletPath, acc1, acc2, acc3) createTestingWallet(t, walletPath, acc1, acc2, acc3)
} }
func TestRegenerateCLITestwallet_NEO3(t *testing.T) { func TestRegenerateCLITestwallet_NEO3(t *testing.T) {
@ -210,15 +215,28 @@ func TestRegenerateCLITestwallet_NEO3(t *testing.T) {
const walletPath = "../../cli/wallet/testdata/testwallet_NEO3.json" const walletPath = "../../cli/wallet/testdata/testwallet_NEO3.json"
pubs := getKeys(t) pubs := getKeys(t)
acc1 := getAccount(t, privnetWIFs[0], passwords[0]) acc1 := getNEP2Account(t, privnetWIFs[0], passwords[0])
acc2 := getAccount(t, privnetWIFs[0], passwords[0]) acc2 := getNEP2Account(t, privnetWIFs[0], passwords[0])
require.NoError(t, acc2.ConvertMultisig(3, pubs)) require.NoError(t, acc2.ConvertMultisig(3, pubs))
createWallet(t, walletPath, acc1, acc2) createNEP2Wallet(t, walletPath, acc1, acc2)
} }
func createWallet(t *testing.T, path string, accs ...*Account) { // createNEP2Wallet creates wallet with provided accounts and NEP2 scrypt parameters.
func createNEP2Wallet(t *testing.T, path string, accs ...*Account) {
createWallet(t, path, keys.NEP2ScryptParams(), accs...)
}
// createTestingWallet creates wallet with provided accounts and scrypt parameters with
// low values which can be used for testing to save time on accounts decryption.
func createTestingWallet(t *testing.T, path string, accs ...*Account) {
createWallet(t, path, naiveScrypt, accs...)
}
func createWallet(t *testing.T, path string, scryptParams keys.ScryptParams, accs ...*Account) {
w, err := NewWallet(path) w, err := NewWallet(path)
require.NoError(t, err) require.NoError(t, err)
w.Scrypt = scryptParams
if len(accs) == 0 { if len(accs) == 0 {
t.Fatal("provide at least 1 account") t.Fatal("provide at least 1 account")
} }
@ -239,17 +257,16 @@ func TestRegenerateCLIWallet1_solo(t *testing.T) {
verifyManifestPath = "../../cli/smartcontract/testdata/verify.manifest.json" verifyManifestPath = "../../cli/smartcontract/testdata/verify.manifest.json"
) )
scrypt := keys.ScryptParams{N: 2, R: 1, P: 1}
wif := privnetWIFs[0] wif := privnetWIFs[0]
acc1 := getAccountWithScrypt(t, wif, "one", scrypt) acc1 := getAccountWithScrypt(t, wif, "one", naiveScrypt)
acc1.Default = true acc1.Default = true
acc2 := getAccountWithScrypt(t, wif, "one", scrypt) acc2 := getAccountWithScrypt(t, wif, "one", naiveScrypt)
require.NoError(t, acc2.ConvertMultisig(3, getKeys(t))) require.NoError(t, acc2.ConvertMultisig(3, getKeys(t)))
acc3 := getAccountWithScrypt(t, wif, "one", scrypt) acc3 := getAccountWithScrypt(t, wif, "one", naiveScrypt)
require.NoError(t, acc3.ConvertMultisig(1, keys.PublicKeys{getKeys(t)[0]})) require.NoError(t, acc3.ConvertMultisig(1, keys.PublicKeys{getKeys(t)[0]}))
acc4 := getAccountWithScrypt(t, verifyWIF, "pass", scrypt) // deployed verify.go contract acc4 := getAccountWithScrypt(t, verifyWIF, "pass", naiveScrypt) // deployed verify.go contract
f, err := os.ReadFile(verifyNEFPath) f, err := os.ReadFile(verifyNEFPath)
require.NoError(t, err) require.NoError(t, err)
nefFile, err := nef.FileFromBytes(f) nefFile, err := nef.FileFromBytes(f)
@ -266,5 +283,5 @@ func TestRegenerateCLIWallet1_solo(t *testing.T) {
Parameters: []ContractParam{}, Parameters: []ContractParam{},
} }
acc4.Label = "verify" acc4.Label = "verify"
createWallet(t, walletPath, acc1, acc2, acc3, acc4) createTestingWallet(t, walletPath, acc1, acc2, acc3, acc4)
} }