From b1c74b34187a0a0451e9f6a47da980c3e9df450e Mon Sep 17 00:00:00 2001
From: peg <ameba23@systemli.org>
Date: Tue, 26 Nov 2024 15:15:26 +0100
Subject: [PATCH] Fix client test

---
 crates/client/src/tests.rs | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/crates/client/src/tests.rs b/crates/client/src/tests.rs
index e079fcfe7..52abd3329 100644
--- 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 {