Companion for: try-runtime::follow-chain - keep connection

tests/common.rs

@@ -15,7 +15,7 @@
 // along with Substrate.  If not, see <http://www.gnu.org/licenses/>.
 
 use polkadot_core_primitives::Block;
-use remote_externalities::rpc_api::get_finalized_head;
+use remote_externalities::rpc_api::RpcService;
 use std::{
 	io::{BufRead, BufReader, Read},
 	process::{Child, ExitStatus},
@@ -56,7 +56,12 @@ async fn wait_n_finalized_blocks_from(n: usize, url: &str) {
 	let mut interval = tokio::time::interval(Duration::from_secs(6));
 
 	loop {
-		if let Ok(block) = get_finalized_head::<Block, _>(url).await {
+		let rpc_service = RpcService::new(url, false).await;
+		if rpc_service.is_err() {
+			continue
+		}
+
+		if let Ok(block) = rpc_service.unwrap().get_finalized_head::<Block>().await {
 			built_blocks.insert(block);
 			if built_blocks.len() > n {
 				break