mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-10 10:05:18 +00:00
115 lines
3.3 KiB
Go
115 lines
3.3 KiB
Go
package evmutil_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
sdkmath "cosmossdk.io/math"
|
|
"github.com/0glabs/0g-chain/x/evmutil"
|
|
"github.com/0glabs/0g-chain/x/evmutil/testutil"
|
|
"github.com/0glabs/0g-chain/x/evmutil/types"
|
|
authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
|
|
)
|
|
|
|
type genesisTestSuite struct {
|
|
testutil.Suite
|
|
}
|
|
|
|
func (suite *genesisTestSuite) SetupTest() {
|
|
suite.Suite.SetupTest()
|
|
}
|
|
|
|
func (s *genesisTestSuite) TestInitGenesis_SetAccounts() {
|
|
gs := types.NewGenesisState(
|
|
[]types.Account{
|
|
{Address: s.Addrs[0], Balance: sdkmath.NewInt(100)},
|
|
},
|
|
types.DefaultParams(),
|
|
)
|
|
accounts := s.Keeper.GetAllAccounts(s.Ctx)
|
|
s.Require().Len(accounts, 0)
|
|
evmutil.InitGenesis(s.Ctx, s.Keeper, gs, s.AccountKeeper)
|
|
accounts = s.Keeper.GetAllAccounts(s.Ctx)
|
|
s.Require().Len(accounts, 1)
|
|
account := s.Keeper.GetAccount(s.Ctx, s.Addrs[0])
|
|
s.Require().Equal(account.Address, s.Addrs[0])
|
|
s.Require().Equal(account.Balance, sdkmath.NewInt(100))
|
|
}
|
|
|
|
func (s *genesisTestSuite) TestInitGenesis_SetParams() {
|
|
params := types.DefaultParams()
|
|
conversionPair := types.ConversionPair{
|
|
ZgChainERC20Address: testutil.MustNewInternalEVMAddressFromString("0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2").Bytes(),
|
|
Denom: "weth",
|
|
}
|
|
params.EnabledConversionPairs = []types.ConversionPair{conversionPair}
|
|
gs := types.NewGenesisState(
|
|
[]types.Account{},
|
|
params,
|
|
)
|
|
evmutil.InitGenesis(s.Ctx, s.Keeper, gs, s.AccountKeeper)
|
|
params = s.Keeper.GetParams(s.Ctx)
|
|
s.Require().Len(params.EnabledConversionPairs, 1)
|
|
s.Require().Equal(conversionPair, params.EnabledConversionPairs[0])
|
|
}
|
|
|
|
func (s *genesisTestSuite) TestInitGenesis_ValidateFail() {
|
|
gs := types.NewGenesisState(
|
|
[]types.Account{
|
|
{Address: s.Addrs[0], Balance: sdkmath.NewInt(-100)},
|
|
},
|
|
types.DefaultParams(),
|
|
)
|
|
s.Require().Panics(func() {
|
|
evmutil.InitGenesis(s.Ctx, s.Keeper, gs, s.AccountKeeper)
|
|
})
|
|
}
|
|
|
|
func (s *genesisTestSuite) TestInitGenesis_ModuleAccount() {
|
|
gs := types.NewGenesisState(
|
|
[]types.Account{},
|
|
types.DefaultParams(),
|
|
)
|
|
s.Require().NotPanics(func() {
|
|
evmutil.InitGenesis(s.Ctx, s.Keeper, gs, s.AccountKeeper)
|
|
})
|
|
// check for module account this way b/c GetModuleAccount creates if not existing.
|
|
acc := s.AccountKeeper.GetAccount(s.Ctx, s.AccountKeeper.GetModuleAddress(types.ModuleName))
|
|
s.Require().NotNil(acc)
|
|
_, ok := acc.(authtypes.ModuleAccountI)
|
|
s.Require().True(ok)
|
|
}
|
|
|
|
func (s *genesisTestSuite) TestExportGenesis() {
|
|
accounts := []types.Account{
|
|
{Address: s.Addrs[0], Balance: sdkmath.NewInt(10)},
|
|
{Address: s.Addrs[1], Balance: sdkmath.NewInt(20)},
|
|
}
|
|
for _, account := range accounts {
|
|
s.Keeper.SetAccount(s.Ctx, account)
|
|
}
|
|
params := types.DefaultParams()
|
|
params.EnabledConversionPairs = []types.ConversionPair{
|
|
{
|
|
ZgChainERC20Address: testutil.MustNewInternalEVMAddressFromString("0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2").Bytes(),
|
|
Denom: "weth"},
|
|
}
|
|
params.AllowedCosmosDenoms = []types.AllowedCosmosCoinERC20Token{
|
|
{
|
|
CosmosDenom: "hard",
|
|
Name: "0G EVM HARD",
|
|
Symbol: "HARD",
|
|
Decimals: 6,
|
|
},
|
|
}
|
|
s.Keeper.SetParams(s.Ctx, params)
|
|
gs := evmutil.ExportGenesis(s.Ctx, s.Keeper)
|
|
s.Require().Equal(gs.Accounts, accounts)
|
|
s.Require().Equal(params, gs.Params)
|
|
}
|
|
|
|
func TestGenesisTestSuite(t *testing.T) {
|
|
suite.Run(t, new(genesisTestSuite))
|
|
}
|