Skip to content

Commit

Permalink
fix upgrade tests after client_v2_genesis
Browse files Browse the repository at this point in the history
  • Loading branch information
gjermundgaraba committed Feb 16, 2025
1 parent 2531648 commit 78e8058
Show file tree
Hide file tree
Showing 2 changed files with 41 additions and 0 deletions.
37 changes: 37 additions & 0 deletions e2e/testsuite/testconfig.go
Original file line number Diff line number Diff line change
Expand Up @@ -751,6 +751,14 @@ func defaultGovv1ModifyGenesis(version string) func(ibc.ChainConfig, []byte) ([]
appState[ibcexported.ModuleName] = ibcGenBz
}

if !testvalues.ClientV2FeatureReleases.IsSupported(version) {
ibcGenBz, err := modifyClientV2GenesisAppState(appState[ibcexported.ModuleName])
if err != nil {
return nil, err
}
appState[ibcexported.ModuleName] = ibcGenBz
}

appGenesis.AppState, err = json.Marshal(appState)
if err != nil {
return nil, err
Expand Down Expand Up @@ -863,6 +871,25 @@ func defaultGovv1Beta1ModifyGenesis(version string) func(ibc.ChainConfig, []byte
appStateMap[ibcexported.ModuleName] = ibcModuleGenesisMap
}

if !testvalues.ClientV2FeatureReleases.IsSupported(version) {
ibcModuleBytes, err := json.Marshal(appStateMap[ibcexported.ModuleName])
if err != nil {
return nil, fmt.Errorf("failed to extract ibc genesis bytes: %s", err)
}

ibcGenesisBytes, err := modifyClientV2GenesisAppState(ibcModuleBytes)
if err != nil {
return nil, err
}

ibcModuleGenesisMap := map[string]interface{}{}
err = json.Unmarshal(ibcGenesisBytes, &ibcModuleGenesisMap)
if err != nil {
return nil, fmt.Errorf("failed to unmarshal gov genesis bytes into map: %w", err)
}
appStateMap[ibcexported.ModuleName] = ibcModuleGenesisMap
}

appStateMap[govtypes.ModuleName] = govModuleGenesisMap
genesisDocMap[appStateKey] = appStateMap

Expand Down Expand Up @@ -975,3 +1002,13 @@ func modifyChannelV2GenesisAppState(ibcAppState []byte) ([]byte, error) {

return json.Marshal(ibcGenesisMap)
}

func modifyClientV2GenesisAppState(ibcAppState []byte) ([]byte, error) {
var ibcGenesisMap map[string]interface{}
if err := json.Unmarshal(ibcAppState, &ibcGenesisMap); err != nil {
return nil, err
}
delete(ibcGenesisMap, "client_v2_genesis")

return json.Marshal(ibcGenesisMap)
}
4 changes: 4 additions & 0 deletions e2e/testvalues/values.go
Original file line number Diff line number Diff line change
Expand Up @@ -135,3 +135,7 @@ var TransactionEventQueryFeatureReleases = semverutil.FeatureReleases{
var ChannelsV2FeatureReleases = semverutil.FeatureReleases{
MajorVersion: "v10",
}

var ClientV2FeatureReleases = semverutil.FeatureReleases{
MajorVersion: "v10",
}

0 comments on commit 78e8058

Please sign in to comment.