cool~!

NamseEnt · Dec 3, 2023 · 6960bdb · 6960bdb
1 parent c90abbf
commit 6960bdb
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/oioi/agent/src/bash.rs b/oioi/agent/src/bash.rs
@@ -4,11 +4,7 @@ pub(crate) async fn run(command: impl AsRef<str>) -> Result<Vec<u8>> {
     let command = command.as_ref();
 
     let output = tokio::process::Command::new("bash")
-        .arg("-c")
-        .arg(format!(
-            "\"{command}\"",
-            command = command.replace('"', "\\\"")
-        ))
+        .args(["-c", command])
         .output()
         .await
         .map_err(|e| anyhow::anyhow!("Failed to run {command}: {e}"))?;
@@ -22,3 +18,14 @@ pub(crate) async fn run(command: impl AsRef<str>) -> Result<Vec<u8>> {
 
     Ok(output.stdout)
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[tokio::test]
+    async fn test_run() {
+        let output = run("echo hello").await.unwrap();
+        assert_eq!(output, b"hello\n");
+    }
+}