package keeper_test import ( "math/big" "testing" sdk "github.com/cosmos/cosmos-sdk/types" "github.com/ethereum/go-ethereum/common/math" "github.com/kava-labs/kava/x/evmutil/keeper" "github.com/kava-labs/kava/x/evmutil/testutil" "github.com/kava-labs/kava/x/evmutil/types" "github.com/stretchr/testify/suite" ) type MsgServerSuite struct { testutil.Suite msgServer types.MsgServer } func (suite *MsgServerSuite) SetupTest() { suite.Suite.SetupTest() suite.msgServer = keeper.NewMsgServerImpl(suite.App.GetEvmutilKeeper()) } func TestMsgServerSuite(t *testing.T) { suite.Run(t, new(MsgServerSuite)) } func (suite *MsgServerSuite) TestConvertCoinToERC20() { invoker, err := sdk.AccAddressFromBech32("kava123fxg0l602etulhhcdm0vt7l57qya5wjcrwhzz") suite.Require().NoError(err) err = suite.App.FundAccount(suite.Ctx, invoker, sdk.NewCoins(sdk.NewCoin("erc20/usdc", sdk.NewInt(10000)))) suite.Require().NoError(err) contractAddr := suite.DeployERC20() pair := types.NewConversionPair( contractAddr, "erc20/usdc", ) // Module account should have starting balance pairStartingBal := big.NewInt(10000) err = suite.Keeper.MintERC20( suite.Ctx, pair.GetAddress(), // contractAddr types.NewInternalEVMAddress(types.ModuleEVMAddress), //receiver pairStartingBal, ) suite.Require().NoError(err) type errArgs struct { expectPass bool contains string } tests := []struct { name string msg types.MsgConvertCoinToERC20 errArgs errArgs }{ { "valid", types.NewMsgConvertCoinToERC20( invoker.String(), "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2", sdk.NewCoin("erc20/usdc", sdk.NewInt(1234)), ), errArgs{ expectPass: true, }, }, { "invalid - odd length hex address", types.NewMsgConvertCoinToERC20( invoker.String(), "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc", sdk.NewCoin("erc20/usdc", sdk.NewInt(1234)), ), errArgs{ expectPass: false, contains: "invalid Receiver address: string is not a hex address", }, }, // Amount coin is not validated by msg_server, but on msg itself } for _, tc := range tests { suite.Run(tc.name, func() { _, err := suite.msgServer.ConvertCoinToERC20(sdk.WrapSDKContext(suite.Ctx), &tc.msg) if tc.errArgs.expectPass { suite.Require().NoError(err) bal := suite.GetERC20BalanceOf( types.ERC20MintableBurnableContract.ABI, pair.GetAddress(), testutil.MustNewInternalEVMAddressFromString(tc.msg.Receiver), ) suite.Require().Equal(tc.msg.Amount.Amount.BigInt(), bal, "balance should match converted amount") // msg server event suite.EventsContains(suite.GetEvents(), sdk.NewEvent( sdk.EventTypeMessage, sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory), sdk.NewAttribute(sdk.AttributeKeySender, tc.msg.Initiator), )) // keeper event suite.EventsContains(suite.GetEvents(), sdk.NewEvent( types.EventTypeConvertCoinToERC20, sdk.NewAttribute(types.AttributeKeyInitiator, tc.msg.Initiator), sdk.NewAttribute(types.AttributeKeyReceiver, tc.msg.Receiver), sdk.NewAttribute(types.AttributeKeyERC20Address, pair.GetAddress().String()), sdk.NewAttribute(types.AttributeKeyAmount, tc.msg.Amount.String()), )) } else { suite.Require().Error(err) suite.Require().Contains(err.Error(), tc.errArgs.contains) } }) } } func (suite *MsgServerSuite) TestConvertERC20ToCoin() { contractAddr := suite.DeployERC20() pair := types.NewConversionPair( contractAddr, "erc20/usdc", ) // give invoker account some erc20 usdc to begin with invoker := testutil.MustNewInternalEVMAddressFromString("0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2") pairStartingBal := big.NewInt(10_000_000) err := suite.Keeper.MintERC20( suite.Ctx, pair.GetAddress(), // contractAddr invoker, //receiver pairStartingBal, ) suite.Require().NoError(err) invokerCosmosAddr, err := sdk.AccAddressFromHex(invoker.String()[2:]) suite.Require().NoError(err) // create user account, otherwise `CallEVMWithData` will fail due to failing to get user account when finding its sequence. err = suite.App.FundAccount(suite.Ctx, invokerCosmosAddr, sdk.NewCoins(sdk.NewCoin(pair.Denom, sdk.ZeroInt()))) suite.Require().NoError(err) type errArgs struct { expectPass bool contains string } tests := []struct { name string msg types.MsgConvertERC20ToCoin approvalAmount *big.Int errArgs errArgs }{ { "valid", types.NewMsgConvertERC20ToCoin( invoker, invokerCosmosAddr, contractAddr, sdk.NewInt(10_000), ), math.MaxBig256, errArgs{ expectPass: true, }, }, { "invalid - invalid hex address", types.MsgConvertERC20ToCoin{ Initiator: "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc", Receiver: invokerCosmosAddr.String(), KavaERC20Address: contractAddr.String(), Amount: sdk.NewInt(10_000), }, math.MaxBig256, errArgs{ expectPass: false, contains: "invalid initiator address: string is not a hex address", }, }, { "invalid - insufficient coins", types.NewMsgConvertERC20ToCoin( invoker, invokerCosmosAddr, contractAddr, sdk.NewIntFromBigInt(pairStartingBal).Add(sdk.OneInt()), ), math.MaxBig256, errArgs{ expectPass: false, contains: "transfer amount exceeds balance", }, }, { "invalid - contract address", types.NewMsgConvertERC20ToCoin( invoker, invokerCosmosAddr, testutil.MustNewInternalEVMAddressFromString("0x7Bbf300890857b8c241b219C6a489431669b3aFA"), sdk.NewInt(10_000), ), math.MaxBig256, errArgs{ expectPass: false, contains: "ERC20 token not enabled to convert to sdk.Coin", }, }, } for _, tc := range tests { suite.Run(tc.name, func() { _, err := suite.msgServer.ConvertERC20ToCoin(sdk.WrapSDKContext(suite.Ctx), &tc.msg) if tc.errArgs.expectPass { suite.Require().NoError(err) // validate user balance after conversion bal := suite.GetERC20BalanceOf( types.ERC20MintableBurnableContract.ABI, pair.GetAddress(), testutil.MustNewInternalEVMAddressFromString(tc.msg.Initiator), ) expectedBal := sdk.NewIntFromBigInt(pairStartingBal).Sub(tc.msg.Amount) suite.Require().Equal(expectedBal.BigInt(), bal, "user erc20 balance is invalid") // validate user coin balance coinBal := suite.App.GetBankKeeper().GetBalance(suite.Ctx, invokerCosmosAddr, pair.Denom) suite.Require().Equal(tc.msg.Amount, coinBal.Amount, "user coin balance is invalid") // msg server event suite.EventsContains(suite.GetEvents(), sdk.NewEvent( sdk.EventTypeMessage, sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory), sdk.NewAttribute(sdk.AttributeKeySender, tc.msg.Initiator), )) // keeper event suite.EventsContains(suite.GetEvents(), sdk.NewEvent( types.EventTypeConvertERC20ToCoin, sdk.NewAttribute(types.AttributeKeyERC20Address, pair.GetAddress().String()), sdk.NewAttribute(types.AttributeKeyInitiator, tc.msg.Initiator), sdk.NewAttribute(types.AttributeKeyReceiver, tc.msg.Receiver), sdk.NewAttribute(types.AttributeKeyAmount, sdk.NewCoin(pair.Denom, tc.msg.Amount).String()), )) } else { suite.Require().Error(err) suite.Require().Contains(err.Error(), tc.errArgs.contains) } }) } }