fix based on PR comments

This commit is contained in:
Jonathan Gimeno
2020-02-26 16:05:12 +01:00
parent 1d18f8839c
commit c94a98f459
3 changed files with 22 additions and 18 deletions
+17 -11
View File
@@ -3,6 +3,7 @@ package simapp
import (
"bytes"
"encoding/hex"
"fmt"
"strconv"
"testing"
@@ -106,7 +107,9 @@ func incremental(accNum int) []sdk.AccAddress {
buffer.WriteString(numString) //adding on final two digits to make addresses unique
res, _ := sdk.AccAddressFromHex(buffer.String())
bech := res.String()
addresses = append(addresses, TestAddr(buffer.String(), bech))
addr, _ := TestAddr(buffer.String(), bech)
addresses = append(addresses, addr)
buffer.Reset()
}
@@ -146,33 +149,36 @@ func addTestAddrs(app *SimApp, ctx sdk.Context, accNum int, accAmt sdk.Int, stra
return testAddrs
}
func ConvertAddrsToValAddrs(addrs []sdk.AccAddress) (valAddrs []sdk.ValAddress) {
//ConvertAddrsToValAddrs converts the provided addresses to
func ConvertAddrsToValAddrs(addrs []sdk.AccAddress) []sdk.ValAddress {
var valAddrs []sdk.ValAddress
for _, addr := range addrs {
valAddrs = append(valAddrs, sdk.ValAddress(addr))
}
return
return valAddrs
}
func TestAddr(addr string, bech string) sdk.AccAddress {
func TestAddr(addr string, bech string) (sdk.AccAddress, error) {
res, err := sdk.AccAddressFromHex(addr)
if err != nil {
panic(err)
return nil, err
}
bechexpected := res.String()
if bech != bechexpected {
panic("Bech encoding doesn't match reference")
return nil, fmt.Errorf("bech encoding doesn't match reference")
}
bechres, err := sdk.AccAddressFromBech32(bech)
if err != nil {
panic(err)
return nil, err
}
if !bytes.Equal(bechres, res) {
panic("Bech decode and hex decode don't match")
return nil, err
}
return res
return res, nil
}
// CheckBalance checks the balance of an account.
@@ -268,14 +274,14 @@ func CreateTestPubKeys(numPubKeys int) []crypto.PubKey {
numString := strconv.Itoa(i)
buffer.WriteString("0B485CFC0EECC619440448436F8FC9DF40566F2369E72400281454CB552AF") //base pubkey string
buffer.WriteString(numString) //adding on final two digits to make pubkeys unique
publicKeys = append(publicKeys, NewPubKey(buffer.String()))
publicKeys = append(publicKeys, NewPubKeyFromHex(buffer.String()))
buffer.Reset()
}
return publicKeys
}
func NewPubKey(pk string) (res crypto.PubKey) {
func NewPubKeyFromHex(pk string) (res crypto.PubKey) {
pkBytes, err := hex.DecodeString(pk)
if err != nil {
panic(err)
+1 -2
View File
@@ -5,11 +5,10 @@ import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/tendermint/tendermint/crypto/ed25519"
"github.com/stretchr/testify/require"
abci "github.com/tendermint/tendermint/abci/types"
"github.com/tendermint/tendermint/crypto/ed25519"
"github.com/cosmos/cosmos-sdk/simapp"
sdk "github.com/cosmos/cosmos-sdk/types"
+4 -5
View File
@@ -5,15 +5,14 @@ import (
"testing"
"time"
gogotypes "github.com/gogo/protobuf/types"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
abci "github.com/tendermint/tendermint/abci/types"
"github.com/tendermint/tendermint/crypto/secp256k1"
tmtypes "github.com/tendermint/tendermint/types"
"github.com/stretchr/testify/assert"
gogotypes "github.com/gogo/protobuf/types"
"github.com/stretchr/testify/require"
"github.com/cosmos/cosmos-sdk/simapp"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/cosmos/cosmos-sdk/x/staking"