From 17021ce8c1fdf47e19dd3bf1340a34f8ab227033 Mon Sep 17 00:00:00 2001
From: dcodesdev <101001810+dcodesdev@users.noreply.github.com>
Date: Sat, 21 Sep 2024 23:31:59 +0300
Subject: [PATCH] bug fix

---
 crates/rustfinity-runner/src/command.rs | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/crates/rustfinity-runner/src/command.rs b/crates/rustfinity-runner/src/command.rs
index a3a7738..e2d2f24 100644
--- a/crates/rustfinity-runner/src/command.rs
+++ b/crates/rustfinity-runner/src/command.rs
@@ -15,14 +15,16 @@ pub async fn run_code(code_base64: &str, challenge: &str) -> anyhow::Result<Stri
     output.push_str(&tests_output);
     output.push_str("\n---\n");
 
-    let test_binary_path = extract_unittest_path(&output).unwrap();
+    let test_binary_path = extract_unittest_path(&output);
 
-    let time_output = benchmark_time(&challenge, &test_binary_path).await?;
-    output.push_str(time_output.to_string().as_str());
-    output.push_str("\n---\n");
+    if let Some(test_binary_path) = test_binary_path {
+        let time_output = benchmark_time(&challenge, &test_binary_path).await?;
+        output.push_str(time_output.to_string().as_str());
+        output.push_str("\n---\n");
 
-    let memory_output = memory_benchmark(&challenge, &test_binary_path).await?;
-    output.push_str(memory_output.as_str());
+        let memory_output = memory_benchmark(&challenge, &test_binary_path).await?;
+        output.push_str(memory_output.as_str());
+    }
 
     Ok(output)
 }