-
Notifications
You must be signed in to change notification settings - Fork 10
/
test_common.go
96 lines (84 loc) · 3.06 KB
/
test_common.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
package types
import (
"encoding/json"
"testing"
"github.com/cosmos/cosmos-sdk/types/address"
"github.com/cosmos/cosmos-sdk/client/tx"
"github.com/cosmos/cosmos-sdk/codec"
codectypes "github.com/cosmos/cosmos-sdk/codec/types"
"github.com/cosmos/cosmos-sdk/crypto/hd"
"github.com/cosmos/cosmos-sdk/crypto/keyring"
cryptotypes "github.com/cosmos/cosmos-sdk/crypto/types"
"github.com/cosmos/cosmos-sdk/server"
simappparams "github.com/cosmos/cosmos-sdk/simapp/params"
"github.com/cosmos/cosmos-sdk/std"
sdk "github.com/cosmos/cosmos-sdk/types"
authtx "github.com/cosmos/cosmos-sdk/x/auth/tx"
"github.com/cosmos/cosmos-sdk/x/genutil"
stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
"github.com/stretchr/testify/require"
cfg "github.com/tendermint/tendermint/config"
"github.com/tendermint/tendermint/libs/rand"
)
func MakeEncodingConfig(_ testing.TB) simappparams.EncodingConfig {
legacyAmino := codec.NewLegacyAmino()
interfaceRegistry := codectypes.NewInterfaceRegistry()
marshaler := codec.NewProtoCodec(interfaceRegistry)
std.RegisterInterfaces(interfaceRegistry)
std.RegisterLegacyAminoCodec(legacyAmino)
RegisterInterfaces(interfaceRegistry)
RegisterLegacyAminoCodec(legacyAmino)
return simappparams.EncodingConfig{
InterfaceRegistry: interfaceRegistry,
Marshaler: marshaler,
TxConfig: authtx.NewTxConfig(marshaler, authtx.DefaultSignModes),
Amino: legacyAmino,
}
}
func RandomAccAddress() sdk.AccAddress {
return rand.Bytes(address.Len)
}
// RandomGenTX returns a signed genesis tx
func RandomGenTX(t *testing.T, power uint32, mutators ...func(*MsgCreateValidator)) (json.RawMessage, sdk.AccAddress, cryptotypes.PubKey) {
t.Helper()
nodeConfig := cfg.TestConfig()
nodeConfig.RootDir = t.TempDir()
nodeConfig.NodeKey = "key.json"
_, valPubKey, err := genutil.InitializeNodeValidatorFiles(nodeConfig)
require.NoError(t, err)
//setup keyring
kb, err := keyring.New(sdk.KeyringServiceName(), keyring.BackendTest, t.TempDir(), nil)
require.NoError(t, err)
keyringAlgos, _ := kb.SupportedAlgorithms()
algo, err := keyring.NewSigningAlgoFromString(string(hd.Secp256k1Type), keyringAlgos)
require.NoError(t, err)
const myKey = "myKey"
addr, _, err := server.GenerateSaveCoinKey(kb, myKey, true, algo)
require.NoError(t, err)
// prepare genesis tx
valTokens := sdk.TokensFromConsensusPower(int64(power), sdk.DefaultPowerReduction)
createValMsg, err := NewMsgCreateValidator(
addr,
valPubKey,
sdk.NewCoin(DefaultBondDenom, valTokens),
stakingtypes.NewDescription("testing", "", "", "", ""),
)
require.NoError(t, err)
for _, m := range mutators {
m(createValMsg)
}
txConfig := MakeEncodingConfig(t).TxConfig
txBuilder := txConfig.NewTxBuilder()
err = txBuilder.SetMsgs(createValMsg)
require.NoError(t, err)
txFactory := tx.Factory{}
txFactory = txFactory.
WithChainID("").
WithKeybase(kb).
WithTxConfig(txConfig)
err = tx.Sign(txFactory, myKey, txBuilder, true)
require.NoError(t, err)
txBz, err := txConfig.TxJSONEncoder()(txBuilder.GetTx())
require.NoError(t, err)
return txBz, addr, valPubKey
}