use custom image in ci.

.github/workflows/tps.yml

@@ -7,6 +7,8 @@ jobs:
   push_to_registry:
     name: TPS zombienet
     runs-on: parity-large
+    container:
+        image: docker.io/paritytech/ci-unified:bullseye-1.84.1-2025-01-28-v202502131220
     steps:
       - name: Check out the repo
         uses: actions/checkout@v4
@@ -23,15 +25,25 @@ jobs:
         run: |
           mkdir -p bins
           cd bins
-          LINKS=$(curl  -H 'Accept: application/vnd.github+json' https://api.github.com/repos/paritytech/polkadot-sdk/releases |jq -r -c  '[.[0].assets.[] | select(.name == "polkadot" or .name == "polkadot-prepare-worker" or .name == "polkadot-execute-worker")| .browser_download_url] | .[]')
+          LINKS=$(curl  -H 'Accept: application/vnd.github+json' https://api.github.com/repos/paritytech/polkadot-sdk/releases |jq -r -c  '[ .[] |select(.target_commitish="polkadot-stable2503") |.assets | .[] | select(.name == "polkadot" or .name == "polkadot-prepare-worker" or .name == "polkadot-execute-worker")| .browser_download_url] | .[0,1,2]')
           for link in $(echo $LINKS);do
             echo "downloading $link"
-            curl -O $link
+            curl -L -O $link
           done;
           chmod +x polkadot*
           cd ..
 
       - name: Run Zombienet
+        timeout-minutes: 30
         run: |
           export PATH=$(pwd)/bins:$PATH
+          export ZOMBIE_SPECS=$(pwd)/zombienet-sdk-tests/tps/chainspec
           cargo run --manifest-path zombienet-sdk-tests/tps/Cargo.toml
+
+      - name: upload_logs
+        uses: actions/upload-artifact@v4
+        if: ${{ ! cancelled() }}
+        with:
+          name: zombienet-logs-${{ github.job }}-${{ github.sha }}
+          path: |
+            /tmp/zn/*/*.log

zombienet-sdk-tests/tps/src/main.rs

@@ -12,6 +12,7 @@ use std::{
 	collections::HashMap,
 	env,
 	error::Error,
+	process::Command,
 	sync::{atomic::AtomicU64, Arc},
 	time::Duration,
 };
@@ -441,6 +442,26 @@ async fn block_subscriber(
 	Err(subxt::Error::Other("Finalized block subscriber exited unexpectedly".to_string()))
 }
 
+fn cargo() -> Command {
+	Command::new(std::env::var("CARGO").unwrap_or_else(|_| "cargo".to_string()))
+}
+
+fn get_specs_directory() -> String {
+	let output = cargo()
+		.arg("metadata")
+		.arg("--format-version=1")
+		.output()
+		.ok()
+		.expect("Cargo metadata should works.");
+
+	let metadata: serde_json::Value =
+		serde_json::from_slice(&output.stdout).ok().expect("metadata should be valid.");
+	let workspace_directory =
+		metadata["workspace_root"].as_str().expect("workspace root should be present");
+
+	format!("{workspace_directory}/zombienet-sdk-tests/tps/chainspec")
+}
+
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn Error>> {
 	env_logger::init_from_env(
@@ -450,10 +471,16 @@ async fn main() -> Result<(), Box<dyn Error>> {
 	let snaps_server =
 		if let Ok(url) = env::var("ZOMBIE_SNAPS") { url } else { SNAPS_BUCKET.to_string() };
 
+	let chain_specs_dir = if let Ok(path) = env::var("ZOMBIE_SPECS") {
+		path
+	} else {
+		get_specs_directory().to_string()
+	};
+
 	let network = NetworkConfigBuilder::new()
 		.with_relaychain(|r| {
 			r.with_chain("paseo-local")
-				.with_chain_spec_path("chainspec/paseo-local.json")
+				.with_chain_spec_path(format!("{chain_specs_dir}/paseo-local.json").as_str())
 				.with_default_command("polkadot")
 				.with_node(|node| {
 					node.with_name("alice")
@@ -469,7 +496,9 @@ async fn main() -> Result<(), Box<dyn Error>> {
 		.with_parachain(|p| {
 			p.with_id(3369)
 				.evm_based(true)
-				.with_chain_spec_path("chainspec/local-v_paseo-local-3369.json")
+				.with_chain_spec_path(
+					format!("{chain_specs_dir}/local-v_paseo-local-3369.json").as_str(),
+				)
 				.with_default_command("mythos-node")
 				.with_collator(|n| {
 					n.with_name("muse-collator01")