Tidy up test code

gnosisguild · Nov 5, 2024 · 06d1d8b · 06d1d8b
1 parent 9f3eeb3
commit 06d1d8b
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 14 deletions.
diff --git a/packages/ciphernode/evm/tests/enclave_sol_reader.rs b/packages/ciphernode/evm/tests/enclave_sol_reader.rs
@@ -0,0 +1 @@
+
diff --git a/packages/ciphernode/evm/tests/evm_reader.rs b/packages/ciphernode/evm/tests/evm_reader.rs
@@ -57,15 +57,18 @@ async fn evm_reader() -> Result<()> {
     // Create a WS provider
     // NOTE: Anvil must be available on $PATH
     let anvil = Anvil::new().block_time(1).try_spawn()?;
-    let ws = WsConnect::new(anvil.ws_endpoint());
-    let provider = ProviderBuilder::new().on_ws(ws).await?;
-    let arc_provider = WithChainId::new(provider).await?;
-    let contract = EmitLogs::deploy(arc_provider.get_provider()).await?;
+    let provider = WithChainId::new(
+        ProviderBuilder::new()
+            .on_ws(WsConnect::new(anvil.ws_endpoint()))
+            .await?,
+    )
+    .await?;
+    let contract = EmitLogs::deploy(provider.get_provider()).await?;
     let bus = EventBus::new(true).start();
-    let unwrapper = EvmEventUnwrapper { bus: bus.clone() }.start();
+
     EvmEventReader::attach(
-        &unwrapper.clone().into(),
-        &arc_provider,
+        &EvmEventUnwrapper { bus: bus.clone() }.start().into(),
+        &provider,
         test_event_extractor,
         &contract.address().to_string(),
         None,
@@ -111,12 +114,15 @@ async fn ensure_historical_events() -> Result<()> {
     // Create a WS provider
     // NOTE: Anvil must be available on $PATH
     let anvil = Anvil::new().block_time(1).try_spawn()?;
-    let ws = WsConnect::new(anvil.ws_endpoint());
-    let provider = ProviderBuilder::new().on_ws(ws).await?;
-    let arc_provider = WithChainId::new(provider).await?;
-    let contract = EmitLogs::deploy(arc_provider.get_provider()).await?;
+    let provider = WithChainId::new(
+        ProviderBuilder::new()
+            .on_ws(WsConnect::new(anvil.ws_endpoint()))
+            .await?,
+    )
+    .await?;
+    let contract = EmitLogs::deploy(provider.get_provider()).await?;
     let bus = EventBus::new(true).start();
-    let unwrapper = EvmEventUnwrapper { bus: bus.clone() }.start();
+
     let historical_msgs = vec!["these", "are", "historical", "events"];
     let live_events = vec!["these", "events", "are", "live"];
 
@@ -130,8 +136,8 @@ async fn ensure_historical_events() -> Result<()> {
     }
 
     EvmEventReader::attach(
-        &unwrapper.clone().into(),
-        &arc_provider,
+        &EvmEventUnwrapper { bus: bus.clone() }.start().into(),
+        &provider,
         test_event_extractor,
         &contract.address().to_string(),
         None,