store kurtosis engine endpoint

informalsystems · Nov 20, 2024 · 7b0a498 · 7b0a498
1 parent 43ef9bd
commit 7b0a498
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/tests/network/ethpkg.rs b/src/tests/network/ethpkg.rs
@@ -19,6 +19,8 @@ use testresult::TestResult;
 
 #[derive(Builder, Debug)]
 pub struct EthPkgKurtosis {
+    #[builder(default = SocketAddr::new(Ipv4Addr::UNSPECIFIED.into(), 9710))]
+    pub kurtosis_engine_endpoint: SocketAddr,
     #[builder(default = "ethpkg".into())]
     pub enclave_name: String,
     pub el_socket: Option<SocketAddr>,
@@ -69,7 +71,7 @@ impl EthereumNetwork for EthPkgKurtosis {
 
         // CONNECT TO ENGINE
         let mut engine =
-            EngineServiceClient::connect(format!("http://{}:{}", Ipv4Addr::UNSPECIFIED, 9710))
+            EngineServiceClient::connect(format!("http://{}", self.kurtosis_engine_endpoint))
                 .await?;
 
         // CREATE ENCLAVE
@@ -208,7 +210,7 @@ impl EthereumNetwork for EthPkgKurtosis {
 
         // CONNECT TO ENGINE
         let mut engine =
-            EngineServiceClient::connect(format!("http://{}:{}", Ipv4Addr::UNSPECIFIED, 9710))
+            EngineServiceClient::connect(format!("http://{}", self.kurtosis_engine_endpoint))
                 .await?;
 
         // DESTROY ENCLAVE