Add upgrade handler for v4.1.0 (#1374)

* Add upgrade handler for v4.1.0

* update upgrade to version

* update package name

protocol/app/upgrades.go

@@ -6,30 +6,29 @@ import (
 	upgradetypes "cosmossdk.io/x/upgrade/types"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/dydxprotocol/v4-chain/protocol/app/upgrades"
-	v4_0_0 "github.com/dydxprotocol/v4-chain/protocol/app/upgrades/v4.0.0"
+	v4_1_0 "github.com/dydxprotocol/v4-chain/protocol/app/upgrades/v4.1.0"
 )
 
 var (
 	// `Upgrades` defines the upgrade handlers and store loaders for the application.
 	// New upgrades should be added to this slice after they are implemented.
 	Upgrades = []upgrades.Upgrade{
-		v4_0_0.Upgrade,
+		v4_1_0.Upgrade,
 	}
 	Forks = []upgrades.Fork{}
 )
 
 // setupUpgradeHandlers registers the upgrade handlers to perform custom upgrade
 // logic and state migrations for software upgrades.
 func (app *App) setupUpgradeHandlers() {
-	if app.UpgradeKeeper.HasHandler(v4_0_0.UpgradeName) {
-		panic(fmt.Sprintf("Cannot register duplicate upgrade handler '%s'", v4_0_0.UpgradeName))
+	if app.UpgradeKeeper.HasHandler(v4_1_0.UpgradeName) {
+		panic(fmt.Sprintf("Cannot register duplicate upgrade handler '%s'", v4_1_0.UpgradeName))
 	}
 	app.UpgradeKeeper.SetUpgradeHandler(
-		v4_0_0.UpgradeName,
-		v4_0_0.CreateUpgradeHandler(
+		v4_1_0.UpgradeName,
+		v4_1_0.CreateUpgradeHandler(
 			app.ModuleManager,
 			app.configurator,
-			app.RatelimitKeeper,
 		),
 	)
 }

protocol/app/upgrades/v4.1.0/constants.go

@@ -0,0 +1,13 @@
+package v_4_1_0
+
+import (
+	"github.com/dydxprotocol/v4-chain/protocol/app/upgrades"
+)
+
+const (
+	UpgradeName = "v4.1.0"
+)
+
+var Upgrade = upgrades.Upgrade{
+	UpgradeName: UpgradeName,
+}

protocol/app/upgrades/v4.1.0/upgrade.go

@@ -0,0 +1,21 @@
+package v_4_1_0
+
+import (
+	"context"
+	"fmt"
+
+	upgradetypes "cosmossdk.io/x/upgrade/types"
+	"github.com/cosmos/cosmos-sdk/types/module"
+	"github.com/dydxprotocol/v4-chain/protocol/lib"
+)
+
+func CreateUpgradeHandler(
+	mm *module.Manager,
+	configurator module.Configurator,
+) upgradetypes.UpgradeHandler {
+	return func(ctx context.Context, plan upgradetypes.Plan, vm module.VersionMap) (module.VersionMap, error) {
+		sdkCtx := lib.UnwrapSDKContext(ctx, "app/upgrades")
+		sdkCtx.Logger().Info(fmt.Sprintf("Running %s Upgrade...", UpgradeName))
+		return mm.RunMigrations(ctx, configurator, vm)
+	}
+}

protocol/testing/containertest/testnet.go

@@ -25,7 +25,7 @@ const persistentPeers = "17e5e45691f0d01449c84fd4ae87279578cdd7ec@testnet-local-
 const resourceLifetimeSecs = 600
 
 // The version of that we're upgrading to (aka the current commit)
-const UpgradeToVersion = "v4.0.0"
+const UpgradeToVersion = "v4.1.0"
 
 func monikers() map[string]string {
 	return map[string]string{