From f5f285379aa232041380bac01dace0be9d72940c Mon Sep 17 00:00:00 2001
From: philip-morlier <philipmorlier@gmail.com>
Date: Fri, 8 Sep 2023 13:14:58 -0700
Subject: [PATCH] Implementation of test function for laconic's
 GetCpontractCode method.

---
 plugins/test-plugin/main.go | 43 +++++++++++++++++++++++++++++++++++--
 1 file changed, 41 insertions(+), 2 deletions(-)

diff --git a/plugins/test-plugin/main.go b/plugins/test-plugin/main.go
index 2e76f84e1..bd6cd6a06 100644
--- a/plugins/test-plugin/main.go
+++ b/plugins/test-plugin/main.go
@@ -5,9 +5,11 @@ import (
 	"math/big"
 	"time"
 	"os"
-	
+	"bytes"
+
 	"github.com/openrelayxyz/plugeth-utils/core"
 	"github.com/openrelayxyz/plugeth-utils/restricted/hexutil"
+	"github.com/openrelayxyz/plugeth-utils/restricted/crypto"
 )
 
 var hookChan chan map[string]struct{} = make(chan map[string]struct{}, 10)
@@ -223,6 +225,8 @@ func txTracer() {
 		log.Error("debug_traceTransaction failed",  "err", err)
 	}
 
+	testGetContractCode(t3)
+
 	debugArg0 := map[string]interface{}{
 		"input": "0x60006000fd",
 		"from": coinBase,
@@ -242,7 +246,6 @@ func txTracer() {
 	var trResult1 interface{}
 	err = client.Call(&trResult1, "debug_traceCall", debugArg1, "latest", t)
 
-
 	final := map[string]interface{}{
 		"input": "0x61520873000000000000000000000000000000000000000060006000600060006000f1",
 		"from": coinBase,
@@ -258,3 +261,39 @@ func txTracer() {
 
 }
 
+func testGetContractCode(hash core.Hash) {
+	
+	cl := apis[0].Service.(*engineService).stack
+	client, err := cl.Attach()
+	if err != nil {
+		log.Error("Error connecting with client testGetContractCode")
+	}
+
+	receipt := map[string]interface{}{}
+	err = client.Call(&receipt, "eth_getTransactionReceipt", hash)
+	if err != nil {
+		log.Error("Error calling getTransactionReciepts, testGetContractCode", "err", err)
+	}
+
+	var controlCode hexutil.Bytes
+	err = client.Call(&controlCode, "eth_getCode", receipt["contractAddress"], receipt["blockNumber"])
+	if err != nil {
+		log.Error("Error calling getCode, testGetContractCode", "err", err)
+	}
+
+	codeHash := crypto.Keccak256Hash(controlCode)
+
+	testCode, err := apis[0].Service.(*engineService).backend.GetContractCode(codeHash)
+	if err != nil {
+		log.Error("Error calling GetContractCode", "err", err)
+	}
+
+	if !bytes.Equal(testCode, controlCode) {
+
+		log.Error("Exit with error, return value from GetContractCode is divergent from control value")
+		os.Exit(1)
+	} 
+
+	log.Info("made it through checkGetContractCode")
+}
+