chore: add upgrade handler v2 + reenable ante for MsgCreateValidator

app/app.go

@@ -898,6 +898,10 @@ func New(
 		panic(err)
 	}
 
+	// RegisterUpgradeHandlers is used for registering any on-chain upgrades.
+	// Make sure it's called after `app.ModuleManager` and `app.configurator` are set.
+	app.RegisterUpgradeHandlers()
+
 	autocliv1.RegisterQueryServer(app.GRPCQueryRouter(), runtimeservices.NewAutoCLIQueryService(app.mm.Modules))
 	reflectionSvc, err := runtimeservices.NewReflectionService()
 	if err != nil {
@@ -956,6 +960,29 @@ func (app *App) setAnteHandler(txConfig client.TxConfig) {
 	app.SetAnteHandler(anteHandler)
 }
 
+func (app *App) RegisterUpgradeHandlers() {
+	const UpgradeName = "v0.2.0"
+
+	app.UpgradeKeeper.SetUpgradeHandler(
+		UpgradeName,
+		func(ctx context.Context, _ upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) {
+			return app.ModuleManager().RunMigrations(ctx, app.Configurator(), fromVM)
+		},
+	)
+
+	upgradeInfo, err := app.UpgradeKeeper.ReadUpgradeInfoFromDisk()
+	if err != nil {
+		panic(err)
+	}
+
+	if upgradeInfo.Name == UpgradeName && !app.UpgradeKeeper.IsSkipHeight(upgradeInfo.Height) {
+		storeUpgrades := storetypes.StoreUpgrades{}
+
+		// configure store loader that checks if version == upgradeHeight and applies store upgrades
+		app.SetStoreLoader(upgradetypes.UpgradeStoreLoader(upgradeInfo.Height, &storeUpgrades))
+	}
+}
+
 // Name returns the name of the App
 func (app *App) Name() string { return app.BaseApp.Name() }
 

x/reporter/ante/ante.go

@@ -31,8 +31,8 @@ func (t TrackStakeChangesDecorator) AnteHandle(ctx sdk.Context, tx sdk.Tx, simul
 	var msgAmount math.Int
 	for _, msg := range tx.GetMsgs() {
 		switch msg := msg.(type) {
-		// case *stakingtypes.MsgCreateValidator:
-		// 	msgAmount = msg.Value.Amount
+		case *stakingtypes.MsgCreateValidator:
+			msgAmount = msg.Value.Amount
 		case *stakingtypes.MsgDelegate:
 			msgAmount = msg.Amount.Amount
 		case *stakingtypes.MsgBeginRedelegate:

x/reporter/ante/ante_test.go

@@ -31,20 +31,20 @@ func TestNewTrackStakeChangesDecorator(t *testing.T) {
 		msg  sdk.Msg
 		err  error
 	}{
-		// {
-		// 	name: "CreateValidator",
-		// 	msg: &stakingtypes.MsgCreateValidator{
-		// 		Value: sdk.Coin{Denom: "loya", Amount: math.NewInt(1)},
-		// 	},
-		// 	err: nil,
-		// },
-		// {
-		// 	name: "CreateValidator",
-		// 	msg: &stakingtypes.MsgCreateValidator{
-		// 		Value: sdk.Coin{Denom: "loya", Amount: math.NewInt(100)},
-		// 	},
-		// 	err: errors.New("amount increases total stake by more than the allowed 5% in a twelve hour period"),
-		// },
+		{
+			name: "CreateValidator",
+			msg: &stakingtypes.MsgCreateValidator{
+				Value: sdk.Coin{Denom: "loya", Amount: math.NewInt(1)},
+			},
+			err: nil,
+		},
+		{
+			name: "CreateValidator",
+			msg: &stakingtypes.MsgCreateValidator{
+				Value: sdk.Coin{Denom: "loya", Amount: math.NewInt(100)},
+			},
+			err: errors.New("amount increases total stake by more than the allowed 5% in a twelve hour period"),
+		},
 		{
 			name: "Delegate",
 			msg: &stakingtypes.MsgDelegate{