fix: set proper key path and tempdir for collator key (#754)

eigerco · Feb 17, 2025 · d8d3a6e · d8d3a6e
1 parent 1a37199
commit d8d3a6e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/maat/src/lib.rs b/maat/src/lib.rs
@@ -16,7 +16,6 @@ use subxt::{
     },
     tx::PairSigner,
 };
-use tempfile::tempdir;
 use tracing::level_filters::LevelFilter;
 use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt, EnvFilter, Layer};
 use zombienet_configuration::shared::node::{Buildable, Initial, NodeConfigBuilder};
@@ -95,14 +94,13 @@ impl NodeConfigBuilderExt for NodeConfigBuilder<Initial> {
 ///
 /// We could use the TOML file if wasn't for not having the same requirements as this description,
 /// for example, when reading the TOML file, [you need to explicitly set a timeout](https://github.com/paritytech/zombienet-sdk/issues/254)
-pub fn local_testnet_config() -> NetworkConfig {
+pub fn local_testnet_config(temp_dir_path: &std::path::Path) -> NetworkConfig {
     let binding = find_polka_storage_node()
         .expect("couldn't find the polka-storage-node binary")
         .display()
         .to_string();
     let polka_storage_node_binary_path = binding.as_str();
-    let temp_dir = tempdir().unwrap();
-    let file_path = temp_dir.path().join("private_key.pem");
+    let file_path = temp_dir_path.join("private_key.pem");
     generate_pem_file(&file_path);
 
     NetworkConfigBuilder::new()
@@ -124,7 +122,8 @@ pub fn local_testnet_config() -> NetworkConfig {
                             ("--pool-type", "fork-aware").into(),
                             ("-lruntime=trace,parachain=debug").into(),
                             ("--p2p-listen-address=/ip4/127.0.0.1/tcp/62649").into(),
-                            (format!("--p2p-key={}", file_path.display()).as_str()).into(),
+                            ("--bootstrap-addresses=/ip4/127.0.0.1/tcp/1337").into(),
+                            (format!("--p2p-key=@{}", file_path.display()).as_str()).into(),
                         ])
                 })
         })

diff --git a/maat/tests/real_world.rs b/maat/tests/real_world.rs
@@ -311,7 +311,10 @@ async fn real_world_use_case() {
     let post_mapped_parameters =
         Arc::new(post::load_groth16_parameters(post_parameters_path).unwrap());
 
-    let network = local_testnet_config().spawn_native().await.unwrap();
+    let network = local_testnet_config(temp_dir.path())
+        .spawn_native()
+        .await
+        .unwrap();
     tracing::debug!("base dir: {:?}", network.base_dir());
     let collator = network.get_node(COLLATOR_NAME).unwrap();
     let client = Arc::new(