package evm_test import ( "math/big" "github.com/ethereum/go-ethereum/common" authtypes "github.com/cosmos/cosmos-sdk/x/auth/types" "github.com/tharsis/ethermint/crypto/ethsecp256k1" "github.com/tharsis/ethermint/x/evm" "github.com/tharsis/ethermint/x/evm/types" ) func (suite *EvmTestSuite) TestInitGenesis() { privkey, err := ethsecp256k1.GenerateKey() suite.Require().NoError(err) address := common.HexToAddress(privkey.PubKey().Address().String()) testCases := []struct { name string malleate func() genState *types.GenesisState expPanic bool }{ { "default", func() {}, types.DefaultGenesisState(), false, }, { "valid account", func() { acc := suite.app.AccountKeeper.NewAccountWithAddress(suite.ctx, address.Bytes()) suite.Require().NotNil(acc) suite.app.EvmKeeper.AddBalance(address, big.NewInt(1)) suite.app.AccountKeeper.SetAccount(suite.ctx, acc) }, &types.GenesisState{ Params: types.DefaultParams(), Accounts: []types.GenesisAccount{ { Address: address.String(), Storage: types.Storage{ {Key: common.BytesToHash([]byte("key")).String(), Value: common.BytesToHash([]byte("value")).String()}, }, }, }, }, false, }, { "account not found", func() {}, &types.GenesisState{ Params: types.DefaultParams(), Accounts: []types.GenesisAccount{ { Address: address.String(), }, }, }, true, }, { "invalid account type", func() { acc := authtypes.NewBaseAccountWithAddress(address.Bytes()) suite.app.AccountKeeper.SetAccount(suite.ctx, acc) }, &types.GenesisState{ Params: types.DefaultParams(), Accounts: []types.GenesisAccount{ { Address: address.String(), }, }, }, true, }, } for _, tc := range testCases { suite.Run(tc.name, func() { suite.SetupTest() // reset values tc.malleate() if tc.expPanic { suite.Require().Panics( func() { _ = evm.InitGenesis(suite.ctx, suite.app.EvmKeeper, suite.app.AccountKeeper, *tc.genState) }, ) } else { suite.Require().NotPanics( func() { _ = evm.InitGenesis(suite.ctx, suite.app.EvmKeeper, suite.app.AccountKeeper, *tc.genState) }, ) } }) } }