fix(walltime): ensure working directory is used when running the cmd

Author: not-matthias

src/run/runner/helpers/command.rs

@@ -55,6 +55,10 @@ impl CommandBuilder {
         self.cwd = Some(dir.as_ref().to_owned());
     }
 
+    pub fn get_current_dir(&self) -> Option<&OsStr> {
+        self.cwd.as_deref()
+    }
+
     pub fn wrap<S, I, T>(&mut self, wrapper: S, wrapper_args: I) -> &mut Self
     where
         S: AsRef<OsStr>,

src/run/runner/tests.rs

@@ -254,4 +254,35 @@ mod walltime {
         })
         .await;
     }
+
+    // Ensure that the working directory is used correctly
+    #[rstest::rstest]
+    #[tokio::test]
+    async fn test_walltime_executor_in_working_dir(#[values(false, true)] enable_perf: bool) {
+        let (system_info, run_data, _temp_dir) = create_test_setup().await;
+        let (_permit, executor) = get_walltime_executor().await;
+
+        let cmd = r#"
+if [ "$(basename "$(pwd)")" != "within_sub_directory" ]; then
+    echo "FAIL: Working directory is not 'within_sub_directory'"
+    exit 1
+fi
+"#;
+
+        let mut config = walltime_config(cmd, enable_perf);
+
+        let dir = TempDir::new().unwrap();
+        config.working_directory = Some(
+            dir.path()
+                .join("within_sub_directory")
+                .to_string_lossy()
+                .to_string(),
+        );
+        std::fs::create_dir_all(config.working_directory.as_ref().unwrap()).unwrap();
+
+        executor
+            .run(&config, &system_info, &run_data, &None)
+            .await
+            .unwrap();
+    }
 }

src/run/runner/wall_time/perf/mod.rs

@@ -156,6 +156,12 @@ impl PerfRunner {
         );
         let mut wrapped_builder = CommandBuilder::new("sh");
         wrapped_builder.args(["-c", &raw_command]);
+
+        // IMPORTANT: Preserve the working directory from the original command
+        if let Some(cwd) = cmd_builder.get_current_dir() {
+            wrapped_builder.current_dir(cwd);
+        }
+
         let cmd = wrap_with_sudo(wrapped_builder)?.build();
         debug!("cmd: {cmd:?}");