From 21c411cdcd7a9754f11984346f364a2cce892290 Mon Sep 17 00:00:00 2001
From: pr0n00gler <programmer10110@gmail.com>
Date: Fri, 7 Feb 2025 11:39:04 +0200
Subject: [PATCH] fix upgrade handler

---
 app/upgrades/v5.1.0/constants.go     |  3 ++-
 app/upgrades/v5.1.0/upgrades.go      | 19 ++++++++++++++++++-
 app/upgrades/v5.1.0/upgrades_test.go |  8 +++++++-
 3 files changed, 27 insertions(+), 3 deletions(-)

diff --git a/app/upgrades/v5.1.0/constants.go b/app/upgrades/v5.1.0/constants.go
index 15f18d2c2..1eaa3f6c8 100644
--- a/app/upgrades/v5.1.0/constants.go
+++ b/app/upgrades/v5.1.0/constants.go
@@ -8,7 +8,8 @@ import (
 
 const (
 	// UpgradeName defines the on-chain upgrade name.
-	UpgradeName = "v5.1.0"
+	UpgradeName                = "v5.1.0"
+	MarketMapAuthorityMultisig = "neutron1ldvrvhyvtssm0ptdq23hhaltprx8ctmjh92kzfs55sz2z997n76s72wr86"
 )
 
 var Upgrade = upgrades.Upgrade{
diff --git a/app/upgrades/v5.1.0/upgrades.go b/app/upgrades/v5.1.0/upgrades.go
index 34618c047..44666882f 100644
--- a/app/upgrades/v5.1.0/upgrades.go
+++ b/app/upgrades/v5.1.0/upgrades.go
@@ -5,9 +5,13 @@ import (
 	"fmt"
 
 	upgradetypes "cosmossdk.io/x/upgrade/types"
+	adminmoduletypes "github.com/cosmos/admin-module/v2/x/adminmodule/types"
 	"github.com/cosmos/cosmos-sdk/codec"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/cosmos/cosmos-sdk/types/module"
+	authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
+	marketmapkeeper "github.com/skip-mev/slinky/x/marketmap/keeper"
+	marketmaptypes "github.com/skip-mev/slinky/x/marketmap/types"
 
 	"github.com/neutron-org/neutron/v5/app/upgrades"
 )
@@ -15,7 +19,7 @@ import (
 func CreateUpgradeHandler(
 	mm *module.Manager,
 	configurator module.Configurator,
-	_ *upgrades.UpgradeKeepers,
+	keepers *upgrades.UpgradeKeepers,
 	_ upgrades.StoreKeys,
 	_ codec.Codec,
 ) upgradetypes.UpgradeHandler {
@@ -29,7 +33,20 @@ func CreateUpgradeHandler(
 			return vm, err
 		}
 
+		err = setMarketMapParams(ctx, keepers.MarketmapKeeper)
+		if err != nil {
+			return nil, err
+		}
+
 		ctx.Logger().Info(fmt.Sprintf("Migration {%s} applied", UpgradeName))
 		return vm, nil
 	}
 }
+
+func setMarketMapParams(ctx sdk.Context, marketmapKeeper *marketmapkeeper.Keeper) error {
+	marketmapParams := marketmaptypes.Params{
+		MarketAuthorities: []string{authtypes.NewModuleAddress(adminmoduletypes.ModuleName).String(), MarketMapAuthorityMultisig},
+		Admin:             authtypes.NewModuleAddress(adminmoduletypes.ModuleName).String(),
+	}
+	return marketmapKeeper.SetParams(ctx, marketmapParams)
+}
diff --git a/app/upgrades/v5.1.0/upgrades_test.go b/app/upgrades/v5.1.0/upgrades_test.go
index c13fcc442..d711f2c33 100644
--- a/app/upgrades/v5.1.0/upgrades_test.go
+++ b/app/upgrades/v5.1.0/upgrades_test.go
@@ -23,7 +23,7 @@ func (suite *UpgradeTestSuite) SetupTest() {
 	suite.IBCConnectionTestSuite.SetupTest()
 }
 
-func (suite *UpgradeTestSuite) TestOracleUpgrade() {
+func (suite *UpgradeTestSuite) TestUpgrade() {
 	app := suite.GetNeutronZoneApp(suite.ChainA)
 	ctx := suite.ChainA.GetContext().WithChainID("neutron-1")
 	t := suite.T()
@@ -34,4 +34,10 @@ func (suite *UpgradeTestSuite) TestOracleUpgrade() {
 		Height: 100,
 	}
 	require.NoError(t, app.UpgradeKeeper.ApplyUpgrade(ctx, upgrade))
+
+	params, err := app.MarketMapKeeper.GetParams(ctx)
+	suite.Require().NoError(err)
+	suite.Require().Equal(params.MarketAuthorities[0], "neutron1hxskfdxpp5hqgtjj6am6nkjefhfzj359x0ar3z")
+	suite.Require().Equal(params.MarketAuthorities[1], v510.MarketMapAuthorityMultisig)
+	suite.Require().Equal(params.Admin, "neutron1hxskfdxpp5hqgtjj6am6nkjefhfzj359x0ar3z")
 }