Display partition tests count

CHANGELOG.md

@@ -7,15 +7,19 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
-## Forge
+### Forge
+
+#### Added
+
+- Partitioned test execution with `--partition <INDEX>/<TOTAL>` flag. Read more [here](https://foundry-rs.github.io/starknet-foundry/snforge-advanced-features/tests-partitioning.html)
 
 #### Changed
 
 - `--detailed-resources` output now includes all gas-related resources
 
 ## [0.55.0] - 2026-01-13
 
-## Forge
+### Forge
 
 #### Changed
 

crates/forge-runner/Cargo.toml

@@ -45,3 +45,7 @@ strum_macros = "0.27"
 scarb-oracle-hint-service.workspace = true
 tracing.workspace = true
 comfy-table.workspace = true
+scarb-api = { path = "../scarb-api" }
+
+[dev-dependencies]
+test-case.workspace = true

crates/forge-runner/src/filtering.rs

@@ -0,0 +1,29 @@
+use crate::package_tests::TestCase;
+
+/// Result of filtering a test case.
+#[derive(Debug)]
+pub enum FilterResult {
+    /// Test case should be included.
+    Included,
+    /// Test case should be excluded for the given reason.
+    Excluded(ExcludeReason),
+}
+
+/// Reason for excluding a test case.
+#[derive(Debug, Clone, Copy, PartialEq, Eq)]
+pub enum ExcludeReason {
+    /// Test case is ignored.
+    Ignored,
+    /// Test case is excluded from the current partition.
+    ExcludedFromPartition,
+}
+
+pub trait TestCaseFilter {
+    fn filter<T>(&self, test_case: &TestCase<T>) -> FilterResult
+    where
+        T: TestCaseIsIgnored;
+}
+
+pub trait TestCaseIsIgnored {
+    fn is_ignored(&self) -> bool;
+}

crates/forge-runner/src/lib.rs

@@ -1,6 +1,5 @@
 use crate::coverage_api::run_coverage;
 use crate::forge_config::{ExecutionDataToSave, ForgeConfig};
-use crate::package_tests::TestCase;
 use crate::running::{run_fuzz_test, run_test};
 use crate::test_case_summary::TestCaseSummary;
 use anyhow::Result;
@@ -25,22 +24,24 @@ use tokio::sync::mpsc::{Sender, channel};
 use tokio::task::JoinHandle;
 use universal_sierra_compiler_api::representation::RawCasmProgram;
 
+pub mod backtrace;
 pub mod build_trace_data;
 pub mod coverage_api;
+pub mod debugging;
 pub mod expected_result;
+pub mod filtering;
 pub mod forge_config;
+mod gas;
+pub mod messages;
 pub mod package_tests;
+pub mod partition;
 pub mod profiler_api;
+pub mod running;
+pub mod scarb;
 pub mod test_case_summary;
 pub mod test_target_summary;
 pub mod tests_summary;
 
-pub mod backtrace;
-pub mod debugging;
-mod gas;
-pub mod messages;
-pub mod running;
-
 pub const CACHE_DIR: &str = ".snfoundry_cache";
 
 const BUILTINS: [&str; 11] = [
@@ -57,16 +58,6 @@ const BUILTINS: [&str; 11] = [
     "MulMod",
 ];
 
-pub trait TestCaseFilter {
-    fn should_be_run<T>(&self, test_case: &TestCase<T>) -> bool
-    where
-        T: TestCaseIsIgnored;
-}
-
-pub trait TestCaseIsIgnored {
-    fn is_ignored(&self) -> bool;
-}
-
 pub fn maybe_save_trace_and_profile(
     result: &AnyTestCaseSummary,
     execution_data_to_save: &ExecutionDataToSave,

crates/forge-runner/src/messages.rs

@@ -13,6 +13,7 @@ enum TestResultStatus {
     Failed,
     Ignored,
     Interrupted,
+    ExcludedFromPartition,
 }
 
 impl From<&AnyTestCaseSummary> for TestResultStatus {
@@ -26,6 +27,10 @@ impl From<&AnyTestCaseSummary> for TestResultStatus {
             | AnyTestCaseSummary::Fuzzing(TestCaseSummary::Ignored { .. }) => Self::Ignored,
             AnyTestCaseSummary::Single(TestCaseSummary::Interrupted { .. })
             | AnyTestCaseSummary::Fuzzing(TestCaseSummary::Interrupted { .. }) => Self::Interrupted,
+            AnyTestCaseSummary::Single(TestCaseSummary::ExcludedFromPartition { .. })
+            | AnyTestCaseSummary::Fuzzing(TestCaseSummary::ExcludedFromPartition { .. }) => {
+                Self::ExcludedFromPartition
+            }
         }
     }
 }
@@ -123,7 +128,9 @@ impl TestResultMessage {
             match self.status {
                 TestResultStatus::Passed => return format!("\n\n{msg}"),
                 TestResultStatus::Failed => return format!("\n\nFailure data:{msg}"),
-                TestResultStatus::Ignored | TestResultStatus::Interrupted => return String::new(),
+                TestResultStatus::Ignored
+                | TestResultStatus::Interrupted
+                | TestResultStatus::ExcludedFromPartition => return String::new(),
             }
         }
         String::new()
@@ -137,6 +144,11 @@ impl TestResultMessage {
             TestResultStatus::Interrupted => {
                 unreachable!("Interrupted tests should not have visible message representation")
             }
+            TestResultStatus::ExcludedFromPartition => {
+                unreachable!(
+                    "Tests excluded from partition should not have visible message representation"
+                )
+            }
         }
     }
 }

crates/forge-runner/src/package_tests/with_config.rs

@@ -1,7 +1,7 @@
 use super::{TestCase, TestTarget};
 use crate::{
-    TestCaseIsIgnored,
     expected_result::{ExpectedPanicValue, ExpectedTestResult},
+    filtering::TestCaseIsIgnored,
 };
 use cheatnet::runtime_extensions::forge_config_extension::config::{
     Expected, RawAvailableResourceBoundsConfig, RawForgeConfig, RawForkConfig, RawFuzzerConfig,

crates/forge-runner/src/package_tests/with_config_resolved.rs

@@ -1,5 +1,7 @@
 use super::{TestCase, TestTarget};
-use crate::{TestCaseIsIgnored, expected_result::ExpectedTestResult, package_tests::TestDetails};
+use crate::{
+    expected_result::ExpectedTestResult, filtering::TestCaseIsIgnored, package_tests::TestDetails,
+};
 use anyhow::Result;
 use cairo_vm::types::program::Program;
 use cheatnet::runtime_extensions::forge_config_extension::config::{
@@ -13,7 +15,8 @@ pub type TestTargetWithResolvedConfig = TestTarget<TestCaseResolvedConfig>;
 
 pub type TestCaseWithResolvedConfig = TestCase<TestCaseResolvedConfig>;
 
-fn sanitize_test_case_name(name: &str) -> String {
+#[must_use]
+pub fn sanitize_test_case_name(name: &str) -> String {
     // Test names generated by `#[test]` and `#[fuzzer]` macros contain internal suffixes
     name.replace("__snforge_internal_test_generated", "")
         .replace("__snforge_internal_fuzzer_generated", "")