Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

allocate multiple genesis coins on bank simulation #9

Merged
merged 7 commits into from
Jun 16, 2021
Merged

allocate multiple genesis coins on bank simulation #9

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
d18156b
allocate multiple genesis coins on bank simulation
mconcat Jun 13, 2021
114c4c4
fix test
mconcat Jun 13, 2021
f25a281
fix genesis
mconcat Jun 13, 2021
4b912ab
fix test final
mconcat Jun 13, 2021
5369f87
increase default tx size
mconcat Jun 13, 2021
53c6757
Update x/bank/simulation/genesis.go
mconcat Jun 13, 2021
b450390
fix test
mconcat Jun 13, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion simapp/helpers/test_helpers.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ import (

// SimAppChainID hardcoded chainID for simulation
const (
DefaultGenTxGas = 1000000
DefaultGenTxGas = 2000000
SimAppChainID = "simulation-app"
)

Expand Down
19 changes: 16 additions & 3 deletions x/bank/simulation/genesis.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,15 @@ func RandomGenesisSendParams(r *rand.Rand) types.SendEnabledParams {
return params.SendEnabled
}

func GenesisCoins(bondDenom string, initialStake int64) sdk.Coins {
coins := make(sdk.Coins, 4)
for i := range coins {
coins[i] = sdk.NewCoin(fmt.Sprintf("%s%02d", bondDenom, i), sdk.NewInt(initialStake))
}
coins = coins.Add(sdk.NewInt64Coin(bondDenom, initialStake))
return coins
}

// RandomGenesisBalances returns a slice of account balances. Each account has
// a balance of simState.InitialStake for sdk.DefaultBondDenom.
func RandomGenesisBalances(simState *module.SimulationState) []types.Balance {
Expand All @@ -43,7 +52,7 @@ func RandomGenesisBalances(simState *module.SimulationState) []types.Balance {
for _, acc := range simState.Accounts {
genesisBalances = append(genesisBalances, types.Balance{
Address: acc.Address.String(),
Coins: sdk.NewCoins(sdk.NewCoin(sdk.DefaultBondDenom, sdk.NewInt(simState.InitialStake))),
Coins: GenesisCoins(sdk.DefaultBondDenom, simState.InitialStake),
})
}

Expand All @@ -65,8 +74,12 @@ func RandomizedGenState(simState *module.SimulationState) {
)

numAccs := int64(len(simState.Accounts))
totalSupply := sdk.NewInt(simState.InitialStake * (numAccs + simState.NumBonded))
supply := sdk.NewCoins(sdk.NewCoin(sdk.DefaultBondDenom, totalSupply))

supply := GenesisCoins(sdk.DefaultBondDenom, simState.InitialStake)
for i, coin := range supply {
supply[i] = sdk.NewCoin(coin.Denom, coin.Amount.MulRaw(numAccs))
}
supply = supply.Add(sdk.NewInt64Coin(sdk.DefaultBondDenom, simState.InitialStake*simState.NumBonded))

bankGenesis := types.GenesisState{
Params: types.Params{
Expand Down
4 changes: 2 additions & 2 deletions x/bank/simulation/genesis_test.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -42,8 +42,8 @@ func TestRandomizedGenState(t *testing.T) {
require.Len(t, bankGenesis.Params.GetSendEnabled(), 1)
require.Len(t, bankGenesis.Balances, 3)
require.Equal(t, "cosmos1ghekyjucln7y67ntx7cf27m9dpuxxemn4c8g4r", bankGenesis.Balances[2].GetAddress().String())
require.Equal(t, "1000stake", bankGenesis.Balances[2].GetCoins().String())
require.Equal(t, "6000stake", bankGenesis.Supply.String())
require.Equal(t, simulation.GenesisCoins("stake", 1000).String(), bankGenesis.Balances[2].GetCoins().String())
require.Equal(t, "6000stake,3000stake00,3000stake01,3000stake02,3000stake03", bankGenesis.Supply.String())
}

// TestRandomizedGenState tests abnormal scenarios of applying RandomizedGenState.
Expand Down