light client tests: use `ChainheadBackendBuilder::build_with_backgrou…

…nd_driver` (#1823) * backend rpc tests: use background task * fix build
paritytech · Oct 14, 2024 · bcdf0bb · bcdf0bb
1 parent da2fcd5
commit bcdf0bb
Showing 1 changed file with 3 additions and 17 deletions.
diff --git a/testing/integration-tests/src/light_client/mod.rs b/testing/integration-tests/src/light_client/mod.rs
@@ -29,7 +29,6 @@
 
 use crate::utils::node_runtime;
 use codec::Compact;
-use futures::StreamExt;
 use std::sync::Arc;
 use subxt::backend::chain_head::ChainHeadBackend;
 use subxt::backend::rpc::RpcClient;
@@ -178,27 +177,14 @@ async fn run_test(backend: BackendType) -> Result<(), subxt::Error> {
     //     .unwrap();
     // let chain_config = chainspec.get();
 
-    tracing::trace!("Init light clinet");
+    tracing::trace!("Init light client");
     let now = std::time::Instant::now();
     let (_lc, rpc) = LightClient::relay_chain(POLKADOT_SPEC)?;
 
     let api = match backend {
         BackendType::Unstable => {
-            let (backend, mut driver) = ChainHeadBackend::builder().build(RpcClient::new(rpc));
-            tokio::spawn(async move {
-                while let Some(val) = driver.next().await {
-                    if let Err(e) = val {
-                        if e.is_disconnected_will_reconnect() {
-                            tracing::info!(
-                                "The RPC connection was lost and we may have missed a few blocks"
-                            );
-                            continue;
-                        }
-
-                        tracing::error!("Error driving unstable backend: {e}");
-                    }
-                }
-            });
+            let backend =
+                ChainHeadBackend::builder().build_with_background_driver(RpcClient::new(rpc));
             let api: OnlineClient<PolkadotConfig> =
                 OnlineClient::from_backend(Arc::new(backend)).await?;
             api