Fix client test

entropyxyz · Nov 26, 2024 · b1c74b3 · b1c74b3
1 parent fbf1fb5
commit b1c74b3
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/crates/client/src/tests.rs b/crates/client/src/tests.rs
@@ -22,7 +22,7 @@ use entropy_testing_utils::{
     constants::{TEST_PROGRAM_WASM_BYTECODE, TSS_ACCOUNTS, X25519_PUBLIC_KEYS},
     helpers::{encode_verifying_key, spawn_tss_nodes_and_start_chain},
     substrate_context::test_context_stationary,
-    ChainSpecType,
+    test_node_process_testing_state, ChainSpecType,
 };
 use rand::{
     rngs::{OsRng, StdRng},
@@ -262,10 +262,12 @@ async fn test_remove_program_reference_counter() {
 #[tokio::test]
 #[serial]
 async fn test_get_oracle_headings() {
-    let substrate_context = test_context_stationary().await;
-
-    let api = get_api(&substrate_context.node_proc.ws_url).await.unwrap();
-    let rpc = get_rpc(&substrate_context.node_proc.ws_url).await.unwrap();
+    let force_authoring = true;
+    let context =
+        test_node_process_testing_state(ChainSpecType::IntegrationJumpStarted, force_authoring)
+            .await;
+    let api = get_api(&context[0].ws_url).await.unwrap();
+    let rpc = get_rpc(&context[0].ws_url).await.unwrap();
 
     let mut current_block = 0;
     while current_block < 2 {