package keys

import (
	"fmt"
	"io/ioutil"
	"strings"
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/cosmos/cosmos-sdk/testutil"
)

func Test_RunMnemonicCmdNormal(t *testing.T) {
	cmd := MnemonicKeyCommand()
	cmd.SetErr(ioutil.Discard)
	cmd.SetOut(ioutil.Discard)
	cmd.SetArgs([]string{})
	require.NoError(t, cmd.Execute())
}

func Test_RunMnemonicCmdUser(t *testing.T) {
	cmd := MnemonicKeyCommand()
	cmd.SetErr(ioutil.Discard)
	cmd.SetOut(ioutil.Discard)

	cmd.SetArgs([]string{fmt.Sprintf("--%s=1", flagUserEntropy)})
	err := cmd.Execute()
	require.Error(t, err)
	require.Equal(t, "EOF", err.Error())

	// Try again
	mockIn, _, _ := testutil.ApplyMockIO(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())

	// Now provide "good" entropy :)
	fakeEntropy := strings.Repeat(":)", 40) + "\ny\n" // entropy + accept count
	mockIn.Reset(fakeEntropy)
	require.NoError(t, cmd.Execute())

	// 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())
}