Added substrate validators warp sync test

Author: DenzelPenzel

substrate/zombienet/zombienet-sdk/tests/utils.rs

@@ -4,6 +4,8 @@
 use zombienet_sdk::{LocalFileSystem, Network, NetworkConfig};
 
 pub const BEST_BLOCK_METRIC: &str = "block_height{status=\"best\"}";
+pub const FINALIZED_BLOCK_METRIC: &str = "substrate_block_height{status=\"finalized\"}";
+pub const BEEFY_BEST_BLOCK_METRIC: &str = "substrate_beefy_best_block";
 pub const DEFAULT_SUBSTRATE_IMAGE: &str = "docker.io/paritypr/substrate:latest";
 
 pub const DEFAULT_DB_SNAPSHOT_URL: &str =

substrate/zombienet/zombienet-sdk/tests/zombie_ci/basic_warp_sync.rs

@@ -3,7 +3,9 @@
 
 use std::time::Duration;
 
-use crate::utils::{initialize_network, BEST_BLOCK_METRIC};
+use crate::utils::{
+	initialize_network, BEST_BLOCK_METRIC, DEFAULT_CHAIN_SPEC, DEFAULT_DB_SNAPSHOT_URL,
+};
 use anyhow::{anyhow, Result};
 use env_logger::Env;
 use zombienet_orchestrator::network::node::LogLineCountOptions;
@@ -24,15 +26,10 @@ const NODE_ROLE_METRIC: &str = "node_roles";
 const PEER_COUNT_METRIC: &str = "substrate_sub_libp2p_peers_count";
 const NODE_NAMES: [&str; 4] = ["alice", "bob", "charlie", "dave"];
 const SNAPSHOT_NODES: [&str; 3] = ["alice", "bob", "charlie"];
-const INTEGRATION_IMAGE_ENV: &str = "ZOMBIENET_INTEGRATION_TEST_IMAGE";
 const DB_SNAPSHOT_ENV: &str = "DB_SNAPSHOT";
 const CHAIN_SPEC_ENV: &str = "WARP_CHAIN_SPEC_PATH";
 const DB_BLOCK_HEIGHT_ENV: &str = "DB_BLOCK_HEIGHT";
 const DEFAULT_SUBSTRATE_IMAGE: &str = "docker.io/paritypr/substrate:latest";
-const DEFAULT_DB_SNAPSHOT_URL: &str =
-	"https://storage.googleapis.com/zombienet-db-snaps/substrate/0001-basic-warp-sync/chains-0bb3f0be2ce41b5615b224215bcc8363aa0416a6.tgz";
-const DEFAULT_CHAIN_SPEC: &str =
-	"https://raw.githubusercontent.com/paritytech/polkadot-sdk/refs/heads/master/substrate/zombienet/0001-basic-warp-sync/chain-spec.json";
 
 #[tokio::test(flavor = "multi_thread")]
 async fn basic_warp_sync() -> Result<()> {

substrate/zombienet/zombienet-sdk/tests/zombie_ci/block_building.rs

@@ -3,14 +3,12 @@
 
 use std::time::Duration;
 
-use crate::utils::initialize_network;
+use crate::utils::{initialize_network, DEFAULT_CHAIN_SPEC, DEFAULT_SUBSTRATE_IMAGE};
 use anyhow::{anyhow, Result};
 use subxt::{config::substrate::SubstrateConfig, dynamic::tx, OnlineClient};
 use subxt_signer::sr25519::dev;
 use zombienet_orchestrator::network::node::LogLineCountOptions;
-use zombienet_sdk::{environment::get_spawn_fn, NetworkConfig, NetworkConfigBuilder, NetworkNode};
-
-const DEFAULT_SUBSTRATE_IMAGE: &str = "docker.io/paritypr/substrate:latest";
+use zombienet_sdk::{NetworkConfig, NetworkConfigBuilder, NetworkNode};
 
 const NODE_NAMES: [&str; 2] = ["alice", "bob"];
 
@@ -28,13 +26,17 @@ const LOG_TIMEOUT_SECS: u64 = 2;
 const SCRIPT_TIMEOUT_SECS: u64 = 30;
 
 const REMARK_PAYLOAD: &[u8] = b"block-building-test";
+const INTEGRATION_IMAGE_ENV: &str = "ZOMBIENET_INTEGRATION_TEST_IMAGE";
+const CHAIN_SPEC_ENV: &str = "WARP_CHAIN_SPEC_PATH";
 
 #[tokio::test(flavor = "multi_thread")]
 async fn block_building_test() -> Result<()> {
 	let _ = env_logger::try_init_from_env(
 		env_logger::Env::default().filter_or(env_logger::DEFAULT_FILTER_ENV, "info"),
 	);
 
+	ensure_env_defaults();
+
 	log::info!("Spawning network");
 	let config = build_network_config()?;
 	let network = initialize_network(config).await?;
@@ -54,13 +56,28 @@ async fn block_building_test() -> Result<()> {
 	Ok(())
 }
 
+fn ensure_env_defaults() {
+	if std::env::var(INTEGRATION_IMAGE_ENV).is_err() {
+		std::env::set_var(INTEGRATION_IMAGE_ENV, DEFAULT_SUBSTRATE_IMAGE);
+	}
+	if std::env::var(CHAIN_SPEC_ENV).is_err() {
+		std::env::set_var(CHAIN_SPEC_ENV, DEFAULT_CHAIN_SPEC);
+	}
+}
+
 fn build_network_config() -> Result<NetworkConfig> {
+	let integration_image = std::env::var(INTEGRATION_IMAGE_ENV)
+		.unwrap_or_else(|_| DEFAULT_SUBSTRATE_IMAGE.to_string());
+	let chain_spec =
+		std::env::var(CHAIN_SPEC_ENV).unwrap_or_else(|_| DEFAULT_CHAIN_SPEC.to_string());
+
 	let config = NetworkConfigBuilder::new()
 		.with_relaychain(|relaychain| {
 			relaychain
 				.with_chain("local")
 				.with_default_command("substrate")
-				.with_default_image(DEFAULT_SUBSTRATE_IMAGE)
+				.with_default_image(integration_image.as_str())
+				.with_chain_spec_path(chain_spec.as_str())
 				.with_default_args(vec!["-lparachain=debug".into()])
 				.with_node(|node| node.with_name("alice"))
 				.with_node(|node| node.with_name("bob"))