cosmos-sdk/tests/integration/genutil/init_test.go

package genutil

import (
	"context"
	"fmt"
	"net"
	"testing"
	"time"

	abci_server "github.com/cometbft/cometbft/abci/server"
	"github.com/spf13/viper"
	"github.com/stretchr/testify/require"

	corectx "cosmossdk.io/core/context"
	"cosmossdk.io/depinject"
	"cosmossdk.io/log"
	_ "cosmossdk.io/x/accounts"
	_ "cosmossdk.io/x/bank"
	_ "cosmossdk.io/x/consensus"
	"cosmossdk.io/x/staking"

	"github.com/cosmos/cosmos-sdk/client"
	"github.com/cosmos/cosmos-sdk/client/flags"
	"github.com/cosmos/cosmos-sdk/codec"
	"github.com/cosmos/cosmos-sdk/codec/types"
	cryptocodec "github.com/cosmos/cosmos-sdk/crypto/codec"
	"github.com/cosmos/cosmos-sdk/crypto/keys/ed25519"
	"github.com/cosmos/cosmos-sdk/testutil"
	"github.com/cosmos/cosmos-sdk/testutil/configurator"
	simtestutil "github.com/cosmos/cosmos-sdk/testutil/sims"
	genutilhelpers "github.com/cosmos/cosmos-sdk/testutil/x/genutil"
	sdk "github.com/cosmos/cosmos-sdk/types"
	"github.com/cosmos/cosmos-sdk/types/module"
	"github.com/cosmos/cosmos-sdk/x/genutil"
	genutilcli "github.com/cosmos/cosmos-sdk/x/genutil/client/cli"
	genutiltypes "github.com/cosmos/cosmos-sdk/x/genutil/types"
	_ "github.com/cosmos/cosmos-sdk/x/validate"
)

var testMbm = module.NewManager(
	staking.NewAppModule(makeCodec(), nil),
	genutil.NewAppModule(makeCodec(), nil, nil, nil, nil, nil),
)

func TestInitCmd(t *testing.T) {
	tests := []struct {
		name      string
		flags     func(dir string) []string
		shouldErr bool
		err       error
	}{
		{
			name: "happy path",
			flags: func(dir string) []string {
				return []string{
					"appnode-test",
				}
			},
			shouldErr: false,
			err:       nil,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			home := t.TempDir()
			logger := log.NewNopLogger()
			viper := viper.New()

			err := writeAndTrackDefaultConfig(viper, home)
			require.NoError(t, err)
			interfaceRegistry := types.NewInterfaceRegistry()
			marshaler := codec.NewProtoCodec(interfaceRegistry)
			clientCtx := client.Context{}.
				WithCodec(marshaler).
				WithLegacyAmino(makeAminoCodec()).
				WithHomeDir(home)

			ctx := context.Background()
			ctx = context.WithValue(ctx, client.ClientContextKey, &clientCtx)
			ctx = context.WithValue(ctx, corectx.ViperContextKey, viper)
			ctx = context.WithValue(ctx, corectx.LoggerContextKey, logger)

			cmd := genutilcli.InitCmd(testMbm)
			cmd.SetArgs(
				tt.flags(home),
			)

			if tt.shouldErr {
				err := cmd.ExecuteContext(ctx)
				require.EqualError(t, err, tt.err.Error())
			} else {
				require.NoError(t, cmd.ExecuteContext(ctx))
			}
		})
	}
}

