From de14bc520e364b7bed443180e9635d10d98f3264 Mon Sep 17 00:00:00 2001
From: Jayden Lee <41176085+tkxkd0159@users.noreply.github.com>
Date: Thu, 30 May 2024 02:03:22 +0900
Subject: [PATCH] Add TransferTxCmd test

---
 x/fbridge/client/cli/tx_test.go | 159 ++++++++++++++++++++++++++++++++
 1 file changed, 159 insertions(+)
 create mode 100644 x/fbridge/client/cli/tx_test.go

diff --git a/x/fbridge/client/cli/tx_test.go b/x/fbridge/client/cli/tx_test.go
new file mode 100644
index 0000000000..30bbc3df66
--- /dev/null
+++ b/x/fbridge/client/cli/tx_test.go
@@ -0,0 +1,159 @@
+package cli_test
+
+import (
+	"fmt"
+	"io"
+	"strings"
+	"testing"
+
+	"github.com/gogo/protobuf/proto"
+	"github.com/golang/mock/gomock"
+	"github.com/stretchr/testify/suite"
+
+	abci "github.com/tendermint/tendermint/abci/types"
+	rpcclientmock "github.com/tendermint/tendermint/rpc/client/mock"
+
+	"github.com/Finschia/finschia-sdk/client"
+	"github.com/Finschia/finschia-sdk/client/flags"
+	"github.com/Finschia/finschia-sdk/crypto/hd"
+	"github.com/Finschia/finschia-sdk/crypto/keyring"
+	"github.com/Finschia/finschia-sdk/tests/mocks"
+	clitestutil "github.com/Finschia/finschia-sdk/testutil/cli"
+	sdk "github.com/Finschia/finschia-sdk/types"
+	testutilmod "github.com/Finschia/finschia-sdk/types/module/testutil"
+	"github.com/Finschia/finschia-sdk/x/fbridge/client/cli"
+	fbridgem "github.com/Finschia/finschia-sdk/x/fbridge/module"
+)
+
+type CLITestSuite struct {
+	suite.Suite
+
+	kr           keyring.Keyring
+	encCfg       testutilmod.TestEncodingConfig
+	baseCtx      client.Context
+	clientCtx    client.Context
+	addrs        []sdk.AccAddress
+	defaultFlags []string
+}
+
+func TestCLITestSuite(t *testing.T) {
+	suite.Run(t, new(CLITestSuite))
+}
+
+func (s *CLITestSuite) SetupSuite() {
+	ctrl := gomock.NewController(s.T())
+	defer ctrl.Finish()
+	ar := mocks.NewMockAccountRetriever(ctrl)
+
+	s.encCfg = testutilmod.MakeTestEncodingConfig(fbridgem.AppModule{})
+	s.kr = keyring.NewInMemory()
+
+	s.baseCtx = client.Context{}.
+		WithKeyring(s.kr).
+		WithTxConfig(s.encCfg.TxConfig).
+		WithCodec(s.encCfg.Codec).
+		WithInterfaceRegistry(s.encCfg.InterfaceRegistry).
+		WithLegacyAmino(s.encCfg.Amino).
+		WithClient(clitestutil.MockTendermintRPC{Client: rpcclientmock.Client{}}).
+		WithAccountRetriever(ar).
+		WithOutput(io.Discard).
+		WithChainID("test-chain")
+
+	ctxGen := func() client.Context {
+		bz, _ := s.encCfg.Codec.Marshal(&sdk.TxResponse{})
+		c := clitestutil.NewMockTendermintRPC(abci.ResponseQuery{
+			Value: bz,
+		})
+		return s.baseCtx.WithClient(c)
+	}
+
+	s.clientCtx = ctxGen()
+	s.addrs = make([]sdk.AccAddress, 0)
+	for i := 0; i < 3; i++ {
+		k, _, err := s.clientCtx.Keyring.NewMnemonic(fmt.Sprintf("TestAccount-%d", i), keyring.English, sdk.FullFundraiserPath, keyring.DefaultBIP39Passphrase, hd.Secp256k1)
+		s.Require().NoError(err)
+		pub := k.GetPubKey()
+		newAddr := sdk.AccAddress(pub.Address())
+		s.addrs = append(s.addrs, newAddr)
+
+		ar.EXPECT().EnsureExists(gomock.Any(), newAddr).Return(nil).AnyTimes()
+		ar.EXPECT().GetAccountNumberSequence(gomock.Any(), newAddr).Return(uint64(i), uint64(1), nil).AnyTimes()
+	}
+
+	s.defaultFlags = []string{
+		fmt.Sprintf("--output=json"),
+		fmt.Sprintf("--%s=home", flags.FlagKeyringDir),
+		fmt.Sprintf("--%s=%s", flags.FlagFrom, s.addrs[0]),
+		fmt.Sprintf("--%s=mynote", flags.FlagNote),
+		fmt.Sprintf("--%s=%s", flags.FlagFees, sdk.NewCoins(sdk.NewCoin(sdk.DefaultBondDenom, sdk.NewInt(10))).String()),
+		fmt.Sprintf("--%s=1.2", flags.FlagGasAdjustment),
+		fmt.Sprintf("--%s=false", flags.FlagUseLedger),
+		fmt.Sprintf("--%s=%s", flags.FlagBroadcastMode, flags.BroadcastSync),
+		fmt.Sprintf("--%s=false", flags.FlagDryRun),
+		fmt.Sprintf("--%s=false", flags.FlagGenerateOnly),
+		fmt.Sprintf("--%s=false", flags.FlagOffline),
+		fmt.Sprintf("--%s=true", flags.FlagSkipConfirmation),
+		fmt.Sprintf("--%s=direct", flags.FlagSignMode),
+		fmt.Sprintf("--%s=%d", flags.FlagTimeoutHeight, 0),
+	}
+}
+
+func (s *CLITestSuite) TestNewTransferTxCmd() {
+	cmd := cli.NewTransferTxCmd()
+	s.Require().NotNil(cmd)
+
+	tcs := []struct {
+		name         string
+		args         []string
+		expectErr    bool
+		respType     proto.Message
+		expectedCode uint32
+	}{
+		{
+			name: "invalid from address",
+			args: []string{
+				s.addrs[0].String(),
+				"10stake",
+				fmt.Sprintf("--%s=%s", flags.FlagFrom, "link1..."),
+			},
+			expectErr: true,
+		},
+		{
+			name: "invalid decimal coin",
+			args: []string{
+				s.addrs[1].String(),
+				fmt.Sprintf("10%s", strings.Repeat("a", 300)),
+				fmt.Sprintf("--%s=%s", flags.FlagFrom, s.addrs[0]),
+			},
+			expectErr: true,
+		},
+		{
+			name: "valid request",
+			args: append([]string{
+				s.addrs[1].String(),
+				"10stake",
+			},
+				s.defaultFlags...,
+			),
+			expectErr:    false,
+			respType:     &sdk.TxResponse{},
+			expectedCode: 0,
+		},
+	}
+
+	for _, tc := range tcs {
+		tc := tc
+		s.Run(tc.name, func() {
+			out, err := clitestutil.ExecTestCLICmd(s.clientCtx, cmd, tc.args)
+			if tc.expectErr {
+				s.Require().Error(err)
+			} else {
+				s.Require().NoError(err, out.String())
+				s.Require().NoError(s.clientCtx.Codec.UnmarshalJSON(out.Bytes(), tc.respType), out.String())
+				tsResp := tc.respType.(*sdk.TxResponse)
+				s.Require().Equal(tc.expectedCode, tsResp.Code, out.String())
+			}
+		})
+
+	}
+}