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")
}