50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
package server_test
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/cosmos/cosmos-sdk/client/keys"
|
|
"github.com/cosmos/cosmos-sdk/server"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGenerateCoinKey(t *testing.T) {
|
|
addr, mnemonic, err := server.GenerateCoinKey()
|
|
require.NoError(t, err)
|
|
|
|
// Test creation
|
|
info, err := keys.NewInMemoryKeyBase().CreateAccount("xxx", mnemonic, "", "012345678", 0, 0)
|
|
require.NoError(t, err)
|
|
require.Equal(t, addr, info.GetAddress())
|
|
}
|
|
|
|
func TestGenerateSaveCoinKey(t *testing.T) {
|
|
dir, cleanup := tempdir(t)
|
|
defer cleanup() // clean after itself
|
|
// Remove the dir to that GenerateSaveCoinKey creates it automatically
|
|
os.RemoveAll(dir)
|
|
|
|
addr, mnemonic, err := server.GenerateSaveCoinKey(dir, "keyname", "012345678", false)
|
|
require.NoError(t, err)
|
|
|
|
// Test key was actually saved
|
|
kb, err := keys.NewKeyBaseFromDir(dir)
|
|
require.NoError(t, err)
|
|
info, err := kb.Get("keyname")
|
|
require.NoError(t, err)
|
|
require.Equal(t, addr, info.GetAddress())
|
|
|
|
// Test in-memory recovery
|
|
info, err = keys.NewInMemoryKeyBase().CreateAccount("xxx", mnemonic, "", "012345678", 0, 0)
|
|
require.NoError(t, err)
|
|
require.Equal(t, addr, info.GetAddress())
|
|
}
|
|
|
|
func tempdir(t *testing.T) (string, func()) {
|
|
dir, err := ioutil.TempDir("", t.Name()+"_")
|
|
require.NoError(t, err)
|
|
return dir, func() { os.RemoveAll(dir) }
|
|
}
|