From fdc64d21cc0f5214a3550eaf891399707fc05927 Mon Sep 17 00:00:00 2001
From: jfldde <168934971+jfldde@users.noreply.github.com>
Date: Mon, 2 Dec 2024 22:46:42 +0000
Subject: [PATCH] Add custom docker image support

---
 src/framework.rs | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/framework.rs b/src/framework.rs
index 948eb77..5da8970 100644
--- a/src/framework.rs
+++ b/src/framework.rs
@@ -434,33 +434,34 @@ fn generate_test_config<T: TestCase>(
         }
     };
 
+    let citrea_docker_image = std::env::var("CITREA_DOCKER_IMAGE").ok();
     Ok(TestConfig {
         bitcoin: bitcoin_confs,
         sequencer: FullSequencerConfig::new(
             sequencer,
             sequencer_rollup,
-            None,
+            citrea_docker_image,
             sequencer_dir,
             env.sequencer(),
         )?,
         batch_prover: FullBatchProverConfig::new(
             batch_prover,
             batch_prover_rollup,
-            None,
+            citrea_docker_image,
             batch_prover_dir,
             env.batch_prover(),
         )?,
         light_client_prover: FullLightClientProverConfig::new(
             light_client_prover,
             light_client_prover_rollup,
-            None,
+            citrea_docker_image,
             light_client_prover_dir,
             env.light_client_prover(),
         )?,
         full_node: FullFullNodeConfig::new(
             (),
             full_node_rollup,
-            None,
+            citrea_docker_image,
             full_node_dir,
             env.full_node(),
         )?,