Prepare migrating testing auxiliary functions from tests to testutil. Remove local duplicates on testutil.WriteToNewTempFile(). Always favor testutil.NewTestCaseDir() over ioutil.TempDir(). Add test cases for the testing auxiliary functions.
70 lines
2.0 KiB
Go
70 lines
2.0 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/testutil"
|
|
"github.com/cosmos/cosmos-sdk/types"
|
|
)
|
|
|
|
func TestGenerateCoinKey(t *testing.T) {
|
|
t.Parallel()
|
|
addr, mnemonic, err := server.GenerateCoinKey()
|
|
require.NoError(t, err)
|
|
|
|
// Test creation
|
|
info, err := keyring.NewInMemory().NewAccount("xxx", mnemonic, "", hd.NewFundraiserParams(0, types.GetConfig().GetCoinType(), 0).String(), hd.Secp256k1)
|
|
require.NoError(t, err)
|
|
require.Equal(t, addr, info.GetAddress())
|
|
}
|
|
|
|
func TestGenerateSaveCoinKey(t *testing.T) {
|
|
t.Parallel()
|
|
dir, cleanup := testutil.NewTestCaseDir(t)
|
|
t.Cleanup(cleanup)
|
|
|
|
kb, err := keyring.New(t.Name(), "test", dir, nil)
|
|
require.NoError(t, err)
|
|
|
|
addr, mnemonic, err := server.GenerateSaveCoinKey(kb, "keyname", "012345678", false)
|
|
require.NoError(t, err)
|
|
|
|
// Test key was actually saved
|
|
info, err := kb.Key("keyname")
|
|
require.NoError(t, err)
|
|
require.Equal(t, addr, info.GetAddress())
|
|
|
|
// Test in-memory recovery
|
|
info, err = keyring.NewInMemory().NewAccount("xxx", mnemonic, "", hd.NewFundraiserParams(0, types.GetConfig().GetCoinType(), 0).String(), hd.Secp256k1)
|
|
require.NoError(t, err)
|
|
require.Equal(t, addr, info.GetAddress())
|
|
}
|
|
|
|
func TestGenerateSaveCoinKeyOverwriteFlag(t *testing.T) {
|
|
t.Parallel()
|
|
dir, cleanup := testutil.NewTestCaseDir(t)
|
|
t.Cleanup(cleanup)
|
|
|
|
kb, err := keyring.New(t.Name(), "test", dir, nil)
|
|
require.NoError(t, err)
|
|
|
|
keyname := "justakey"
|
|
addr1, _, err := server.GenerateSaveCoinKey(kb, keyname, "012345678", false)
|
|
require.NoError(t, err)
|
|
|
|
// Test overwrite with overwrite=false
|
|
_, _, err = server.GenerateSaveCoinKey(kb, keyname, "012345678", false)
|
|
require.Error(t, err)
|
|
|
|
// Test overwrite with overwrite=true
|
|
addr2, _, err := server.GenerateSaveCoinKey(kb, keyname, "012345678", true)
|
|
require.NoError(t, err)
|
|
|
|
require.NotEqual(t, addr1, addr2)
|
|
}
|