104 lines
2.7 KiB
Go
104 lines
2.7 KiB
Go
package keys
|
|
|
|
import (
|
|
"bufio"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/spf13/viper"
|
|
"github.com/tendermint/tendermint/libs/cli"
|
|
|
|
"github.com/cosmos/cosmos-sdk/tests"
|
|
|
|
"github.com/cosmos/cosmos-sdk/client"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test_runAddCmdBasic(t *testing.T) {
|
|
cmd := addKeyCommand()
|
|
assert.NotNil(t, cmd)
|
|
|
|
// Missing input (enter password)
|
|
err := runAddCmd(cmd, []string{"keyname"})
|
|
assert.EqualError(t, err, "EOF")
|
|
|
|
// Prepare a keybase
|
|
kbHome, kbCleanUp := tests.NewTestCaseDir(t)
|
|
assert.NotNil(t, kbHome)
|
|
defer kbCleanUp()
|
|
viper.Set(cli.HomeFlag, kbHome)
|
|
|
|
/// Test Text
|
|
viper.Set(cli.OutputFlag, OutputFormatText)
|
|
// Now enter password
|
|
cleanUp1 := client.OverrideStdin(bufio.NewReader(strings.NewReader("test1234\ntest1234\n")))
|
|
defer cleanUp1()
|
|
err = runAddCmd(cmd, []string{"keyname1"})
|
|
assert.NoError(t, err)
|
|
|
|
/// Test Text - Replace? >> FAIL
|
|
viper.Set(cli.OutputFlag, OutputFormatText)
|
|
// Now enter password
|
|
cleanUp2 := client.OverrideStdin(bufio.NewReader(strings.NewReader("test1234\ntest1234\n")))
|
|
defer cleanUp2()
|
|
err = runAddCmd(cmd, []string{"keyname1"})
|
|
assert.Error(t, err)
|
|
|
|
/// Test Text - Replace? Answer >> PASS
|
|
viper.Set(cli.OutputFlag, OutputFormatText)
|
|
// Now enter password
|
|
cleanUp3 := client.OverrideStdin(bufio.NewReader(strings.NewReader("y\ntest1234\ntest1234\n")))
|
|
defer cleanUp3()
|
|
err = runAddCmd(cmd, []string{"keyname1"})
|
|
assert.NoError(t, err)
|
|
|
|
// Check JSON
|
|
viper.Set(cli.OutputFlag, OutputFormatJSON)
|
|
// Now enter password
|
|
cleanUp4 := client.OverrideStdin(bufio.NewReader(strings.NewReader("test1234\ntest1234\n")))
|
|
defer cleanUp4()
|
|
err = runAddCmd(cmd, []string{"keyname2"})
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
type MockResponseWriter struct {
|
|
dataHeaderStatus int
|
|
dataBody []byte
|
|
}
|
|
|
|
func (MockResponseWriter) Header() http.Header {
|
|
panic("Unexpected call!")
|
|
}
|
|
|
|
func (w *MockResponseWriter) Write(data []byte) (int, error) {
|
|
w.dataBody = append(w.dataBody, data...)
|
|
return len(data), nil
|
|
}
|
|
|
|
func (w *MockResponseWriter) WriteHeader(statusCode int) {
|
|
w.dataHeaderStatus = statusCode
|
|
}
|
|
|
|
func TestCheckAndWriteErrorResponse(t *testing.T) {
|
|
mockRW := MockResponseWriter{}
|
|
|
|
mockRW.WriteHeader(100)
|
|
assert.Equal(t, 100, mockRW.dataHeaderStatus)
|
|
|
|
detected := CheckAndWriteErrorResponse(&mockRW, http.StatusBadRequest, errors.New("some ERROR"))
|
|
require.True(t, detected)
|
|
require.Equal(t, http.StatusBadRequest, mockRW.dataHeaderStatus)
|
|
require.Equal(t, "some ERROR", string(mockRW.dataBody[:]))
|
|
|
|
mockRW = MockResponseWriter{}
|
|
detected = CheckAndWriteErrorResponse(&mockRW, http.StatusBadRequest, nil)
|
|
require.False(t, detected)
|
|
require.Equal(t, 0, mockRW.dataHeaderStatus)
|
|
require.Equal(t, "", string(mockRW.dataBody[:]))
|
|
}
|