func TestInitRecover(t *testing.T) {
	home := t.TempDir()
	logger := log.NewNopLogger()
	viper := viper.New()

	err := writeAndTrackDefaultConfig(viper, home)
	require.NoError(t, err)
	interfaceRegistry := types.NewInterfaceRegistry()
	marshaler := codec.NewProtoCodec(interfaceRegistry)
	clientCtx := client.Context{}.
		WithCodec(marshaler).
		WithLegacyAmino(makeAminoCodec()).
		WithHomeDir(home)

	ctx := context.Background()
	ctx = context.WithValue(ctx, client.ClientContextKey, &clientCtx)
	ctx = context.WithValue(ctx, corectx.ViperContextKey, viper)
	ctx = context.WithValue(ctx, corectx.LoggerContextKey, logger)

	cmd := genutilcli.InitCmd(testMbm)
	cmd.SetContext(ctx)
	mockIn := testutil.ApplyMockIODiscardOutErr(cmd)

	cmd.SetArgs([]string{
		"appnode-test",
		fmt.Sprintf("--%s=true", genutilcli.FlagRecover),
	})

	// use valid mnemonic and complete recovery key generation successfully
	mockIn.Reset(
		"decide praise business actor peasant farm drastic weather extend front hurt later song give verb rhythm worry fun pond reform school tumble august one\n",
	)
	require.NoError(t, cmd.ExecuteContext(ctx))
}

func TestInitDefaultBondDenom(t *testing.T) {
	home := t.TempDir()
	logger := log.NewNopLogger()
	viper := viper.New()

	err := writeAndTrackDefaultConfig(viper, home)
	require.NoError(t, err)
	interfaceRegistry := types.NewInterfaceRegistry()
	marshaler := codec.NewProtoCodec(interfaceRegistry)
	clientCtx := client.Context{}.
		WithCodec(marshaler).
		WithLegacyAmino(makeAminoCodec()).
		WithHomeDir(home)

	ctx := context.Background()
	ctx = context.WithValue(ctx, client.ClientContextKey, &clientCtx)
	ctx = context.WithValue(ctx, corectx.ViperContextKey, viper)
	ctx = context.WithValue(ctx, corectx.LoggerContextKey, logger)

	cmd := genutilcli.InitCmd(testMbm)

	cmd.SetArgs([]string{
		"appnode-test",
		fmt.Sprintf("--%s=testtoken", genutilcli.FlagDefaultBondDenom),
	})
	require.NoError(t, cmd.ExecuteContext(ctx))
}

func TestEmptyState(t *testing.T) {
	// TODO: rewrite for v2: https://github.com/cosmos/cosmos-sdk/issues/20799

	// home := t.TempDir()
	// logger := log.NewNopLogger()
	// viper := viper.New()

	// err := writeAndTrackDefaultConfig(viper, home)
	// require.NoError(t, err)
	// interfaceRegistry := types.NewInterfaceRegistry()
	// marshaler := codec.NewProtoCodec(interfaceRegistry)
	// clientCtx := client.Context{}.
	// 	WithCodec(marshaler).
	// 	WithLegacyAmino(makeAminoCodec()).
	// 	WithHomeDir(home)

	// ctx := context.Background()
	// ctx = context.WithValue(ctx, client.ClientContextKey, &clientCtx)
	// ctx = context.WithValue(ctx, corectx.ViperContextKey, viper)
	// ctx = context.WithValue(ctx, corectx.LoggerContextKey, logger)

	// cmd := genutilcli.InitCmd(testMbm)
	// cmd.SetArgs([]string{"appnode-test"})

	// require.NoError(t, cmd.ExecuteContext(ctx))

	// old := os.Stdout
	// r, w, _ := os.Pipe()
	// os.Stdout = w

	// cmd = genutilcli.ExportCmd(nil)
	// require.NoError(t, cmd.ExecuteContext(ctx))

	// outC := make(chan string)
	// go func() {
	// 	var buf bytes.Buffer
	// 	_, err := io.Copy(&buf, r)
	// 	require.NoError(t, err)

	// 	outC <- buf.String()
	// }()

	// w.Close()
	// os.Stdout = old
	// out := <-outC

	// require.Contains(t, out, "genesis_time")
	// require.Contains(t, out, "chain_id")
	// require.Contains(t, out, "consensus")
	// require.Contains(t, out, "app_hash")
	// require.Contains(t, out, "app_state")
}

