refactor ordering functions

This commit is contained in:
Jonathan Gimeno 2020-02-25 18:01:18 +01:00
parent 194b97b1c1
commit 660ffc8be5
1 changed files with 6 additions and 4 deletions

View File

@ -82,7 +82,8 @@ func SetupWithGenesisAccounts(genAccs []authexported.GenesisAccount, balances ..
type GenerateAccountStrategy func(int) []sdk.AccAddress
func Random(accNum int) []sdk.AccAddress {
// random is a strategy used by addTestAddrs() in order to generated addresses in random order.
func random(accNum int) []sdk.AccAddress {
testAddrs := make([]sdk.AccAddress, accNum)
for i := 0; i < accNum; i++ {
pk := ed25519.GenPrivKey().PubKey()
@ -92,7 +93,8 @@ func Random(accNum int) []sdk.AccAddress {
return testAddrs
}
func Incremental(accNum int) []sdk.AccAddress {
// incremental is a strategy used by addTestAddrs() in order to generated addresses in ascending order.
func incremental(accNum int) []sdk.AccAddress {
var addresses []sdk.AccAddress
var buffer bytes.Buffer
@ -114,13 +116,13 @@ func Incremental(accNum int) []sdk.AccAddress {
// AddTestAddrs constructs and returns accNum amount of accounts with an
// initial balance of accAmt in random order
func AddTestAddrs(app *SimApp, ctx sdk.Context, accNum int, accAmt sdk.Int) []sdk.AccAddress {
return addTestAddrs(app, ctx, accNum, accAmt, Random)
return addTestAddrs(app, ctx, accNum, accAmt, random)
}
// AddTestAddrs constructs and returns accNum amount of accounts with an
// initial balance of accAmt in random order
func AddTestAddrsIncremental(app *SimApp, ctx sdk.Context, accNum int, accAmt sdk.Int) []sdk.AccAddress {
return addTestAddrs(app, ctx, accNum, accAmt, Incremental)
return addTestAddrs(app, ctx, accNum, accAmt, incremental)
}
func addTestAddrs(app *SimApp, ctx sdk.Context, accNum int, accAmt sdk.Int, strategy GenerateAccountStrategy) []sdk.AccAddress {