github.com/Finschia/finschia-sdk@v0.48.1/x/gov/keeper/common_test.go (about)

     1  package keeper_test
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/stretchr/testify/require"
     7  
     8  	"github.com/Finschia/finschia-sdk/simapp"
     9  	sdk "github.com/Finschia/finschia-sdk/types"
    10  	"github.com/Finschia/finschia-sdk/x/gov/types"
    11  	"github.com/Finschia/finschia-sdk/x/staking"
    12  	stakingkeeper "github.com/Finschia/finschia-sdk/x/staking/keeper"
    13  	stakingtypes "github.com/Finschia/finschia-sdk/x/staking/types"
    14  )
    15  
    16  var TestProposal = types.NewTextProposal("Test", "description")
    17  
    18  func createValidators(t *testing.T, ctx sdk.Context, app *simapp.SimApp, powers []int64) ([]sdk.AccAddress, []sdk.ValAddress) {
    19  	addrs := simapp.AddTestAddrsIncremental(app, ctx, 5, sdk.NewInt(30000000))
    20  	valAddrs := simapp.ConvertAddrsToValAddrs(addrs)
    21  	pks := simapp.CreateTestPubKeys(5)
    22  	cdc := simapp.MakeTestEncodingConfig().Marshaler
    23  
    24  	app.StakingKeeper = stakingkeeper.NewKeeper(
    25  		cdc,
    26  		app.GetKey(stakingtypes.StoreKey),
    27  		app.AccountKeeper,
    28  		app.BankKeeper,
    29  		app.GetSubspace(stakingtypes.ModuleName),
    30  	)
    31  
    32  	val1, err := stakingtypes.NewValidator(valAddrs[0], pks[0], stakingtypes.Description{})
    33  	require.NoError(t, err)
    34  	val2, err := stakingtypes.NewValidator(valAddrs[1], pks[1], stakingtypes.Description{})
    35  	require.NoError(t, err)
    36  	val3, err := stakingtypes.NewValidator(valAddrs[2], pks[2], stakingtypes.Description{})
    37  	require.NoError(t, err)
    38  
    39  	app.StakingKeeper.SetValidator(ctx, val1)
    40  	app.StakingKeeper.SetValidator(ctx, val2)
    41  	app.StakingKeeper.SetValidator(ctx, val3)
    42  	app.StakingKeeper.SetValidatorByConsAddr(ctx, val1)
    43  	app.StakingKeeper.SetValidatorByConsAddr(ctx, val2)
    44  	app.StakingKeeper.SetValidatorByConsAddr(ctx, val3)
    45  	app.StakingKeeper.SetNewValidatorByPowerIndex(ctx, val1)
    46  	app.StakingKeeper.SetNewValidatorByPowerIndex(ctx, val2)
    47  	app.StakingKeeper.SetNewValidatorByPowerIndex(ctx, val3)
    48  
    49  	_, _ = app.StakingKeeper.Delegate(ctx, addrs[0], app.StakingKeeper.TokensFromConsensusPower(ctx, powers[0]), stakingtypes.Unbonded, val1, true)
    50  	_, _ = app.StakingKeeper.Delegate(ctx, addrs[1], app.StakingKeeper.TokensFromConsensusPower(ctx, powers[1]), stakingtypes.Unbonded, val2, true)
    51  	_, _ = app.StakingKeeper.Delegate(ctx, addrs[2], app.StakingKeeper.TokensFromConsensusPower(ctx, powers[2]), stakingtypes.Unbonded, val3, true)
    52  
    53  	_ = staking.EndBlocker(ctx, app.StakingKeeper)
    54  
    55  	return addrs, valAddrs
    56  }