github.com/Finschia/finschia-sdk@v0.49.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  	t.Helper()
    20  	addrs := simapp.AddTestAddrsIncremental(app, ctx, 5, sdk.NewInt(30000000))
    21  	valAddrs := simapp.ConvertAddrsToValAddrs(addrs)
    22  	pks := simapp.CreateTestPubKeys(5)
    23  	cdc := simapp.MakeTestEncodingConfig().Marshaler
    24  
    25  	app.StakingKeeper = stakingkeeper.NewKeeper(
    26  		cdc,
    27  		app.GetKey(stakingtypes.StoreKey),
    28  		app.AccountKeeper,
    29  		app.BankKeeper,
    30  		app.GetSubspace(stakingtypes.ModuleName),
    31  	)
    32  
    33  	val1, err := stakingtypes.NewValidator(valAddrs[0], pks[0], stakingtypes.Description{})
    34  	require.NoError(t, err)
    35  	val2, err := stakingtypes.NewValidator(valAddrs[1], pks[1], stakingtypes.Description{})
    36  	require.NoError(t, err)
    37  	val3, err := stakingtypes.NewValidator(valAddrs[2], pks[2], stakingtypes.Description{})
    38  	require.NoError(t, err)
    39  
    40  	app.StakingKeeper.SetValidator(ctx, val1)
    41  	app.StakingKeeper.SetValidator(ctx, val2)
    42  	app.StakingKeeper.SetValidator(ctx, val3)
    43  	err = app.StakingKeeper.SetValidatorByConsAddr(ctx, val1)
    44  	require.NoError(t, err)
    45  	err = app.StakingKeeper.SetValidatorByConsAddr(ctx, val2)
    46  	require.NoError(t, err)
    47  	err = app.StakingKeeper.SetValidatorByConsAddr(ctx, val3)
    48  	require.NoError(t, err)
    49  	app.StakingKeeper.SetNewValidatorByPowerIndex(ctx, val1)
    50  	app.StakingKeeper.SetNewValidatorByPowerIndex(ctx, val2)
    51  	app.StakingKeeper.SetNewValidatorByPowerIndex(ctx, val3)
    52  
    53  	_, _ = app.StakingKeeper.Delegate(ctx, addrs[0], app.StakingKeeper.TokensFromConsensusPower(ctx, powers[0]), stakingtypes.Unbonded, val1, true)
    54  	_, _ = app.StakingKeeper.Delegate(ctx, addrs[1], app.StakingKeeper.TokensFromConsensusPower(ctx, powers[1]), stakingtypes.Unbonded, val2, true)
    55  	_, _ = app.StakingKeeper.Delegate(ctx, addrs[2], app.StakingKeeper.TokensFromConsensusPower(ctx, powers[2]), stakingtypes.Unbonded, val3, true)
    56  
    57  	_ = staking.EndBlocker(ctx, app.StakingKeeper)
    58  
    59  	return addrs, valAddrs
    60  }