0g-chain/precompiles/testutil/suite.go

package testutil

import (
	"math/big"
	"strings"
	"time"

	"github.com/0glabs/0g-chain/app"
	"github.com/0glabs/0g-chain/chaincfg"
	precopmiles_common "github.com/0glabs/0g-chain/precompiles/common"
	"github.com/0glabs/0g-chain/x/bep3/types"
	"github.com/cosmos/cosmos-sdk/crypto/keyring"
	"github.com/ethereum/go-ethereum/common"
	"github.com/ethereum/go-ethereum/crypto"
	emtests "github.com/evmos/ethermint/tests"
	evmkeeper "github.com/evmos/ethermint/x/evm/keeper"
	"github.com/evmos/ethermint/x/evm/statedb"
	"github.com/stretchr/testify/suite"

	cryptotypes "github.com/cosmos/cosmos-sdk/crypto/types"
	sdk "github.com/cosmos/cosmos-sdk/types"
	stakingkeeper "github.com/cosmos/cosmos-sdk/x/staking/keeper"
	ethtypes "github.com/ethereum/go-ethereum/core/types"
	"github.com/evmos/ethermint/crypto/ethsecp256k1"

	tmproto "github.com/cometbft/cometbft/proto/tendermint/types"
	stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
)

type PrecompileTestSuite struct {
	suite.Suite

	StakingKeeper *stakingkeeper.Keeper
	App           app.TestApp
	Ctx           sdk.Context
	QueryClient   types.QueryClient
	Addresses     []sdk.AccAddress

	EvmKeeper *evmkeeper.Keeper
	EthSigner ethtypes.Signer
	Statedb   *statedb.StateDB
}

type TestSigner struct {
	Addr    common.Address
	HexAddr string
	PrivKey cryptotypes.PrivKey
	Signer  keyring.Signer
	ValAddr sdk.ValAddress
	AccAddr sdk.AccAddress
}

func (suite *PrecompileTestSuite) GenSigner() *TestSigner {
	var s TestSigner
	addr, priv := emtests.NewAddrKey()
	s.PrivKey = priv
	s.Addr = addr
	s.HexAddr = precopmiles_common.ToLowerHexWithoutPrefix(s.Addr)
	s.Signer = emtests.NewSigner(priv)
	valAddr, _ := sdk.ValAddressFromHex(s.HexAddr)
	accAddr, _ := sdk.AccAddressFromHexUnsafe(s.HexAddr)
	s.ValAddr = valAddr
	s.AccAddr = accAddr

	// 10000 a0gi for test
	suite.App.GetBankKeeper().MintCoins(suite.Ctx, "mint", sdk.NewCoins(chaincfg.MakeCoinForGasDenom(big.NewInt(10000000000))))
	suite.App.GetBankKeeper().SendCoinsFromModuleToAccount(suite.Ctx, "mint", accAddr, sdk.NewCoins(chaincfg.MakeCoinForGasDenom(big.NewInt(10000000000))))
	return &s
}

func (suite *PrecompileTestSuite) SetupTest() {
	chaincfg.SetSDKConfig()
	suite.App = app.NewTestApp()
	suite.App.InitializeFromGenesisStates()
	suite.StakingKeeper = suite.App.GetStakingKeeper()

	// make block header
	privkey, _ := ethsecp256k1.GenerateKey()
	consAddress := sdk.ConsAddress(privkey.PubKey().Address())
	key, err := privkey.ToECDSA()
	suite.Assert().NoError(err)
	hexAddr := strings.ToLower(crypto.PubkeyToAddress(key.PublicKey).Hex()[2:])
	valAddr, err := sdk.ValAddressFromHex(hexAddr)
	suite.Assert().NoError(err)
	suite.Ctx = suite.App.NewContext(true, tmproto.Header{Height: 1, ChainID: app.TestChainId, ProposerAddress: consAddress, Time: time.Now()})
	newValidator, err := stakingtypes.NewValidator(valAddr, privkey.PubKey(), stakingtypes.Description{})
	suite.Assert().NoError(err)
	err = suite.StakingKeeper.SetValidatorByConsAddr(suite.Ctx, newValidator)
	suite.Assert().NoError(err)
	suite.StakingKeeper.SetValidator(suite.Ctx, newValidator)

	_, accAddresses := app.GeneratePrivKeyAddressPairs(10)
	suite.Addresses = accAddresses

	suite.EvmKeeper = suite.App.GetEvmKeeper()

	suite.EthSigner = ethtypes.LatestSignerForChainID(suite.EvmKeeper.ChainID())

	suite.Statedb = statedb.New(suite.Ctx, suite.EvmKeeper, statedb.NewEmptyTxConfig(common.BytesToHash(suite.Ctx.HeaderHash().Bytes())))
}