mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-27 08:45:19 +00:00
101 lines
3.4 KiB
Go
101 lines
3.4 KiB
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())))
|
|
}
|