mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-13 03:25:20 +00:00
70 lines
2.4 KiB
Go
70 lines
2.4 KiB
Go
package testutil
|
|
|
|
import (
|
|
"cosmossdk.io/math"
|
|
tmproto "github.com/cometbft/cometbft/proto/tendermint/types"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
stakingkeeper "github.com/cosmos/cosmos-sdk/x/staking/keeper"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/0glabs/0g-chain/app"
|
|
"github.com/0glabs/0g-chain/chaincfg"
|
|
"github.com/0glabs/0g-chain/x/dasigners/v1/keeper"
|
|
"github.com/0glabs/0g-chain/x/dasigners/v1/types"
|
|
stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
|
|
"github.com/evmos/ethermint/crypto/ethsecp256k1"
|
|
)
|
|
|
|
// Suite implements a test suite for the module integration tests
|
|
type Suite struct {
|
|
suite.Suite
|
|
|
|
Keeper keeper.Keeper
|
|
StakingKeeper *stakingkeeper.Keeper
|
|
App app.TestApp
|
|
Ctx sdk.Context
|
|
QueryClient types.QueryClient
|
|
Addresses []sdk.AccAddress
|
|
}
|
|
|
|
// SetupTest instantiates a new app, keepers, and sets suite state
|
|
func (suite *Suite) SetupTest() {
|
|
chaincfg.SetSDKConfig()
|
|
suite.App = app.NewTestApp()
|
|
suite.Keeper = suite.App.GetDASignersKeeper()
|
|
suite.StakingKeeper = suite.App.GetStakingKeeper()
|
|
suite.Ctx = suite.App.NewContext(true, tmproto.Header{})
|
|
_, accAddresses := app.GeneratePrivKeyAddressPairs(10)
|
|
suite.Addresses = accAddresses
|
|
|
|
// Set query client
|
|
queryHelper := suite.App.NewQueryServerTestHelper(suite.Ctx)
|
|
queryHandler := suite.Keeper
|
|
types.RegisterQueryServer(queryHelper, queryHandler)
|
|
suite.QueryClient = types.NewQueryClient(queryHelper)
|
|
}
|
|
|
|
func (suite *Suite) AddDelegation(from string, to string, amount math.Int) {
|
|
accAddr, err := sdk.AccAddressFromHexUnsafe(from)
|
|
suite.Require().NoError(err)
|
|
valAddr, err := sdk.ValAddressFromHex(to)
|
|
suite.Require().NoError(err)
|
|
validator, found := suite.StakingKeeper.GetValidator(suite.Ctx, valAddr)
|
|
if !found {
|
|
consPriv, err := ethsecp256k1.GenerateKey()
|
|
suite.Require().NoError(err)
|
|
newValidator, err := stakingtypes.NewValidator(valAddr, consPriv.PubKey(), stakingtypes.Description{})
|
|
suite.Require().NoError(err)
|
|
validator = newValidator
|
|
}
|
|
validator.Tokens = validator.Tokens.Add(amount)
|
|
validator.DelegatorShares = validator.DelegatorShares.Add(amount.ToLegacyDec())
|
|
suite.StakingKeeper.SetValidator(suite.Ctx, validator)
|
|
bonded := suite.Keeper.GetDelegatorBonded(suite.Ctx, accAddr)
|
|
suite.StakingKeeper.SetDelegation(suite.Ctx, stakingtypes.Delegation{
|
|
DelegatorAddress: accAddr.String(),
|
|
ValidatorAddress: valAddr.String(),
|
|
Shares: bonded.Add(amount).ToLegacyDec(),
|
|
})
|
|
}
|