89 lines
3.1 KiB
Go
89 lines
3.1 KiB
Go
package keys
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/cosmos/cosmos-sdk/testutil"
|
|
)
|
|
|
|
func Test_RunMnemonicCmdNormal(t *testing.T) {
|
|
cmd := MnemonicKeyCommand()
|
|
_ = testutil.ApplyMockIODiscardOutErr(cmd)
|
|
cmd.SetArgs([]string{})
|
|
require.NoError(t, cmd.Execute())
|
|
}
|
|
|
|
func Test_RunMnemonicCmdUser(t *testing.T) {
|
|
cmd := MnemonicKeyCommand()
|
|
_ = testutil.ApplyMockIODiscardOutErr(cmd)
|
|
|
|
cmd.SetArgs([]string{fmt.Sprintf("--%s", flagUserEntropy), fmt.Sprintf("--%s", flagIndiscreet)})
|
|
err := cmd.Execute()
|
|
require.Error(t, err)
|
|
require.Equal(t, "EOF", err.Error())
|
|
|
|
// Try again
|
|
mockIn := testutil.ApplyMockIODiscardOutErr(cmd)
|
|
mockIn.Reset("Hi!\n")
|
|
err = cmd.Execute()
|
|
require.Error(t, err)
|
|
require.Equal(t,
|
|
"256-bits is 43 characters in Base-64, and 100 in Base-6. You entered 3, and probably want more",
|
|
err.Error())
|
|
|
|
mockIn, mockOut := testutil.ApplyMockIO(cmd)
|
|
// Now provide "good" entropy :)
|
|
fakeEntropy := strings.Repeat(":)", 40) + "\ny\n" // entropy + accept count
|
|
mockIn.Reset(fakeEntropy)
|
|
require.NoError(t, cmd.Execute())
|
|
require.Equal(t, "volcano hungry midnight divorce post ship bicycle fitness hospital critic protect ring trim alien there safe fine subway style impulse identify right improve print\n", mockOut.String())
|
|
|
|
mockIn = testutil.ApplyMockIODiscardOutErr(cmd)
|
|
// Now provide "good" entropy but no answer
|
|
fakeEntropy = strings.Repeat(":)", 40) + "\n" // entropy + accept count
|
|
mockIn.Reset(fakeEntropy)
|
|
require.Error(t, cmd.Execute())
|
|
|
|
// Now provide "good" entropy but say no
|
|
fakeEntropy = strings.Repeat(":)", 40) + "\nn\n" // entropy + accept count
|
|
mockIn.Reset(fakeEntropy)
|
|
require.NoError(t, cmd.Execute())
|
|
|
|
// test for skip confirmation
|
|
cmd.SetArgs([]string{fmt.Sprintf("--%s", flagUserEntropy), fmt.Sprintf("--%s", flagIndiscreet), fmt.Sprintf("--%s", flagYes)})
|
|
fakeEntropy = strings.Repeat(":)", 40) + "\n" // entropy + accept count
|
|
mockIn.Reset(fakeEntropy)
|
|
require.NoError(t, cmd.Execute())
|
|
require.Equal(t, "volcano hungry midnight divorce post ship bicycle fitness hospital critic protect ring trim alien there safe fine subway style impulse identify right improve print\n", mockOut.String())
|
|
}
|
|
|
|
func Test_RunMnemonicCmdUserDiscreetly(t *testing.T) {
|
|
cmd := MnemonicKeyCommand()
|
|
_ = testutil.ApplyMockIODiscardOutErr(cmd)
|
|
|
|
cmd.SetArgs([]string{fmt.Sprintf("--%s", flagUserEntropy)})
|
|
err := cmd.Execute()
|
|
require.Error(t, err)
|
|
require.Equal(t, "EOF", err.Error())
|
|
|
|
// Try again
|
|
mockIn := testutil.ApplyMockIODiscardOutErr(cmd)
|
|
mockIn.Reset("Hi!\n")
|
|
err = cmd.Execute()
|
|
require.Error(t, err)
|
|
require.Equal(t,
|
|
"256-bits is 43 characters in Base-64, and 100 in Base-6. You entered 3, and probably want more",
|
|
err.Error())
|
|
|
|
mockIn, mockOut := testutil.ApplyMockIO(cmd)
|
|
// Now provide "good" entropy :)
|
|
fakeEntropy := strings.Repeat(":)", 40) + "\ny\n" // entropy + accept count
|
|
mockIn.Reset(fakeEntropy)
|
|
require.NoError(t, cmd.Execute())
|
|
require.Contains(t, mockOut.String(), "volcano hungry midnight divorce post ship bicycle fitness hospital critic protect ring trim alien there safe fine subway style impulse identify right improve print")
|
|
}
|