Fix tests and create migration logic for compute

scrtlabs · Dec 4, 2024 · 5a82dc2 · 5a82dc2
1 parent 8132332
commit 5a82dc2
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 1 deletion.
diff --git a/x/compute/internal/keeper/migrations.go b/x/compute/internal/keeper/migrations.go
@@ -135,6 +135,18 @@ func (m Migrator) Migrate4to5(ctx sdk.Context) error {
 	return nil
 }
 
+func (m Migrator) Migrate5to6(ctx sdk.Context) error {
+	store := m.keeper.storeService.OpenKVStore(ctx)
+	defaultParams := types.DefaultParams()
+	bz, err := m.keeper.cdc.Marshal(&defaultParams)
+	if err != nil {
+		return err
+	}
+	store.Set(types.ParamsKey, bz)
+
+	return nil
+}
+
 const progressPartSize = 1000
 
 func logMigrationProgress(ctx sdk.Context, formatter *message.Printer, migratedContracts uint64, totalContracts uint64, previousTime int64) {

diff --git a/x/compute/internal/keeper/test_common.go b/x/compute/internal/keeper/test_common.go
@@ -607,6 +607,7 @@ func CreateTestInput(t *testing.T, isCheckTx bool, supportedFeatures string, enc
 	random := make([]byte, 32)
 	_, _ = rand.Read(random)
 	keeper.SetRandomSeed(ctx, random)
+	keeper.SetParams(ctx, wasmtypes.DefaultParams())
 
 	govSubSp, _ := paramsKeeper.GetSubspace(govtypes.ModuleName)
 

diff --git a/x/compute/module.go b/x/compute/module.go
@@ -93,7 +93,7 @@ func NewAppModule(keeper Keeper) AppModule {
 }
 
 // ConsensusVersion implements AppModule/ConsensusVersion.
-func (AppModule) ConsensusVersion() uint64 { return 5 }
+func (AppModule) ConsensusVersion() uint64 { return 6 }
 
 func (am AppModule) RegisterServices(configurator module.Configurator) {
 	types.RegisterMsgServer(configurator.MsgServer(), keeper.NewMsgServerImpl(am.keeper))
@@ -118,6 +118,10 @@ func (am AppModule) RegisterServices(configurator module.Configurator) {
 	if err != nil {
 		panic(err)
 	}
+	err = configurator.RegisterMigration(types.ModuleName, 5, m.Migrate5to6)
+	if err != nil {
+		panic(err)
+	}
 }
 
 // InitGenesis performs genesis initialization for the compute module. It returns