Add trie cache warmup

.github/workflows/zombienet_polkadot.yml

@@ -879,3 +879,33 @@ jobs:
           ref-slug: ${{ needs.preflight.outputs.SOURCE_REF_SLUG }}
           test: "functional::approved_peer_mixed_validators::approved_peer_mixed_validators_test"
           prefix: "polkadot"
+
+  #
+  #
+  zombienet-polkadot-parachains-warm-up-trie-cache:
+    needs: [preflight]
+    if: ${{ (needs.preflight.outputs.changes_substrate || needs.preflight.outputs.changes_polkadot) &&  ! contains(needs.preflight.outputs.FLAKY_TESTS, 'zombienet-polkadot-parachains-warm-up-trie-cache') }}
+    runs-on: ${{ needs.preflight.outputs.ZOMBIENET_RUNNER }} # NOTE: should be zombienet-arc-runner (without quotes)
+    timeout-minutes: 60
+    container:
+      image: ${{ needs.preflight.outputs.ZOMBIENET_IMAGE }}
+    env:
+      # sdk tests are looking for POLKADOT_IMAGE
+      POLKADOT_IMAGE: "${{ needs.preflight.outputs.TEMP_IMAGES_BASE }}/polkadot-debug:${{ needs.preflight.outputs.DOCKER_IMAGES_VERSION }}"
+      COL_IMAGE: "${{ needs.preflight.outputs.TEMP_IMAGES_BASE }}/colander:${{ needs.preflight.outputs.DOCKER_IMAGES_VERSION }}"
+      CUMULUS_IMAGE: "${{ needs.preflight.outputs.TEMP_IMAGES_BASE }}/polkadot-parachain-debug:${{ needs.preflight.outputs.DOCKER_IMAGES_VERSION }}"
+      MALUS_IMAGE: "${{ needs.preflight.outputs.TEMP_IMAGES_BASE }}/malus:${{ needs.preflight.outputs.DOCKER_IMAGES_VERSION }}"
+      RUST_LOG: ${{ needs.preflight.outputs.RUST_LOG }}
+      ZOMBIE_PROVIDER: ${{ needs.preflight.outputs.ZOMBIE_PROVIDER }}
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+
+      - name: zombienet_test
+        uses: ./.github/actions/zombienet-sdk
+        with:
+          gh-token: ${{ secrets.GITHUB_TOKEN }}
+          build-id: ${{ needs.preflight.outputs.BUILD_RUN_ID }}
+          ref-slug: ${{ needs.preflight.outputs.SOURCE_REF_SLUG }}
+          test: "parachains::warm_up_trie_cache::warm_up_trie_cache_test"
+          prefix: "polkadot"

Cargo.toml

@@ -1377,6 +1377,7 @@ substrate-test-runtime-transaction-pool = { path = "substrate/test-utils/runtime
 substrate-test-utils = { path = "substrate/test-utils" }
 substrate-wasm-builder = { path = "substrate/utils/wasm-builder", default-features = false }
 subxt = { version = "0.38.1", default-features = false }
+subxt-macro = { version = "0.38.1", default-features = false }
 subxt-metadata = { version = "0.38.0", default-features = false }
 subxt-signer = { version = "0.38" }
 syn = { version = "2.0.87" }

cumulus/test/service/src/lib.rs

@@ -876,6 +876,7 @@ pub fn node_config(
 		keystore: KeystoreConfig::InMemory,
 		database: DatabaseSource::RocksDb { path: root.join("db"), cache_size: 128 },
 		trie_cache_maximum_size: Some(64 * 1024 * 1024),
+		warm_up_trie_cache: None,
 		state_pruning: Some(PruningMode::ArchiveAll),
 		blocks_pruning: BlocksPruning::KeepAll,
 		chain_spec: spec,

polkadot/node/test/service/src/lib.rs

@@ -205,6 +205,7 @@ pub fn node_config(
 		keystore: KeystoreConfig::InMemory,
 		database: DatabaseSource::RocksDb { path: root.join("db"), cache_size: 128 },
 		trie_cache_maximum_size: Some(64 * 1024 * 1024),
+		warm_up_trie_cache: None,
 		state_pruning: Default::default(),
 		blocks_pruning: BlocksPruning::KeepFinalized,
 		chain_spec: Box::new(spec),

polkadot/zombienet-sdk-tests/Cargo.toml

@@ -9,14 +9,19 @@ publish = false
 
 [dependencies]
 anyhow = { workspace = true }
+asset-hub-westend-runtime = { workspace = true }
 codec = { workspace = true, features = ["derive"] }
 cumulus-zombienet-sdk-helpers = { workspace = true }
 env_logger = { workspace = true }
+futures = { workspace = true }
 log = { workspace = true }
+pallet-revive = { workspace = true, features = ["std"] }
 polkadot-primitives = { workspace = true, default-features = true }
 serde = { workspace = true }
 serde_json = { workspace = true }
+sp-core = { workspace = true }
 subxt = { workspace = true }
+subxt-macro = { workspace = true, features = ["runtime-metadata-insecure-url"] }
 subxt-signer = { workspace = true }
 tokio = { workspace = true, features = ["rt-multi-thread"] }
 tokio-util = { workspace = true, features = ["time"] }

polkadot/zombienet-sdk-tests/build.rs

@@ -119,7 +119,7 @@ fn main() {
 	// Ensure we have the needed metadata files in place to run zombienet tests
 	let manifest_path = env::var("CARGO_MANIFEST_DIR").unwrap();
 	const METADATA_DIR: &str = "metadata-files";
-	const CHAINS: [&str; 2] = ["rococo", "coretime-rococo"];
+	const CHAINS: [&str; 3] = ["rococo", "coretime-rococo", "asset-hub-westend"];
 
 	let metadata_path = format!("{manifest_path}/{METADATA_DIR}");
 

polkadot/zombienet-sdk-tests/tests/lib.rs

@@ -7,4 +7,6 @@ mod elastic_scaling;
 #[cfg(feature = "zombie-ci")]
 mod functional;
 #[cfg(feature = "zombie-ci")]
+mod parachains;
+#[cfg(feature = "zombie-ci")]
 mod smoke;

polkadot/zombienet-sdk-tests/tests/parachains/contract.txt

@@ -0,0 +1,16 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.20;
+
+import "@openzeppelin/contracts/token/ERC20/ERC20.sol";
+
+contract TestToken2 is ERC20 {
+    constructor() ERC20("TestToken2", "TT2") {}
+
+    function mint(uint256 amount) public {
+        _mint(msg.sender, amount);
+    }
+
+    function burn(uint256 amount) public {
+        _burn(msg.sender, amount);
+    }
+}

polkadot/zombienet-sdk-tests/tests/parachains/mod.rs

@@ -0,0 +1,4 @@
+// Copyright (C) Parity Technologies (UK) Ltd.
+// SPDX-License-Identifier: Apache-2.0
+
+mod warm_up_trie_cache;