github.com/fibonacci-chain/fbc@v0.0.0-20231124064014-c7636198c1e9/libs/cosmos-sdk/baseapp/evm2cm_test.go

package baseapp

import (
	"encoding/json"
	"os"
	"testing"

	sdk "github.com/fibonacci-chain/fbc/libs/cosmos-sdk/types"
	"github.com/fibonacci-chain/fbc/libs/tendermint/libs/log"
	db "github.com/fibonacci-chain/fbc/libs/tm-db"
	"github.com/stretchr/testify/require"
)

type testMsg struct {
	route string
}

func (msg testMsg) Route() string                { return msg.route }
func (msg testMsg) Type() string                 { return "testMsg" }
func (msg testMsg) GetSigners() []sdk.AccAddress { return nil }
func (msg testMsg) GetSignBytes() []byte         { return nil }
func (msg testMsg) ValidateBasic() error         { return nil }

func TestRegisterCmHandle_ConvertMsg(t *testing.T) {
	testcases := []struct {
		module      string
		funcName    string
		blockHeight int64
		setHeight   int64
		success     bool
	}{
		{
			module:      "module1",
			funcName:    "test1",
			blockHeight: 0,
			setHeight:   0,
			success:     true,
		},
		{
			module:      "module1",
			funcName:    "test2",
			blockHeight: 0,
			setHeight:   0,
			success:     true,
		},
		{
			module:      "module3",
			funcName:    "test1",
			blockHeight: 0,
			setHeight:   0,
			success:     true,
		},
		{
			module:      "module4",
			funcName:    "test1",
			blockHeight: 0,
			setHeight:   0,
			success:     true,
		},
		{
			module:      "module4",
			funcName:    "test2",
			blockHeight: 0,
			setHeight:   0,
			success:     true,
		},
		{
			module:      "module4",
			funcName:    "test3",
			blockHeight: 0,
			setHeight:   0,
			success:     true,
		},

		// error
		{
			module:      "module5",
			funcName:    "test1",
			blockHeight: 4,
			setHeight:   5,
			success:     false,
		},
		{
			module:      "module5",
			funcName:    "test2",
			blockHeight: 5,
			setHeight:   5,
			success:     true,
		},
		{
			module:      "module5",
			funcName:    "test3",
			blockHeight: 6,
			setHeight:   5,
			success:     true,
		},

		//{ // test for panic
		//	module:      "module5",
		//	funcName:    "test3",
		//	blockHeight: 6,
		//	setHeight:   5,
		//	success:     true,
		//},
	}
	for _, ts := range testcases {
		RegisterCmHandle(ts.module+ts.funcName,
			NewCMHandle(func(data []byte, signers []sdk.AccAddress) (sdk.Msg, error) {
				return nil, nil
			}, ts.setHeight))
	}

	// check
	for _, ts := range testcases {
		RegisterEvmParamParse(func(msg sdk.Msg) ([]byte, error) {
			mw := &MsgWrapper{
				Name: ts.module + ts.funcName,
				Data: []byte("123"),
			}
			v, err := json.Marshal(mw)
			require.NoError(t, err)
			return v, nil
		})
		_, err := ConvertMsg(testMsg{}, ts.blockHeight)
		require.Equal(t, ts.success, err == nil)
	}
}

func TestJudgeEvmConvert(t *testing.T) {
	logger := log.NewTMLogger(log.NewSyncWriter(os.Stdout)).
		With("module", "mock")
	testcases := []struct {
		app     *BaseApp
		fnInit  func(app *BaseApp)
		fnCheck func(is bool)
	}{
		{
			app:    NewBaseApp("test", logger, db.NewMemDB(), nil),
			fnInit: func(app *BaseApp) {},
			fnCheck: func(is bool) {
				require.False(t, is)
			},
		},
		{
			app: NewBaseApp("test", logger, db.NewMemDB(), nil),
			fnInit: func(app *BaseApp) {
				app.SetEvmSysContractAddressHandler(func(ctx sdk.Context, addr sdk.AccAddress) bool {
					return true
				})
				RegisterEvmParamParse(func(msg sdk.Msg) ([]byte, error) {
					return nil, nil
				})
				RegisterEvmResultConverter(func(txHash, data []byte) ([]byte, error) {
					return nil, nil
				})
				RegisterEvmConvertJudge(func(msg sdk.Msg) ([]byte, bool) {
					return []byte{1, 2, 3}, true
				})
			},
			fnCheck: func(is bool) {
				require.True(t, is)
			},
		},
		{
			app:    NewBaseApp("test", logger, db.NewMemDB(), nil),
			fnInit: func(app *BaseApp) {},
			fnCheck: func(is bool) {
				require.False(t, is)
			},
		},
	}

	for _, ts := range testcases {
		ts.fnInit(ts.app)
		re := ts.app.JudgeEvmConvert(sdk.Context{}, nil)
		ts.fnCheck(re)
	}
}

func TestParseMsgWrapper(t *testing.T) {
	testcases := []struct {
		input   string
		fnCheck func(ret *MsgWrapper, err error)
	}{
		{
			input: `{"type": "fbexchain/staking/MsgDeposit","value": {"delegator_address": "0x4375D630687C83471829227b5C1Ea92217FD6265","quantity": {"denom": "fibo","amount": "1"}}}`,
			fnCheck: func(ret *MsgWrapper, err error) {
				require.NoError(t, err)
				require.Equal(t, "fbexchain/staking/MsgDeposit", ret.Name)
				require.Equal(t, "{\"delegator_address\": \"0x4375D630687C83471829227b5C1Ea92217FD6265\",\"quantity\": {\"denom\": \"fibo\",\"amount\": \"1\"}}", string(ret.Data))
			},
		},
		{
			input: `{"type": "fbexchain/staking/MsgWithdraw","value": {"delegator_address": "0x4375D630687C83471829227b5C1Ea92217FD6265","quantity": {"denom": "fibo","amount": "1"}}}`,
			fnCheck: func(ret *MsgWrapper, err error) {
				require.NoError(t, err)
				require.Equal(t, "fbexchain/staking/MsgWithdraw", ret.Name)
				require.Equal(t, "{\"delegator_address\": \"0x4375D630687C83471829227b5C1Ea92217FD6265\",\"quantity\": {\"denom\": \"fibo\",\"amount\": \"1\"}}", string(ret.Data))
			},
		},
		// error
		{
			input: `{"type1": "fbexchain/staking/MsgWithdraw","value":""}`,
			fnCheck: func(ret *MsgWrapper, err error) {
				require.NotNil(t, err)
			},
		},
		{
			input: `{"type": "fbexchain/staking/MsgWithdraw","value1":"123"}`,
			fnCheck: func(ret *MsgWrapper, err error) {
				require.NotNil(t, err)
			},
		},
	}

	for _, ts := range testcases {
		ret, err := ParseMsgWrapper([]byte(ts.input))
		ts.fnCheck(ret, err)
	}
}