feat: add test execution in debug mode

Author: flferretti

src/lib.rs

@@ -1072,6 +1072,9 @@ pub async fn debug_recipe(
 ) -> miette::Result<()> {
     let recipe_path = get_recipe_path(&debug_data.recipe_path)?;
 
+    let is_test_mode = debug_data.test_index.is_some();
+    let test_index = debug_data.test_index.unwrap_or(0);
+
     let build_data = BuildData {
         build_platform: debug_data.build_platform,
         target_platform: debug_data.target_platform,
@@ -1177,22 +1180,64 @@ pub async fn debug_recipe(
             }
         }
 
-        tracing::info!("\nTo run the actual build, use:");
-        tracing::info!(
-            "rattler-build build --recipe {}",
-            output.build_configuration.directories.recipe_path.display()
-        );
-        tracing::info!("Or run the build script directly with:");
-        if cfg!(windows) {
-            tracing::info!(
-                "cd {} && ./conda_build.bat",
-                output.build_configuration.directories.work_dir.display()
-            );
+        if is_test_mode {
+            // Test mode: setup test environment and run specific test
+            let tests = &output.recipe.tests;
+
+            if tests.is_empty() {
+                return Err(miette::miette!(
+                    "No tests found in recipe. Cannot run tests in debug mode."
+                ));
+            }
+
+            if test_index >= tests.len() {
+                return Err(miette::miette!(
+                    "Test index {} out of range. Recipe has {} test(s).",
+                    test_index,
+                    tests.len()
+                ));
+            }
+
+            tracing::info!("\n=== Test Debug Mode ===");
+            tracing::info!("Running test {} of {}", test_index, tests.len() - 1);
+            tracing::info!("Available tests:");
+            for (idx, test) in tests.iter().enumerate() {
+                let test_type = match test {
+                    crate::recipe::parser::TestType::Python { .. } => "Python",
+                    crate::recipe::parser::TestType::Perl { .. } => "Perl",
+                    crate::recipe::parser::TestType::R { .. } => "R",
+                    crate::recipe::parser::TestType::Ruby { .. } => "Ruby",
+                    crate::recipe::parser::TestType::Command(_) => "Command",
+                    crate::recipe::parser::TestType::Downstream(_) => "Downstream",
+                    crate::recipe::parser::TestType::PackageContents { .. } => "PackageContents",
+                };
+                let marker = if idx == test_index { ">>>" } else { "   " };
+                tracing::info!("{} Test {}: {}", marker, idx, test_type);
+            }
+
+            tracing::info!("\nTest environment will be prepared for test index {}.", test_index);
+            tracing::info!("\nTo execute tests after setting up the environment, use:");
+            tracing::info!("  rattler-build test <package_file>");
+            tracing::info!("\nNote: The test environment has been created, but the package must be built first before running tests.");
         } else {
+            // Build mode: provide instructions to run the build script
+            tracing::info!("\nTo run the actual build, use:");
             tracing::info!(
-                "cd {} && ./conda_build.sh",
-                output.build_configuration.directories.work_dir.display()
+                "rattler-build build --recipe {}",
+                output.build_configuration.directories.recipe_path.display()
             );
+            tracing::info!("Or run the build script directly with:");
+            if cfg!(windows) {
+                tracing::info!(
+                    "cd {} && ./conda_build.bat",
+                    output.build_configuration.directories.work_dir.display()
+                );
+            } else {
+                tracing::info!(
+                    "cd {} && ./conda_build.sh",
+                    output.build_configuration.directories.work_dir.display()
+                );
+            }
         }
     }
 

src/opt.rs

@@ -895,6 +895,10 @@ pub struct DebugOpts {
     /// Name of the specific output to debug (only required when a recipe has multiple outputs)
     #[arg(long, help = "Name of the specific output to debug")]
     pub output_name: Option<String>,
+
+    /// Run tests in debug mode.
+    #[arg(long, help = "Run tests in debug mode. Optionally specify the test index (default: 0)")]
+    pub test: Option<Option<usize>>,
 }
 
 #[derive(Debug, Clone)]
@@ -916,6 +920,8 @@ pub struct DebugData {
     pub common: CommonData,
     /// Name of the specific output to debug (if recipe has multiple outputs)
     pub output_name: Option<String>,
+    /// Test index to run in debug mode (if Some). None means no test execution.
+    pub test_index: Option<usize>,
 }
 
 impl DebugData {
@@ -933,6 +939,7 @@ impl DebugData {
             channels: opts.channels,
             common: CommonData::from_opts_and_config(opts.common, config.unwrap_or_default()),
             output_name: opts.output_name,
+            test_index: opts.test.flatten().or(opts.test.and(Some(0))),
         }
     }
 }