refactor: expsoe SetAppVersion on the upgrade keeper to allow overwri…

…ting during the upgrade (#244)
cosmos · May 24, 2022 · 9a05fa0 · 9a05fa0
1 parent a68d28e
commit 9a05fa0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 8 deletions.
diff --git a/x/upgrade/keeper/keeper.go b/x/upgrade/keeper/keeper.go
@@ -83,6 +83,11 @@ func (k Keeper) SetModuleVersionMap(ctx sdk.Context, vm module.VersionMap) {
 	}
 }
 
+// SetAppVersion sets app version to version
+func (k Keeper) SetAppVersion(ctx sdk.Context, version uint64) error {
+	return k.versionManager.SetAppVersion(ctx, version)
+}
+
 // GetModuleVersionMap returns a map of key module name and value module consensus version
 // as defined in ADR-041.
 func (k Keeper) GetModuleVersionMap(ctx sdk.Context) module.VersionMap {

diff --git a/x/upgrade/keeper/util_test.go b/x/upgrade/keeper/util_test.go