75 lines
2.2 KiB
Go
75 lines
2.2 KiB
Go
package server_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/cosmos/cosmos-sdk/crypto/hd"
|
|
"github.com/cosmos/cosmos-sdk/crypto/keyring"
|
|
"github.com/cosmos/cosmos-sdk/server"
|
|
"github.com/cosmos/cosmos-sdk/simapp"
|
|
"github.com/cosmos/cosmos-sdk/types"
|
|
)
|
|
|
|
func TestGenerateCoinKey(t *testing.T) {
|
|
t.Parallel()
|
|
cdc := simapp.MakeTestEncodingConfig().Codec
|
|
addr, mnemonic, err := server.GenerateCoinKey(hd.Secp256k1, cdc)
|
|
require.NoError(t, err)
|
|
|
|
// Test creation
|
|
k, err := keyring.NewInMemory(cdc).NewAccount("xxx", mnemonic, "", hd.NewFundraiserParams(0, types.GetConfig().GetCoinType(), 0).String(), hd.Secp256k1)
|
|
require.NoError(t, err)
|
|
addr1, err := k.GetAddress()
|
|
require.NoError(t, err)
|
|
require.Equal(t, addr, addr1)
|
|
}
|
|
|
|
func TestGenerateSaveCoinKey(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
encCfg := simapp.MakeTestEncodingConfig()
|
|
kb, err := keyring.New(t.Name(), "test", t.TempDir(), nil, encCfg.Codec)
|
|
require.NoError(t, err)
|
|
|
|
addr, mnemonic, err := server.GenerateSaveCoinKey(kb, "keyname", false, hd.Secp256k1)
|
|
require.NoError(t, err)
|
|
|
|
// Test key was actually saved
|
|
k, err := kb.Key("keyname")
|
|
require.NoError(t, err)
|
|
addr1, err := k.GetAddress()
|
|
require.NoError(t, err)
|
|
require.Equal(t, addr, addr1)
|
|
|
|
// Test in-memory recovery
|
|
k, err = keyring.NewInMemory(encCfg.Codec).NewAccount("xxx", mnemonic, "", hd.NewFundraiserParams(0, types.GetConfig().GetCoinType(), 0).String(), hd.Secp256k1)
|
|
require.NoError(t, err)
|
|
addr1, err = k.GetAddress()
|
|
require.NoError(t, err)
|
|
require.Equal(t, addr, addr1)
|
|
}
|
|
|
|
func TestGenerateSaveCoinKeyOverwriteFlag(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
encCfg := simapp.MakeTestEncodingConfig()
|
|
kb, err := keyring.New(t.Name(), "test", t.TempDir(), nil, encCfg.Codec)
|
|
require.NoError(t, err)
|
|
|
|
keyname := "justakey"
|
|
addr1, _, err := server.GenerateSaveCoinKey(kb, keyname, false, hd.Secp256k1)
|
|
require.NoError(t, err)
|
|
|
|
// Test overwrite with overwrite=false
|
|
_, _, err = server.GenerateSaveCoinKey(kb, keyname, false, hd.Secp256k1)
|
|
require.Error(t, err)
|
|
|
|
// Test overwrite with overwrite=true
|
|
addr2, _, err := server.GenerateSaveCoinKey(kb, keyname, true, hd.Secp256k1)
|
|
require.NoError(t, err)
|
|
|
|
require.NotEqual(t, addr1, addr2)
|
|
}
|