feat(sui): add support for contract upgrades (#480)

## Description

Add support for upgrading Sui contracts (and the corresponding e2e
tests).

---------

Co-authored-by: RodrigoAD <15104916+RodrigoAD@users.noreply.github.com>

Author: JohnChangUK

.changeset/sweet-steaks-study.md

@@ -0,0 +1,5 @@
+---
+"@smartcontractkit/mcms": minor
+---
+
+add support for Sui contract upgrades operations

.mockery.yaml

@@ -128,6 +128,19 @@ packages:
         config:
           dir: "./sdk/sui/mocks/feequoter"
           filename: "feequoterencoder.go"
+  github.com/smartcontractkit/chainlink-sui/bindings/generated/mcms/mcms_deployer:
+    config:
+      all: false
+      outpkg: "mock_module_mcmsdeployer"
+    interfaces:
+      IMcmsDeployer:
+        config:
+          dir: "./sdk/sui/mocks/mcmsdeployer"
+          filename: "imcmsdeployer.go"
+      McmsDeployerEncoder:
+        config:
+          dir: "./sdk/sui/mocks/mcmsdeployer"
+          filename: "mcmsdeployerencoder.go"
 
 # Required to fix the following deprecation warning:
 # https://vektra.github.io/mockery/v2.48/deprecations/#issue-845-fix

e2e/tests/runner_test.go

@@ -38,4 +38,5 @@ func TestSuiSuite(t *testing.T) {
 	suite.Run(t, new(suie2e.SetRootTestSuite))
 	suite.Run(t, new(suie2e.MCMSUserTestSuite))
 	suite.Run(t, new(suie2e.TimelockCancelProposalTestSuite))
+	suite.Run(t, new(suie2e.MCMSUserUpgradeTestSuite))
 }

e2e/tests/setup.go

@@ -41,7 +41,8 @@ type Config struct {
 	SuiChain    *blockchain.Input `toml:"sui_config"`
 
 	Settings struct {
-		PrivateKeys []string `toml:"private_keys"`
+		PrivateKeys     []string `toml:"private_keys"`
+		LocalSuiNodeURL string   `toml:"local_sui_node_url,omitempty"`
 	} `toml:"settings"`
 }
 
@@ -180,7 +181,12 @@ func InitializeSharedTestSetup(t *testing.T) *TestSetup {
 			suiClient           sui.ISuiAPI
 			suiBlockchainOutput *blockchain.Output
 		)
-		if in.SuiChain != nil {
+		if in.Settings.LocalSuiNodeURL != "" {
+			// Connect to local Sui node (highest priority)
+			suiClient = sui.NewSuiClient(in.Settings.LocalSuiNodeURL)
+			t.Logf("Connected to local Sui node @ %s", in.Settings.LocalSuiNodeURL)
+		} else if in.SuiChain != nil {
+			// Use blockchain network setup (fallback)
 			ports := freeport.GetN(t, 2)
 			port := ports[0]
 			faucetPort := ports[1]

e2e/tests/sui/common.go

@@ -48,9 +48,10 @@ type SuiTestSuite struct {
 	mcmsAccount modulemcmsaccount.IMcmsAccount
 
 	// MCMS User
-	mcmsUserPackageId   string
-	mcmsUser            modulemcmsuser.IMcmsUser
-	mcmsUserOwnerCapObj string
+	mcmsUserPackageID     string
+	mcmsUser              modulemcmsuser.IMcmsUser
+	mcmsUserOwnerCapObj   string
+	mcmsUserUpgradeCapObj string
 
 	// State Object passed into `mcms_entrypoint`
 	stateObj string
@@ -131,16 +132,19 @@ func (s *SuiTestSuite) DeployMCMSUserContract() {
 	}, s.client, s.mcmsPackageID, signerAddress)
 	s.Require().NoError(err, "Failed to publish MCMS user package")
 
-	s.mcmsUserPackageId = mcmsUserPackage.Address()
+	s.mcmsUserPackageID = mcmsUserPackage.Address()
 	s.mcmsUser = mcmsUserPackage.MCMSUser()
 
 	userDataObj, err := bind.FindObjectIdFromPublishTx(*tx, "mcms_user", "UserData")
 	s.Require().NoError(err, "Failed to find object IDs in publish tx")
-	mcmsUserOwnerCapObj, err := bind.FindObjectIdFromPublishTx(*tx, "mcms_user", "OwnerCap")
+	mcmsUserOwnerCapObj, err := bind.FindObjectIdFromPublishTx(*tx, "ownable", "OwnerCap")
+	s.Require().NoError(err, "Failed to find object IDs in publish tx")
+	mcmsUserUpgradeCapObj, err := bind.FindObjectIdFromPublishTx(*tx, "package", "UpgradeCap")
 	s.Require().NoError(err, "Failed to find object IDs in publish tx")
 
 	s.mcmsUserOwnerCapObj = mcmsUserOwnerCapObj
 	s.stateObj = userDataObj
+	s.mcmsUserUpgradeCapObj = mcmsUserUpgradeCapObj
 
 	// For executing, We need to register OwnerCap with MCMS
 	{
@@ -181,8 +185,8 @@ type TestEntrypointArgEncoder struct {
 }
 
 func (e *TestEntrypointArgEncoder) EncodeEntryPointArg(executingCallbackParams *transaction.Argument, target, module, function, stateObjID string, data []byte) (*bind.EncodedCall, error) {
-	// For simplicity, we only support MCMSUser as target in this test encoder
-	if module != "MCMSUser" {
+	// For simplicity, we only support mcms_user as target in this test encoder
+	if module != "mcms_user" {
 		return nil, fmt.Errorf("unsupported module: %s", module)
 	}
 	mcmsUser, err := mcmsuser.NewMCMSUser(target, e.client)