0g-chain/x/evmutil/keeper/conversion_evm_native_test.go
2024-05-01 11:17:24 +08:00

354 lines
11 KiB
Go

package keeper_test
import (
"math/big"
"testing"
sdkmath "cosmossdk.io/math"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/ethereum/go-ethereum/common"
"github.com/stretchr/testify/suite"
"github.com/0glabs/0g-chain/x/evmutil/testutil"
"github.com/0glabs/0g-chain/x/evmutil/types"
)
type ConversionTestSuite struct {
testutil.Suite
}
func TestConversionTestSuite(t *testing.T) {
suite.Run(t, new(ConversionTestSuite))
}
func (suite *ConversionTestSuite) TestMint() {
pair := types.NewConversionPair(
testutil.MustNewInternalEVMAddressFromString("0x000000000000000000000000000000000000000A"),
"erc20/usdc",
)
amount := big.NewInt(100)
recipient := suite.Key1.PubKey().Address().Bytes()
coin, err := suite.Keeper.MintConversionPairCoin(suite.Ctx, pair, amount, recipient)
suite.Require().NoError(err)
suite.Require().Equal(sdk.NewCoin(pair.Denom, sdkmath.NewIntFromBigInt(amount)), coin)
bal := suite.App.GetBankKeeper().GetBalance(suite.Ctx, recipient, pair.Denom)
suite.Require().Equal(amount, bal.Amount.BigInt(), "minted amount should increase balance")
}
func (suite *ConversionTestSuite) TestBurn_InsufficientBalance() {
pair := types.NewConversionPair(
testutil.MustNewInternalEVMAddressFromString("0x000000000000000000000000000000000000000A"),
"erc20/usdc",
)
amount := sdkmath.NewInt(100)
recipient := suite.Key1.PubKey().Address().Bytes()
err := suite.Keeper.BurnConversionPairCoin(suite.Ctx, pair, sdk.NewCoin(pair.Denom, amount), recipient)
suite.Require().Error(err)
suite.Require().Equal("0erc20/usdc is smaller than 100erc20/usdc: insufficient funds", err.Error())
}
func (suite *ConversionTestSuite) TestBurn() {
pair := types.NewConversionPair(
testutil.MustNewInternalEVMAddressFromString("0x000000000000000000000000000000000000000A"),
"erc20/usdc",
)
amount := sdkmath.NewInt(100)
recipient := suite.Key1.PubKey().Address().Bytes()
coin, err := suite.Keeper.MintConversionPairCoin(suite.Ctx, pair, amount.BigInt(), recipient)
suite.Require().NoError(err)
suite.Require().Equal(sdk.NewCoin(pair.Denom, amount), coin)
bal := suite.App.GetBankKeeper().GetBalance(suite.Ctx, recipient, pair.Denom)
suite.Require().Equal(amount, bal.Amount, "minted amount should increase balance")
err = suite.Keeper.BurnConversionPairCoin(suite.Ctx, pair, sdk.NewCoin(pair.Denom, amount), recipient)
suite.Require().NoError(err)
bal = suite.App.GetBankKeeper().GetBalance(suite.Ctx, recipient, pair.Denom)
suite.Require().Equal(sdk.ZeroInt(), bal.Amount, "balance should be zero after burn")
}
func (suite *ConversionTestSuite) TestUnlockERC20Tokens() {
contractAddr := suite.DeployERC20()
pair := types.NewConversionPair(
contractAddr,
"erc20/usdc",
)
amount := big.NewInt(100)
recipient := types.NewInternalEVMAddress(common.BytesToAddress(suite.Key1.PubKey().Address()))
moduleAddr := types.NewInternalEVMAddress(types.ModuleEVMAddress)
// Mint some initial balance for module account to transfer
err := suite.Keeper.MintERC20(
suite.Ctx,
pair.GetAddress(), // contractAddr
moduleAddr, //receiver
amount,
)
suite.Require().NoError(err)
err = suite.Keeper.UnlockERC20Tokens(suite.Ctx, pair, amount, recipient)
suite.Require().NoError(err)
// Check balance of recipient
bal := suite.GetERC20BalanceOf(
types.ERC20MintableBurnableContract.ABI,
pair.GetAddress(),
recipient,
)
suite.Require().Equal(amount, bal, "balance should increase by unlock amount")
// Check balance of module account
bal = suite.GetERC20BalanceOf(
types.ERC20MintableBurnableContract.ABI,
pair.GetAddress(),
moduleAddr,
)
suite.Require().Equal(
// String() due to non-equal struct values for 0
big.NewInt(0).String(),
bal.String(),
"balance should decrease module account by unlock amount",
)
}
func (suite *ConversionTestSuite) TestUnlockERC20Tokens_Insufficient() {
contractAddr := suite.DeployERC20()
pair := types.NewConversionPair(
contractAddr,
"erc20/usdc",
)
amount := big.NewInt(100)
recipient := types.NewInternalEVMAddress(common.BytesToAddress(suite.Key1.PubKey().Address()))
// Module account has 0 balance, cannot unlock
err := suite.Keeper.UnlockERC20Tokens(suite.Ctx, pair, amount, recipient)
suite.Require().Error(err)
suite.Require().Contains(err.Error(), "execution reverted: ERC20: transfer amount exceeds balance")
}
func (suite *ConversionTestSuite) TestConvertCoinToERC20() {
contractAddr := suite.DeployERC20()
pair := types.NewConversionPair(
contractAddr,
"erc20/usdc",
)
amount := big.NewInt(100)
originAcc := sdk.AccAddress(suite.Key1.PubKey().Address().Bytes())
recipientAcc := types.NewInternalEVMAddress(common.BytesToAddress(suite.Key2.PubKey().Address()))
moduleAddr := types.NewInternalEVMAddress(types.ModuleEVMAddress)
// Starting balance of origin account
coin, err := suite.Keeper.MintConversionPairCoin(suite.Ctx, pair, amount, originAcc)
suite.Require().NoError(err)
suite.Require().Equal(sdk.NewCoin(pair.Denom, sdkmath.NewIntFromBigInt(amount)), coin)
// Mint same initial balance for module account as backing erc20 supply
err = suite.Keeper.MintERC20(
suite.Ctx,
pair.GetAddress(), // contractAddr
moduleAddr, //receiver
amount,
)
suite.Require().NoError(err)
// convert coin to erc20
ctx := suite.Ctx.WithGasMeter(sdk.NewInfiniteGasMeter())
err = suite.Keeper.ConvertCoinToERC20(
ctx,
originAcc,
recipientAcc,
sdk.NewCoin(pair.Denom, sdkmath.NewIntFromBigInt(amount)),
)
suite.Require().NoError(err)
suite.Require().LessOrEqual(ctx.GasMeter().GasConsumed(), uint64(500000))
suite.Require().GreaterOrEqual(ctx.GasMeter().GasConsumed(), uint64(50000))
// Source should decrease
bal := suite.App.GetBankKeeper().GetBalance(suite.Ctx, originAcc, pair.Denom)
suite.Require().Equal(sdk.ZeroInt(), bal.Amount, "conversion should decrease source balance")
// Module bal should also decrease
moduleBal := suite.GetERC20BalanceOf(
types.ERC20MintableBurnableContract.ABI,
pair.GetAddress(),
moduleAddr,
)
suite.Require().Equal(
// String() due to non-equal struct values for 0
big.NewInt(0).String(),
moduleBal.String(),
"balance should decrease module account by unlock amount",
)
// Recipient balance should increase by same amount
recipientBal := suite.GetERC20BalanceOf(
types.ERC20MintableBurnableContract.ABI,
pair.GetAddress(),
recipientAcc,
)
suite.Require().Equal(
// String() due to non-equal struct values for 0
amount,
recipientBal,
"recipient balance should increase",
)
suite.EventsContains(suite.GetEvents(),
sdk.NewEvent(
types.EventTypeConvertCoinToERC20,
sdk.NewAttribute(types.AttributeKeyInitiator, originAcc.String()),
sdk.NewAttribute(types.AttributeKeyReceiver, recipientAcc.String()),
sdk.NewAttribute(types.AttributeKeyERC20Address, pair.GetAddress().String()),
sdk.NewAttribute(types.AttributeKeyAmount, coin.String()),
))
}
func (suite *ConversionTestSuite) TestConvertCoinToERC20_InsufficientBalance() {
contractAddr := suite.DeployERC20()
pair := types.NewConversionPair(
contractAddr,
"erc20/usdc",
)
amount := big.NewInt(100)
originAcc := sdk.AccAddress(suite.Key1.PubKey().Address().Bytes())
recipientAcc := types.NewInternalEVMAddress(common.BytesToAddress(suite.Key2.PubKey().Address()))
err := suite.Keeper.ConvertCoinToERC20(
suite.Ctx,
originAcc,
recipientAcc,
sdk.NewCoin(pair.Denom, sdkmath.NewIntFromBigInt(amount)),
)
suite.Require().Error(err)
suite.Require().Equal("0erc20/usdc is smaller than 100erc20/usdc: insufficient funds", err.Error())
}
func (suite *ConversionTestSuite) TestConvertCoinToERC20_NotEnabled() {
contractAddr := suite.DeployERC20()
pair := types.NewConversionPair(
contractAddr,
"erc20/notenabled",
)
amount := big.NewInt(100)
originAcc := sdk.AccAddress(suite.Key1.PubKey().Address().Bytes())
recipientAcc := types.NewInternalEVMAddress(common.BytesToAddress(suite.Key2.PubKey().Address()))
err := suite.Keeper.ConvertCoinToERC20(
suite.Ctx,
originAcc,
recipientAcc,
sdk.NewCoin(pair.Denom, sdkmath.NewIntFromBigInt(amount)),
)
suite.Require().Error(err)
suite.Require().Equal("erc20/notenabled: ERC20 token not enabled to convert to sdk.Coin", err.Error())
}
func (suite *ConversionTestSuite) TestConvertERC20ToCoin() {
contractAddr := suite.DeployERC20()
pair := types.NewConversionPair(
contractAddr,
"erc20/usdc",
)
totalAmt := big.NewInt(100)
userAddr := sdk.AccAddress(suite.Key1.PubKey().Address().Bytes())
userEvmAddr := types.NewInternalEVMAddress(common.BytesToAddress(suite.Key1.PubKey().Address()))
// Mint same initial balance for user account
err := suite.Keeper.MintERC20(
suite.Ctx,
pair.GetAddress(), // contractAddr
userEvmAddr, //receiver
totalAmt,
)
suite.Require().NoError(err)
ctx := suite.Ctx.WithGasMeter(sdk.NewInfiniteGasMeter())
convertAmt := sdkmath.NewInt(50)
err = suite.Keeper.ConvertERC20ToCoin(
ctx,
userEvmAddr,
userAddr,
pair.GetAddress(),
convertAmt,
)
suite.Require().NoError(err)
suite.Require().LessOrEqual(ctx.GasMeter().GasConsumed(), uint64(500000))
suite.Require().GreaterOrEqual(ctx.GasMeter().GasConsumed(), uint64(50000))
// bank balance should decrease
bal := suite.App.GetBankKeeper().GetBalance(suite.Ctx, userAddr, pair.Denom)
suite.Require().Equal(convertAmt, bal.Amount, "conversion should decrease source balance")
// Module bal should also decrease
userBal := suite.GetERC20BalanceOf(
types.ERC20MintableBurnableContract.ABI,
pair.GetAddress(),
userEvmAddr,
)
suite.Require().Equal(
// String() due to non-equal struct values for 0
big.NewInt(50).String(),
userBal.String(),
"balance should decrease module account by unlock amount",
)
suite.EventsContains(suite.GetEvents(),
sdk.NewEvent(
types.EventTypeConvertERC20ToCoin,
sdk.NewAttribute(types.AttributeKeyERC20Address, pair.GetAddress().String()),
sdk.NewAttribute(types.AttributeKeyInitiator, userEvmAddr.String()),
sdk.NewAttribute(types.AttributeKeyReceiver, userAddr.String()),
sdk.NewAttribute(types.AttributeKeyAmount, sdk.NewCoin(pair.Denom, convertAmt).String()),
),
)
}
func (suite *ConversionTestSuite) TestConvertERC20ToCoin_EmptyContract() {
contractAddr := testutil.MustNewInternalEVMAddressFromString("0x15932E26f5BD4923d46a2b205191C4b5d5f43FE3")
pair := types.NewConversionPair(
contractAddr,
"erc20/usdc",
)
userAddr := sdk.AccAddress(suite.Key1.PubKey().Address().Bytes())
userEvmAddr := types.NewInternalEVMAddress(common.BytesToAddress(suite.Key1.PubKey().Address()))
convertAmt := sdkmath.NewInt(100)
// Trying to convert erc20 from an empty contract should fail
err := suite.Keeper.ConvertERC20ToCoin(
suite.Ctx,
userEvmAddr,
userAddr,
pair.GetAddress(),
convertAmt,
)
suite.Require().Error(err)
suite.Require().ErrorContains(err, "failed to retrieve balance: failed to unpack method balanceOf")
// bank balance should not change
bal := suite.App.GetBankKeeper().GetBalance(suite.Ctx, userAddr, pair.Denom)
suite.Require().Equal(sdk.ZeroInt(), bal.Amount)
}