test: fix integration & unit tests

execution.go

@@ -16,14 +16,14 @@
 	"github.com/ethereum/go-ethereum/ethclient"
 	"github.com/ethereum/go-ethereum/rpc"
 
 	"github.com/golang-jwt/jwt/v5"
 	execution "github.com/rollkit/go-execution"
 	proxy_json_rpc "github.com/rollkit/go-execution/proxy/jsonrpc"
 	execution_types "github.com/rollkit/go-execution/types"
 )
 
 var (
 	ErrNilPayloadStatus     = errors.New("nil payload status")
 	ErrInvalidPayloadStatus = errors.New("invalid payload status")
 )
 
@@ -103,7 +103,7 @@
 
 // Stop stops the execution client and closes all connections
 func (c *EngineAPIExecutionClient) Stop() {
 	c.proxyClient.Stop()
 
 	if c.engineClient != nil {
 		c.engineClient.Close()
@@ -145,7 +145,7 @@
 		return execution_types.Hash{}, 0, fmt.Errorf("engine_getPayloadV3 failed: %w", err)
 	}
 
-	stateRoot := common.HexToHash(payloadResult.ExecutionPayload.StateRoot.Hex())
+	stateRoot := payloadResult.ExecutionPayload.StateRoot
 	rollkitStateRoot := execution_types.Hash(stateRoot[:])
 
 	gasLimit := payloadResult.ExecutionPayload.GasLimit

execution_test.go

@@ -13,7 +13,7 @@
 	"encoding/hex"
 
 	"github.com/ethereum/go-ethereum/common"
 	"github.com/ethereum/go-ethereum/core/types"
 	"github.com/rollkit/go-execution-evm/mocks"
 	proxy_json_rpc "github.com/rollkit/go-execution/proxy/jsonrpc"
 	execution_types "github.com/rollkit/go-execution/types"
@@ -77,21 +77,21 @@
 	defer mockEth.Close()
 
 	jwtSecret := generateTestJWTSecret()
+	prevStateRoot := execution_types.Hash(common.Hex2Bytes("111122223333444455556666777788889999aaaabbbbccccddddeeeeffff0000"))
+
 	client, err := NewEngineAPIExecutionClient(
 		&proxy_json_rpc.Config{},
 		mockEth.URL,
 		mockEngine.URL,
 		jwtSecret,
-		common.Hash{},
+		common.Hash(prevStateRoot),
 		common.Address{},
 	)
 	require.NoError(t, err)
 
 	blockHeight := uint64(1)
 	timestamp := time.Now().UTC().Truncate(time.Second)
 
-	prevStateRoot := execution_types.Hash(common.Hex2Bytes("111122223333444455556666777788889999aaaabbbbccccddddeeeeffff0000"))
-
 	testTxBytes, err := types.NewTransaction(1, common.Address{}, big.NewInt(0), 1000, big.NewInt(875000000), nil).MarshalBinary()
 	require.NoError(t, err)
 	testTx := execution_types.Tx(testTxBytes)
@@ -113,10 +113,8 @@
 	require.Equal(t, common.BytesToHash(prevStateRoot[:]).Hex(), lastCall.FinalizedBlockHash)
 
 	mockStateRoot := common.HexToHash("0x0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef")
-	var expectedStateRoot execution_types.Hash
-	copy(expectedStateRoot[:], mockStateRoot.Bytes())
 
-	require.Equal(t, expectedStateRoot, stateRoot)
+	require.Equal(t, execution_types.Hash(mockStateRoot[:]), stateRoot)
 	require.Equal(t, uint64(21000), gasUsed)
 }
 
@@ -174,7 +172,7 @@
 			}
 		}
 
 		json.NewEncoder(w).Encode(map[string]interface{}{
 			"jsonrpc": "2.0",
 			"id":      req["id"],
 			"result":  resp,

integration_test.go

@@ -201,7 +201,7 @@ func TestExecutionClientLifecycle(t *testing.T) {
 		stateroot, gasUsed, err := executionClient.ExecuteTxs(context.Background(), []rollkit_types.Tx{rollkit_types.Tx(txBytes)}, blockHeight, blockTime, rollkitGenesisStateRoot)
 		require.NoError(t, err)
 		assert.Greater(t, gasLimit, gasUsed)
-		assert.Equal(t, newStateroot.Bytes(), []byte(stateroot))
+		assert.Equal(t, rollkit_types.Hash(newStateroot[:]), stateroot)
 	})
 }