mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-15 20:45:18 +00:00
354 lines
11 KiB
Go
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)
|
|
}
|