109 lines
2.8 KiB
Go
109 lines
2.8 KiB
Go
package cmd_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/cosmos/cosmos-sdk/crypto/hd"
|
|
"github.com/cosmos/cosmos-sdk/crypto/keyring"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"testing"
|
|
|
|
"github.com/spf13/viper"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/tendermint/tendermint/libs/log"
|
|
|
|
"github.com/cosmos/cosmos-sdk/client"
|
|
"github.com/cosmos/cosmos-sdk/client/flags"
|
|
"github.com/cosmos/cosmos-sdk/server"
|
|
"github.com/cosmos/cosmos-sdk/simapp"
|
|
simcmd "github.com/cosmos/cosmos-sdk/simapp/simd/cmd"
|
|
"github.com/cosmos/cosmos-sdk/testutil/testdata"
|
|
"github.com/cosmos/cosmos-sdk/types/module"
|
|
"github.com/cosmos/cosmos-sdk/x/genutil"
|
|
genutiltest "github.com/cosmos/cosmos-sdk/x/genutil/client/testutil"
|
|
)
|
|
|
|
var testMbm = module.NewBasicManager(genutil.AppModuleBasic{})
|
|
|
|
func TestAddGenesisAccountCmd(t *testing.T) {
|
|
_, _, addr1 := testdata.KeyTestPubAddr()
|
|
tests := []struct {
|
|
name string
|
|
addr string
|
|
denom string
|
|
withKeyring bool
|
|
expectErr bool
|
|
}{
|
|
{
|
|
name: "invalid address",
|
|
addr: "",
|
|
denom: "1000atom",
|
|
withKeyring: false,
|
|
expectErr: true,
|
|
},
|
|
{
|
|
name: "valid address",
|
|
addr: addr1.String(),
|
|
denom: "1000atom",
|
|
withKeyring: false,
|
|
expectErr: false,
|
|
},
|
|
{
|
|
name: "multiple denoms",
|
|
addr: addr1.String(),
|
|
denom: "1000atom, 2000stake",
|
|
withKeyring: false,
|
|
expectErr: false,
|
|
},
|
|
{
|
|
name: "with keyring",
|
|
addr: "ser",
|
|
denom: "1000atom",
|
|
withKeyring: true,
|
|
expectErr: false,
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
tc := tc
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
home := t.TempDir()
|
|
logger := log.NewNopLogger()
|
|
cfg, err := genutiltest.CreateDefaultTendermintConfig(home)
|
|
require.NoError(t, err)
|
|
|
|
appCodec := simapp.MakeTestEncodingConfig().Codec
|
|
err = genutiltest.ExecInitCmd(testMbm, home, appCodec)
|
|
require.NoError(t, err)
|
|
|
|
serverCtx := server.NewContext(viper.New(), cfg, logger)
|
|
clientCtx := client.Context{}.WithCodec(appCodec).WithHomeDir(home)
|
|
|
|
if tc.withKeyring {
|
|
path := hd.CreateHDPath(118, 0, 0).String()
|
|
kr, err := keyring.New(sdk.KeyringServiceName(), keyring.BackendMemory, home, nil)
|
|
require.NoError(t, err)
|
|
_, _, err = kr.NewMnemonic(tc.addr, keyring.English, path, keyring.DefaultBIP39Passphrase, hd.Secp256k1)
|
|
require.NoError(t, err)
|
|
clientCtx = clientCtx.WithKeyring(kr)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
ctx = context.WithValue(ctx, client.ClientContextKey, &clientCtx)
|
|
ctx = context.WithValue(ctx, server.ServerContextKey, serverCtx)
|
|
|
|
cmd := simcmd.AddGenesisAccountCmd(home)
|
|
cmd.SetArgs([]string{
|
|
tc.addr,
|
|
tc.denom,
|
|
fmt.Sprintf("--%s=home", flags.FlagHome)})
|
|
|
|
if tc.expectErr {
|
|
require.Error(t, cmd.ExecuteContext(ctx))
|
|
} else {
|
|
require.NoError(t, cmd.ExecuteContext(ctx))
|
|
}
|
|
})
|
|
}
|
|
}
|