65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
|
package keys
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bufio"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/cosmos/cosmos-sdk/client"
|
||
|
|
|
||
|
|
"github.com/cosmos/cosmos-sdk/tests"
|
||
|
|
"github.com/spf13/viper"
|
||
|
|
"github.com/tendermint/tendermint/libs/cli"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Test_updateKeyCommand(t *testing.T) {
|
||
|
|
cmd := updateKeyCommand()
|
||
|
|
assert.NotNil(t, cmd)
|
||
|
|
// No flags or defaults to validate
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_runUpdateCmd(t *testing.T) {
|
||
|
|
fakeKeyName1 := "runUpdateCmd_Key1"
|
||
|
|
fakeKeyName2 := "runUpdateCmd_Key2"
|
||
|
|
|
||
|
|
cmd := updateKeyCommand()
|
||
|
|
|
||
|
|
// fails because it requests a password
|
||
|
|
err := runUpdateCmd(cmd, []string{fakeKeyName1})
|
||
|
|
assert.EqualError(t, err, "EOF")
|
||
|
|
|
||
|
|
cleanUp := client.OverrideStdin(bufio.NewReader(strings.NewReader("pass1234\n")))
|
||
|
|
defer cleanUp()
|
||
|
|
|
||
|
|
// try again
|
||
|
|
err = runUpdateCmd(cmd, []string{fakeKeyName1})
|
||
|
|
assert.EqualError(t, err, "Key runUpdateCmd_Key1 not found")
|
||
|
|
|
||
|
|
// Prepare a key base
|
||
|
|
// Now add a temporary keybase
|
||
|
|
kbHome, cleanUp1, err := tests.GetTempDir("Test_runShowCmd")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
defer cleanUp1()
|
||
|
|
viper.Set(cli.HomeFlag, kbHome)
|
||
|
|
|
||
|
|
kb, err := NewKeyBaseFromHomeFlag()
|
||
|
|
assert.NoError(t, err)
|
||
|
|
_, err = kb.CreateAccount(fakeKeyName1, tests.TestMnemonic, "", "", 0, 0)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
_, err = kb.CreateAccount(fakeKeyName2, tests.TestMnemonic, "", "", 0, 1)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
// Try again now that we have keys
|
||
|
|
cleanUp2 := client.OverrideStdin(bufio.NewReader(strings.NewReader("pass1234\nNew1234\nNew1234")))
|
||
|
|
defer cleanUp2()
|
||
|
|
|
||
|
|
// Incorrect key type
|
||
|
|
err = runUpdateCmd(cmd, []string{fakeKeyName1})
|
||
|
|
assert.EqualError(t, err, "locally stored key required. Received: keys.offlineInfo")
|
||
|
|
|
||
|
|
// TODO: Check for other type types?
|
||
|
|
|
||
|
|
}
|