From 17ec4a9e8c4fc55eae29f217e179d3047fda2ca5 Mon Sep 17 00:00:00 2001
From: Oliver Nordbjerg <hi@notbjerg.me>
Date: Mon, 23 Sep 2024 08:59:12 +0200
Subject: [PATCH] feat: add set code tx helper to e2e utils

---
 crates/e2e-test-utils/src/transaction.rs | 48 +++++++++++++++++++++---
 1 file changed, 42 insertions(+), 6 deletions(-)

diff --git a/crates/e2e-test-utils/src/transaction.rs b/crates/e2e-test-utils/src/transaction.rs
index 71f2e132d07a..f5cfd5d743a9 100644
--- a/crates/e2e-test-utils/src/transaction.rs
+++ b/crates/e2e-test-utils/src/transaction.rs
@@ -3,7 +3,8 @@ use alloy_network::{
     eip2718::Encodable2718, Ethereum, EthereumWallet, TransactionBuilder, TransactionBuilder4844,
 };
 use alloy_primitives::{hex, Address, Bytes, TxKind, B256, U256};
-use alloy_rpc_types::{TransactionInput, TransactionRequest};
+use alloy_rpc_types::{Authorization, TransactionInput, TransactionRequest};
+use alloy_signer::SignerSync;
 use alloy_signer_local::PrivateKeySigner;
 use eyre::Ok;
 
@@ -14,7 +15,7 @@ pub struct TransactionTestContext;
 impl TransactionTestContext {
     /// Creates a static transfer and signs it, returning an envelope.
     pub async fn transfer_tx(chain_id: u64, wallet: PrivateKeySigner) -> TxEnvelope {
-        let tx = tx(chain_id, 21000, None, 0);
+        let tx = tx(chain_id, 21000, None, None, 0);
         Self::sign_tx(wallet, tx).await
     }
 
@@ -31,7 +32,7 @@ impl TransactionTestContext {
         init_code: Bytes,
         wallet: PrivateKeySigner,
     ) -> TxEnvelope {
-        let tx = tx(chain_id, gas, Some(init_code), 0);
+        let tx = tx(chain_id, gas, Some(init_code), None, 0);
         Self::sign_tx(wallet, tx).await
     }
 
@@ -46,12 +47,41 @@ impl TransactionTestContext {
         signed.encoded_2718().into()
     }
 
+    /// Creates an EIP-7702 set code transaction and signs it, returning an envelope.
+    ///
+    /// The EIP-7702 will delegate the code of the signer to the contract at `delegate_to`.
+    pub async fn set_code_tx(
+        chain_id: u64,
+        delegate_to: Address,
+        wallet: PrivateKeySigner,
+    ) -> TxEnvelope {
+        let authorization =
+            Authorization { chain_id: U256::from(chain_id), address: delegate_to, nonce: 0 };
+        let signature = wallet
+            .sign_hash_sync(&authorization.signature_hash())
+            .expect("could not sign authorization");
+        let tx = tx(chain_id, 48100, None, Some(authorization.into_signed(signature)), 0);
+        Self::sign_tx(wallet, tx).await
+    }
+
+    /// Creates an EIP-7702 set code transaction and signs it, returning bytes.
+    ///
+    /// The EIP-7702 will delegate the code of the signer to the contract at `delegate_to`.
+    pub async fn set_code_tx_bytes(
+        chain_id: u64,
+        delegate_to: Address,
+        wallet: PrivateKeySigner,
+    ) -> Bytes {
+        let signed = Self::set_code_tx(chain_id, delegate_to, wallet).await;
+        signed.encoded_2718().into()
+    }
+
     /// Creates a tx with blob sidecar and sign it
     pub async fn tx_with_blobs(
         chain_id: u64,
         wallet: PrivateKeySigner,
     ) -> eyre::Result<TxEnvelope> {
-        let mut tx = tx(chain_id, 21000, None, 0);
+        let mut tx = tx(chain_id, 21000, None, None, 0);
 
         let mut builder = SidecarBuilder::<SimpleCoder>::new();
         builder.ingest(b"dummy blob");
@@ -85,7 +115,7 @@ impl TransactionTestContext {
         nonce: u64,
     ) -> Bytes {
         let l1_block_info = Bytes::from_static(&hex!("7ef9015aa044bae9d41b8380d781187b426c6fe43df5fb2fb57bd4466ef6a701e1f01e015694deaddeaddeaddeaddeaddeaddeaddeaddead000194420000000000000000000000000000000000001580808408f0d18001b90104015d8eb900000000000000000000000000000000000000000000000000000000008057650000000000000000000000000000000000000000000000000000000063d96d10000000000000000000000000000000000000000000000000000000000009f35273d89754a1e0387b89520d989d3be9c37c1f32495a88faf1ea05c61121ab0d1900000000000000000000000000000000000000000000000000000000000000010000000000000000000000002d679b567db6187c0c8323fa982cfb88b74dbcc7000000000000000000000000000000000000000000000000000000000000083400000000000000000000000000000000000000000000000000000000000f4240"));
-        let tx = tx(chain_id, 21000, Some(l1_block_info), nonce);
+        let tx = tx(chain_id, 21000, Some(l1_block_info), None, nonce);
         let signer = EthereumWallet::from(wallet);
         <TransactionRequest as TransactionBuilder<Ethereum>>::build(tx, &signer)
             .await
@@ -112,7 +142,13 @@ impl TransactionTestContext {
 }
 
 /// Creates a type 2 transaction
-fn tx(chain_id: u64, gas: u128, data: Option<Bytes>, nonce: u64) -> TransactionRequest {
+fn tx(
+    chain_id: u64,
+    gas: u128,
+    data: Option<Bytes>,
+    delegate_to: Option<Authorization>,
+    nonce: u64,
+) -> TransactionRequest {
     TransactionRequest {
         nonce: Some(nonce),
         value: Some(U256::from(100)),