certusone
/
cosmos-sdk
mirror of https://github.com/certusone/cosmos-sdk.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
simapp/test_helpers.go
View File

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