tests

Author: pr0n00gler

app/app.go

@@ -13,6 +13,7 @@ import (
 	v700 "github.com/neutron-org/neutron/v8/app/upgrades/v7.0.0"
 	v800 "github.com/neutron-org/neutron/v8/app/upgrades/v8.0.0"
 	v800_rc0 "github.com/neutron-org/neutron/v8/app/upgrades/v8.0.0-rc0"
+	v810 "github.com/neutron-org/neutron/v8/app/upgrades/v8.1.0"
 	dynamicfeestypes "github.com/neutron-org/neutron/v8/x/dynamicfees/types"
 	stateverifier "github.com/neutron-org/neutron/v8/x/state-verifier"
 	svkeeper "github.com/neutron-org/neutron/v8/x/state-verifier/keeper"
@@ -239,6 +240,7 @@ var (
 		v700.Upgrade,
 		v800_rc0.Upgrade,
 		v800.Upgrade,
+		v810.Upgrade,
 	}
 
 	// DefaultNodeHome default home directories for the application daemon

app/upgrades/v8.1.0/constants.go

@@ -0,0 +1,18 @@
+package v810
+
+import (
+	storetypes "cosmossdk.io/store/types"
+
+	"github.com/neutron-org/neutron/v8/app/upgrades"
+)
+
+const (
+	// UpgradeName defines the on-chain upgrade name.
+	UpgradeName = "v8.1.0"
+)
+
+var Upgrade = upgrades.Upgrade{
+	UpgradeName:          UpgradeName,
+	CreateUpgradeHandler: CreateUpgradeHandler,
+	StoreUpgrades:        storetypes.StoreUpgrades{},
+}

app/upgrades/v8.1.0/upgrades.go

@@ -0,0 +1,35 @@
+package v810
+
+import (
+	"context"
+	"fmt"
+
+	upgradetypes "cosmossdk.io/x/upgrade/types"
+	"github.com/cosmos/cosmos-sdk/codec"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/cosmos/cosmos-sdk/types/module"
+
+	"github.com/neutron-org/neutron/v8/app/upgrades"
+)
+
+func CreateUpgradeHandler(
+	mm *module.Manager,
+	configurator module.Configurator,
+	keepers *upgrades.UpgradeKeepers,
+	_ upgrades.StoreKeys,
+	cdc codec.Codec,
+) upgradetypes.UpgradeHandler {
+	return func(c context.Context, _ upgradetypes.Plan, vm module.VersionMap) (module.VersionMap, error) {
+		ctx := sdk.UnwrapSDKContext(c)
+
+		ctx.Logger().Info("Starting module migrations...")
+
+		vm, err := mm.RunMigrations(ctx, configurator, vm)
+		if err != nil {
+			return vm, err
+		}
+
+		ctx.Logger().Info(fmt.Sprintf("Migration {%s} applied", UpgradeName))
+		return vm, nil
+	}
+}

app/upgrades/v8.1.0/upgrades_test.go

@@ -0,0 +1,37 @@
+package v810_test
+
+import (
+	"testing"
+
+	upgradetypes "cosmossdk.io/x/upgrade/types"
+	"github.com/stretchr/testify/require"
+	"github.com/stretchr/testify/suite"
+
+	v810 "github.com/neutron-org/neutron/v8/app/upgrades/v8.1.0"
+	"github.com/neutron-org/neutron/v8/testutil"
+)
+
+type UpgradeTestSuite struct {
+	testutil.IBCConnectionTestSuite
+}
+
+func TestKeeperTestSuite(t *testing.T) {
+	suite.Run(t, new(UpgradeTestSuite))
+}
+
+func (suite *UpgradeTestSuite) SetupTest() {
+	suite.IBCConnectionTestSuite.SetupTest()
+}
+
+func (suite *UpgradeTestSuite) TestUpgrade() {
+	app := suite.GetNeutronZoneApp(suite.ChainA)
+	ctx := suite.ChainA.GetContext().WithChainID("neutron-1")
+	t := suite.T()
+
+	upgrade := upgradetypes.Plan{
+		Name:   v810.UpgradeName,
+		Info:   "some text here",
+		Height: 100,
+	}
+	require.NoError(t, app.UpgradeKeeper.ApplyUpgrade(ctx, upgrade))
+}