mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-27 08:45:19 +00:00
267 lines
7.3 KiB
Go
267 lines
7.3 KiB
Go
|
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)
|
||
|
}
|
||
|
})
|
||
|
}
|
||
|
}
|