Add tests precollection logic

crates/forge-runner/src/package_tests.rs

@@ -128,18 +128,20 @@ impl TestDetails {
     }
 }
 
+// TODO: Remove in next PRs
 #[derive(Debug, Clone)]
-pub struct TestTarget<C> {
+pub struct TestTargetDeprecated<C> {
     pub tests_location: TestTargetLocation,
     pub sierra_program: ProgramArtifact,
     pub sierra_program_path: Arc<Utf8PathBuf>,
     pub casm_program: Arc<RawCasmProgram>,
     pub test_cases: Vec<TestCase<C>>,
 }
 
-impl TestTarget<TestCaseWithConfig> {
+// TODO: Remove in next PRs
+impl TestTargetDeprecated<TestCaseWithConfig> {
     #[tracing::instrument(skip_all, level = "debug")]
-    pub fn from_raw(
+    pub fn from_raw_deprecated(
         test_target_raw: TestTargetRaw,
         tracked_resource: &ForgeTrackedResource,
     ) -> Result<TestTargetWithConfig> {
@@ -211,3 +213,72 @@ pub struct TestCase<C> {
     pub name: String,
     pub config: C,
 }
+
+#[derive(Debug, Clone)]
+pub struct TestTarget<T> {
+    pub tests_location: TestTargetLocation,
+    pub sierra_program: ProgramArtifact,
+    pub sierra_program_path: Arc<Utf8PathBuf>,
+    pub casm_program: Option<Arc<RawCasmProgram>>,
+    pub test_cases: Vec<T>,
+}
+
+impl TestTarget<TestCandidate> {
+    #[tracing::instrument(skip_all, level = "debug")]
+    pub fn from_raw(test_target_raw: TestTargetRaw) -> Result<TestTarget<TestCandidate>> {
+        macro_rules! by_id {
+            ($field:ident) => {{
+                let temp: HashMap<_, _> = test_target_raw
+                    .sierra_program
+                    .program
+                    .$field
+                    .iter()
+                    .map(|f| (f.id.id, f))
+                    .collect();
+
+                temp
+            }};
+        }
+        let funcs = by_id!(funcs);
+        let type_declarations = by_id!(type_declarations);
+
+        let sierra_program_registry =
+            ProgramRegistry::<CoreType, CoreLibfunc>::new(&test_target_raw.sierra_program.program)?;
+        let type_size_map = get_type_size_map(
+            &test_target_raw.sierra_program.program,
+            &sierra_program_registry,
+        )
+        .ok_or_else(|| anyhow!("can not get type size map"))?;
+
+        let default_executables = vec![];
+        let debug_info = test_target_raw.sierra_program.debug_info.clone();
+        let executables = debug_info
+            .as_ref()
+            .and_then(|info| info.executables.get("snforge_internal_test_executable"))
+            .unwrap_or(&default_executables);
+
+        let test_cases = executables
+            .par_iter()
+            .map(|case| {
+                let func = funcs[&case.id];
+                let name = case.debug_name.clone().unwrap().into();
+                let test_details = TestDetails::build(func, &type_declarations, &type_size_map);
+
+                Ok(TestCandidate { name, test_details })
+            })
+            .collect::<Result<_>>()?;
+
+        Ok(TestTarget {
+            tests_location: test_target_raw.tests_location,
+            test_cases,
+            sierra_program: test_target_raw.sierra_program,
+            sierra_program_path: test_target_raw.sierra_program_path.into(),
+            casm_program: None,
+        })
+    }
+}
+
+pub struct TestCandidate {
+    pub name: String,
+    pub test_details: TestDetails,
+}

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

@@ -1,4 +1,4 @@
-use super::{TestCase, TestTarget};
+use super::{TestCase, TestTargetDeprecated};
 use crate::{
     TestCaseIsIgnored,
     expected_result::{ExpectedPanicValue, ExpectedTestResult},
@@ -9,7 +9,7 @@ use cheatnet::runtime_extensions::forge_config_extension::config::{
 };
 use conversions::serde::serialize::SerializeToFeltVec;
 
-pub type TestTargetWithConfig = TestTarget<TestCaseConfig>;
+pub type TestTargetWithConfig = TestTargetDeprecated<TestCaseConfig>;
 
 pub type TestCaseWithConfig = TestCase<TestCaseConfig>;
 

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

@@ -1,4 +1,4 @@
-use super::{TestCase, TestTarget};
+use super::{TestCase, TestTargetDeprecated};
 use crate::{TestCaseIsIgnored, expected_result::ExpectedTestResult, package_tests::TestDetails};
 use anyhow::Result;
 use cairo_vm::types::program::Program;
@@ -9,7 +9,7 @@ use starknet_api::block::BlockNumber;
 use universal_sierra_compiler_api::representation::RawCasmProgram;
 use url::Url;
 
-pub type TestTargetWithResolvedConfig = TestTarget<TestCaseResolvedConfig>;
+pub type TestTargetWithResolvedConfig = TestTargetDeprecated<TestCaseResolvedConfig>;
 
 pub type TestCaseWithResolvedConfig = TestCase<TestCaseResolvedConfig>;
 

crates/forge/Cargo.toml

@@ -56,6 +56,7 @@ chrono.workspace = true
 tracing-subscriber.workspace = true
 tracing.workspace = true
 tracing-chrome.workspace = true
+rayon.workspace = true
 
 [[bin]]
 name = "snforge"

crates/forge/src/run_tests/package.rs

@@ -27,7 +27,8 @@ use console::Style;
 use forge_runner::{
     forge_config::ForgeConfig,
     package_tests::{
-        TestTarget, raw::TestTargetRaw, with_config_resolved::TestTargetWithResolvedConfig,
+        TestTargetDeprecated, raw::TestTargetRaw,
+        with_config_resolved::TestTargetWithResolvedConfig,
     },
     test_case_summary::AnyTestCaseSummary,
     test_target_summary::TestTargetSummary,
@@ -144,7 +145,7 @@ async fn test_package_with_config_resolved(
     let mut test_targets_with_resolved_config = Vec::with_capacity(test_targets.len());
 
     for test_target in test_targets {
-        let test_target = TestTarget::from_raw(
+        let test_target = TestTargetDeprecated::from_raw_deprecated(
             test_target,
             &forge_config.test_runner_config.tracked_resource,
         )?;

crates/forge/src/run_tests/workspace.rs

@@ -2,15 +2,18 @@ use super::package::RunForPackageArgs;
 use crate::run_tests::messages::latest_blocks_numbers::LatestBlocksNumbersMessage;
 use crate::run_tests::messages::tests_failure_summary::TestsFailureSummaryMessage;
 use crate::run_tests::messages::workspace_summary::WorkspaceSummaryMessage;
+use crate::scarb::load_test_artifacts;
 use crate::{
     ExitStatus, TestArgs, block_number_map::BlockNumberMap, run_tests::package::run_for_package,
     scarb::build_artifacts_with_scarb, shared_cache::FailedTestsCache,
 };
 use anyhow::{Context, Result};
 use camino::Utf8PathBuf;
+use forge_runner::package_tests::{TestCandidate, TestTarget};
 use forge_runner::test_case_summary::AnyTestCaseSummary;
 use forge_runner::{CACHE_DIR, test_target_summary::TestTargetSummary};
 use foundry_ui::UI;
+use rayon::iter::{IntoParallelRefIterator, ParallelIterator};
 use scarb_api::{
     metadata::{Metadata, PackageMetadata},
     target_dir_for_workspace,
@@ -42,6 +45,26 @@ impl WorkspaceDirs {
     }
 }
 
+type TestTargets = Vec<TestTarget<TestCandidate>>;
+
+#[expect(dead_code)]
+fn collect_packages_with_tests(
+    workspace_dirs: &WorkspaceDirs,
+    packages: &[PackageMetadata],
+) -> Result<Vec<(PackageMetadata, TestTargets)>> {
+    packages
+        .par_iter()
+        .map(|package| {
+            let test_targets_raw = load_test_artifacts(&workspace_dirs.artifacts_dir, package)?;
+            let test_targets = test_targets_raw
+                .into_iter()
+                .map(TestTarget::from_raw)
+                .collect::<Result<Vec<_>>>()?;
+            Ok((package.clone(), test_targets))
+        })
+        .collect()
+}
+
 #[tracing::instrument(skip_all, level = "debug")]
 pub async fn run_for_workspace(
     scarb_metadata: &Metadata,