From fd976422777b74010485ef980ef77e940f32d415 Mon Sep 17 00:00:00 2001
From: Shrenuj Bansal <shrenuj@dydx.exchange>
Date: Tue, 26 Nov 2024 14:05:37 -0500
Subject: [PATCH] add files

Signed-off-by: Shrenuj Bansal <shrenuj@dydx.exchange>
---
 protocol/app/upgrades/v8.0/constants.go       | 20 ++++
 .../upgrades/v8.0/upgrade_container_test.go   | 93 +++++++++++++++++++
 2 files changed, 113 insertions(+)
 create mode 100644 protocol/app/upgrades/v8.0/constants.go
 create mode 100644 protocol/app/upgrades/v8.0/upgrade_container_test.go

diff --git a/protocol/app/upgrades/v8.0/constants.go b/protocol/app/upgrades/v8.0/constants.go
new file mode 100644
index 0000000000..644d31c87c
--- /dev/null
+++ b/protocol/app/upgrades/v8.0/constants.go
@@ -0,0 +1,20 @@
+package v_8_0
+
+import (
+	store "cosmossdk.io/store/types"
+	"github.com/dydxprotocol/v4-chain/protocol/app/upgrades"
+	affiliatetypes "github.com/dydxprotocol/v4-chain/protocol/x/affiliates/types"
+)
+
+const (
+	UpgradeName = "v8.0"
+)
+
+var Upgrade = upgrades.Upgrade{
+	UpgradeName: UpgradeName,
+	StoreUpgrades: store.StoreUpgrades{
+		Added: []string{
+			affiliatetypes.StoreKey,
+		},
+	},
+}
diff --git a/protocol/app/upgrades/v8.0/upgrade_container_test.go b/protocol/app/upgrades/v8.0/upgrade_container_test.go
new file mode 100644
index 0000000000..5cb69acc65
--- /dev/null
+++ b/protocol/app/upgrades/v8.0/upgrade_container_test.go
@@ -0,0 +1,93 @@
+package v_8_0_test
+
+import (
+	"testing"
+
+	"github.com/cosmos/gogoproto/proto"
+
+	v_7_1 "github.com/dydxprotocol/v4-chain/protocol/app/upgrades/v7.1"
+	clobtypes "github.com/dydxprotocol/v4-chain/protocol/x/clob/types"
+
+	perptypes "github.com/dydxprotocol/v4-chain/protocol/x/perpetuals/types"
+	pricetypes "github.com/dydxprotocol/v4-chain/protocol/x/prices/types"
+
+	"github.com/dydxprotocol/v4-chain/protocol/testing/containertest"
+	"github.com/dydxprotocol/v4-chain/protocol/testutil/constants"
+	"github.com/stretchr/testify/require"
+)
+
+func TestStateUpgrade(t *testing.T) {
+	testnet, err := containertest.NewTestnetWithPreupgradeGenesis()
+	require.NoError(t, err, "failed to create testnet - is docker daemon running?")
+	err = testnet.Start()
+	require.NoError(t, err)
+	defer testnet.MustCleanUp()
+	node := testnet.Nodes["alice"]
+	nodeAddress := constants.AliceAccAddress.String()
+
+	preUpgradeSetups(node, t)
+	preUpgradeChecks(node, t)
+
+	err = containertest.UpgradeTestnet(nodeAddress, t, node, UpgradeName)
+	require.NoError(t, err)
+
+	postUpgradeChecks(node, t)
+}
+
+func preUpgradeSetups(node *containertest.Node, t *testing.T) {}
+
+func preUpgradeChecks(node *containertest.Node, t *testing.T) {
+	// Add test for your upgrade handler logic below
+}
+
+func postUpgradeChecks(node *containertest.Node, t *testing.T) {
+	// Check that the listing module state has been initialized with the hard cap and default deposit params.
+	postUpgradeMarketIdsCheck(node, t)
+}
+
+func postUpgradeMarketIdsCheck(node *containertest.Node, t *testing.T) {
+	// query the next market id
+	resp, err := containertest.Query(
+		node,
+		pricetypes.NewQueryClient,
+		pricetypes.QueryClient.NextMarketId,
+		&pricetypes.QueryNextMarketIdRequest{},
+	)
+	require.NoError(t, err)
+	require.NotNil(t, resp)
+
+	nextMarketIdResp := pricetypes.QueryNextMarketIdResponse{}
+	err = proto.UnmarshalText(resp.String(), &nextMarketIdResp)
+	require.NoError(t, err)
+	require.Equal(t, uint32(v_7_1.ID_NUM), nextMarketIdResp.NextMarketId)
+
+	// query the next perpetual id
+	resp, err = containertest.Query(
+		node,
+		perptypes.NewQueryClient,
+		perptypes.QueryClient.NextPerpetualId,
+		&perptypes.QueryNextPerpetualIdRequest{},
+	)
+	require.NoError(t, err)
+	require.NotNil(t, resp)
+
+	nextPerpIdResp := perptypes.QueryNextPerpetualIdResponse{}
+	err = proto.UnmarshalText(resp.String(), &nextPerpIdResp)
+	require.NoError(t, err)
+	require.Equal(t, uint32(v_7_1.ID_NUM), nextPerpIdResp.NextPerpetualId)
+
+	// query the next clob pair id
+	resp, err = containertest.Query(
+		node,
+		clobtypes.NewQueryClient,
+		clobtypes.QueryClient.NextClobPairId,
+		&clobtypes.QueryNextClobPairIdRequest{},
+	)
+	require.NoError(t, err)
+	require.NotNil(t, resp)
+
+	nextClobPairIdResp := clobtypes.QueryNextClobPairIdResponse{}
+	err = proto.UnmarshalText(resp.String(), &nextClobPairIdResp)
+	require.NoError(t, err)
+	require.Equal(t, uint32(v_7_1.ID_NUM), nextClobPairIdResp.NextClobPairId)
+}