func TestStartStandAlone(t *testing.T) {
	home := t.TempDir()
	interfaceRegistry := types.NewInterfaceRegistry()
	marshaler := codec.NewProtoCodec(interfaceRegistry)
	err := genutilhelpers.ExecInitCmd(testMbm, home, marshaler)
	require.NoError(t, err)

	app, err := simtestutil.SetupWithConfiguration(
		depinject.Configs(
			configurator.NewAppConfig(
				configurator.AccountsModule(),
				configurator.AuthModule(),
				configurator.StakingModule(),
				configurator.TxModule(),
				configurator.ValidateModule(),
				configurator.ConsensusModule(),
				configurator.BankModule(),
			),
			depinject.Supply(log.NewNopLogger()),
		), simtestutil.DefaultStartUpConfig())
	require.NoError(t, err)

	svrAddr, _, closeFn, err := freeTCPAddr()
	require.NoError(t, err)
	require.NoError(t, closeFn())

	cmtApp := cometABCIWrapper{app}
	svr, err := abci_server.NewServer(svrAddr, "socket", cmtApp)
	require.NoError(t, err, "error creating listener")

	err = svr.Start()
	require.NoError(t, err)

	timer := time.NewTimer(time.Duration(2) * time.Second)
	for range timer.C {
		err = svr.Stop()
		require.NoError(t, err)
		break
	}
}

func TestInitNodeValidatorFiles(t *testing.T) {
	home := t.TempDir()
	cfg, err := genutilhelpers.CreateDefaultCometConfig(home)
	require.NoError(t, err)

	nodeID, valPubKey, err := genutil.InitializeNodeValidatorFiles(cfg, ed25519.KeyType)
	require.NoError(t, err)

	require.NotEqual(t, "", nodeID)
	require.NotEqual(t, 0, len(valPubKey.Bytes()))
}

func TestInitConfig(t *testing.T) {
	// TODO: rewrite for v2: https://github.com/cosmos/cosmos-sdk/issues/20799

	// home := t.TempDir()
	// logger := log.NewNopLogger()
	// viper := viper.New()

	// err := writeAndTrackDefaultConfig(viper, home)
	// require.NoError(t, err)
	// interfaceRegistry := types.NewInterfaceRegistry()
	// marshaler := codec.NewProtoCodec(interfaceRegistry)
	// clientCtx := client.Context{}.
	// 	WithCodec(marshaler).
	// 	WithLegacyAmino(makeAminoCodec()).
	// 	WithChainID("foo"). // add chain-id to clientCtx
	// 	WithHomeDir(home)

	// ctx := context.Background()
	// ctx = context.WithValue(ctx, client.ClientContextKey, &clientCtx)
	// ctx = context.WithValue(ctx, corectx.ViperContextKey, viper)
	// ctx = context.WithValue(ctx, corectx.LoggerContextKey, logger)

	// cmd := genutilcli.InitCmd(testMbm)
	// cmd.SetArgs([]string{"testnode"})

	// err = cmd.ExecuteContext(ctx)
	// require.NoError(t, err)

	// old := os.Stdout
	// r, w, _ := os.Pipe()
	// os.Stdout = w

	// cmd = genutilcli.ExportCmd(nil)
	// require.NoError(t, cmd.ExecuteContext(ctx))

	// outC := make(chan string)
	// go func() {
	// 	var buf bytes.Buffer
	// 	_, err := io.Copy(&buf, r)
	// 	require.NoError(t, err)
	// 	outC <- buf.String()
	// }()

	// w.Close()
	// os.Stdout = old
	// out := <-outC

	// require.Contains(t, out, "\"chain_id\": \"foo\"")
}

