From b56d9bbd7a3976a988ff88632511f88144554a57 Mon Sep 17 00:00:00 2001
From: lumtis <lucas.bertrand.22@gmail.com>
Date: Wed, 10 Apr 2024 16:40:26 +0200
Subject: [PATCH] e2e test

---
 e2e/e2etests/e2etests.go                      |  6 ++
 .../test_update_bytecode_connector.go         | 59 +++++++++++++++++++
 2 files changed, 65 insertions(+)
 create mode 100644 e2e/e2etests/test_update_bytecode_connector.go

diff --git a/e2e/e2etests/e2etests.go b/e2e/e2etests/e2etests.go
index a54a9a6490..765194c425 100644
--- a/e2e/e2etests/e2etests.go
+++ b/e2e/e2etests/e2etests.go
@@ -366,4 +366,10 @@ var AllE2ETests = []runner.E2ETest{
 		[]runner.ArgDefinition{},
 		TestMigrateChainSupport,
 	),
+	runner.NewE2ETest(
+		TestUpdateBytecodeConnectorName,
+		"update zevm connector bytecode",
+		[]runner.ArgDefinition{},
+		TestUpdateBytecodeConnector,
+	),
 }
diff --git a/e2e/e2etests/test_update_bytecode_connector.go b/e2e/e2etests/test_update_bytecode_connector.go
new file mode 100644
index 0000000000..a6e719948e
--- /dev/null
+++ b/e2e/e2etests/test_update_bytecode_connector.go
@@ -0,0 +1,59 @@
+package e2etests
+
+import (
+	"github.com/ethereum/go-ethereum/accounts/abi/bind"
+	"github.com/zeta-chain/zetacore/e2e/contracts/testconnectorzevm"
+	"github.com/zeta-chain/zetacore/e2e/runner"
+	"github.com/zeta-chain/zetacore/e2e/utils"
+	fungibletypes "github.com/zeta-chain/zetacore/x/fungible/types"
+)
+
+// TestUpdateBytecodeConnector tests updating the bytecode of a connector and interact with it
+func TestUpdateBytecodeConnector(r *runner.E2ERunner, _ []string) {
+	// Deploy the test contract
+	newTestConnectorAddr, tx, _, err := testconnectorzevm.DeployTestZetaConnectorZEVM(
+		r.ZEVMAuth,
+		r.ZEVMClient,
+		r.WZetaAddr,
+	)
+	if err != nil {
+		panic(err)
+	}
+
+	// Wait for the contract to be deployed
+	receipt := utils.MustWaitForTxReceipt(r.Ctx, r.ZEVMClient, tx, r.Logger, r.ReceiptTimeout)
+	if receipt.Status != 1 {
+		panic("contract deployment failed")
+	}
+
+	// Get the code hash of the new contract
+	codeHashRes, err := r.FungibleClient.CodeHash(r.Ctx, &fungibletypes.QueryCodeHashRequest{
+		Address: newTestConnectorAddr.String(),
+	})
+	if err != nil {
+		panic(err)
+	}
+	r.Logger.Info("New contract code hash: %s", codeHashRes.CodeHash)
+
+	r.Logger.Info("Updating the bytecode of the Connector")
+	msg := fungibletypes.NewMsgUpdateContractBytecode(
+		r.ZetaTxServer.GetAccountAddress(0),
+		r.ConnectorZEVMAddr.Hex(),
+		codeHashRes.CodeHash,
+	)
+	res, err := r.ZetaTxServer.BroadcastTx(utils.FungibleAdminName, msg)
+	if err != nil {
+		panic(err)
+	}
+	r.Logger.Info("Update connector bytecode tx hash: %s", res.TxHash)
+
+	r.Logger.Info("Can interact with the new code of the contract")
+	testConnectorContract, err := testconnectorzevm.NewTestZetaConnectorZEVM(r.ConnectorZEVMAddr, r.ZEVMClient)
+	if err != nil {
+		panic(err)
+	}
+	_, err = testConnectorContract.Foo(&bind.CallOpts{})
+	if err != nil {
+		panic(err)
+	}
+}