/
test_helpers.go
91 lines (75 loc) · 3.64 KB
/
test_helpers.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
package consumer
import (
"encoding/json"
"time"
cryptocodec "github.com/cosmos/cosmos-sdk/crypto/codec"
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
"github.com/cosmos/cosmos-sdk/x/genutil"
genutiltypes "github.com/cosmos/cosmos-sdk/x/genutil/types"
ibcclienttypes "github.com/cosmos/ibc-go/v4/modules/core/02-client/types"
ibccommitmenttypes "github.com/cosmos/ibc-go/v4/modules/core/23-commitment/types"
ibctmtypes "github.com/cosmos/ibc-go/v4/modules/light-clients/07-tendermint/types"
ccvconsumertypes "github.com/cosmos/interchain-security/x/ccv/consumer/types"
ccvprovidertypes "github.com/cosmos/interchain-security/x/ccv/provider/types"
"github.com/cosmos/interchain-security/x/ccv/types"
types1 "github.com/tendermint/tendermint/abci/types"
tmtypes "github.com/tendermint/tendermint/types"
"github.com/cosmos/cosmos-sdk/testutil/network"
"github.com/merlin-network/petri/app"
)
// This function creates consumer module genesis state that is used as starting point for modifications
// that allow petri chain to be started locally without having to start the provider chain and the relayer.
// It is also used in tests that are starting the chain node.
func CreateMinimalConsumerTestGenesis() *ccvconsumertypes.GenesisState {
genesisState := ccvconsumertypes.DefaultGenesisState()
genesisState.Params.Enabled = true
genesisState.NewChain = true
genesisState.ProviderClientState = ccvprovidertypes.DefaultParams().TemplateClient
genesisState.ProviderClientState.ChainId = app.Name
genesisState.ProviderClientState.LatestHeight = ibcclienttypes.Height{RevisionNumber: 0, RevisionHeight: 1}
genesisState.ProviderClientState.TrustingPeriod, _ = types.CalculateTrustPeriod(genesisState.Params.UnbondingPeriod, ccvprovidertypes.DefaultTrustingPeriodFraction)
genesisState.ProviderClientState.UnbondingPeriod = genesisState.Params.UnbondingPeriod
genesisState.ProviderClientState.MaxClockDrift = ccvprovidertypes.DefaultMaxClockDrift
genesisState.ProviderConsensusState = &ibctmtypes.ConsensusState{
Timestamp: time.Now().UTC(),
Root: ibccommitmenttypes.MerkleRoot{Hash: []byte("dummy")},
}
return genesisState
}
func ModifyConsumerGenesis(val network.Validator) error {
genFile := val.Ctx.Config.GenesisFile()
appState, genDoc, err := genutiltypes.GenesisStateFromGenFile(genFile)
if err != nil {
return sdkerrors.Wrap(err, "failed to read genesis from the file")
}
tmProtoPublicKey, err := cryptocodec.ToTmProtoPublicKey(val.PubKey)
if err != nil {
return sdkerrors.Wrap(err, "invalid public key")
}
initialValset := []types1.ValidatorUpdate{{PubKey: tmProtoPublicKey, Power: 100}}
vals, err := tmtypes.PB2TM.ValidatorUpdates(initialValset)
if err != nil {
return sdkerrors.Wrap(err, "could not convert val updates to validator set")
}
consumerGenesisState := CreateMinimalConsumerTestGenesis()
consumerGenesisState.InitialValSet = initialValset
consumerGenesisState.ProviderConsensusState.NextValidatorsHash = tmtypes.NewValidatorSet(vals).Hash()
if err := consumerGenesisState.Validate(); err != nil {
return sdkerrors.Wrap(err, "invalid consumer genesis")
}
consumerGenStateBz, err := val.ClientCtx.Codec.MarshalJSON(consumerGenesisState)
if err != nil {
return sdkerrors.Wrap(err, "failed to marshal consumer genesis state into JSON")
}
appState[ccvconsumertypes.ModuleName] = consumerGenStateBz
appStateJSON, err := json.Marshal(appState)
if err != nil {
return sdkerrors.Wrap(err, "failed to marshal application genesis state into JSON")
}
genDoc.AppState = appStateJSON
err = genutil.ExportGenesisFile(genDoc, genFile)
if err != nil {
return sdkerrors.Wrap(err, "failed to export genesis state")
}
return nil
}