func TestInitWithHeight(t *testing.T) {
	home := t.TempDir()
	logger := log.NewNopLogger()
	viper := viper.New()
	cfg, err := genutilhelpers.CreateDefaultCometConfig(home)
	require.NoError(t, err)

	err = writeAndTrackDefaultConfig(viper, home)
	require.NoError(t, err)
	interfaceRegistry := types.NewInterfaceRegistry()
	marshaler := codec.NewProtoCodec(interfaceRegistry)
	clientCtx := client.Context{}.
		WithCodec(marshaler).
		WithLegacyAmino(makeAminoCodec()).
		WithChainID("foo"). // add chain-id to clientCtx
		WithHomeDir(home)

	ctx := context.Background()
	ctx = context.WithValue(ctx, client.ClientContextKey, &clientCtx)
	ctx = context.WithValue(ctx, corectx.ViperContextKey, viper)
	ctx = context.WithValue(ctx, corectx.LoggerContextKey, logger)

	testInitialHeight := int64(333)

	cmd := genutilcli.InitCmd(testMbm)

	fmt.Println("RootDir", viper.Get(flags.FlagHome))
	cmd.SetArgs([]string{"init-height-test", fmt.Sprintf("--%s=%d", flags.FlagInitHeight, testInitialHeight)})

	require.NoError(t, cmd.ExecuteContext(ctx))

	appGenesis, importErr := genutiltypes.AppGenesisFromFile(cfg.GenesisFile())
	require.NoError(t, importErr)

	require.Equal(t, testInitialHeight, appGenesis.InitialHeight)
}

func TestInitWithNegativeHeight(t *testing.T) {
	home := t.TempDir()
	logger := log.NewNopLogger()
	viper := viper.New()
	cfg, err := genutilhelpers.CreateDefaultCometConfig(home)
	require.NoError(t, err)

	err = writeAndTrackDefaultConfig(viper, home)
	require.NoError(t, err)
	interfaceRegistry := types.NewInterfaceRegistry()
	marshaler := codec.NewProtoCodec(interfaceRegistry)
	clientCtx := client.Context{}.
		WithCodec(marshaler).
		WithLegacyAmino(makeAminoCodec()).
		WithChainID("foo"). // add chain-id to clientCtx
		WithHomeDir(home)

	ctx := context.Background()
	ctx = context.WithValue(ctx, client.ClientContextKey, &clientCtx)
	ctx = context.WithValue(ctx, corectx.ViperContextKey, viper)
	ctx = context.WithValue(ctx, corectx.LoggerContextKey, logger)

	testInitialHeight := int64(-333)

	cmd := genutilcli.InitCmd(testMbm)
	cmd.SetArgs([]string{"init-height-test", fmt.Sprintf("--%s=%d", flags.FlagInitHeight, testInitialHeight)})

	require.NoError(t, cmd.ExecuteContext(ctx))

	appGenesis, importErr := genutiltypes.AppGenesisFromFile(cfg.GenesisFile())
	require.NoError(t, importErr)

	require.Equal(t, int64(1), appGenesis.InitialHeight)
}

// custom tx codec
func makeAminoCodec() *codec.LegacyAmino {
	cdc := codec.NewLegacyAmino()
	sdk.RegisterLegacyAminoCodec(cdc)
	cryptocodec.RegisterCrypto(cdc)
	return cdc
}

func makeCodec() codec.Codec {
	interfaceRegistry := types.NewInterfaceRegistry()
	return codec.NewProtoCodec(interfaceRegistry)
}

func writeAndTrackDefaultConfig(v *viper.Viper, home string) error {
	cfg, err := genutilhelpers.CreateDefaultCometConfig(home)
	if err != nil {
		return err
	}
	return genutilhelpers.WriteAndTrackCometConfig(v, home, cfg)
}

// Get a free address for a test CometBFT server
// protocol is either tcp, http, etc
func freeTCPAddr() (addr, port string, closeFn func() error, err error) {
	l, err := net.Listen("tcp", "127.0.0.1:0")
	if err != nil {
		return "", "", nil, err
	}

	closeFn = func() error {
		return l.Close()
	}

	portI := l.Addr().(*net.TCPAddr).Port
	port = fmt.Sprintf("%d", portI)
	addr = fmt.Sprintf("tcp://127.0.0.1:%s", port)
	return
}