1 files changed, 47 insertions, 1 deletions
diff --git a/Omni/Agent/Tools.hs b/Omni/Agent/Tools.hs
index e132c86..0312924 100644
--- a/Omni/Agent/Tools.hs
+++ b/Omni/Agent/Tools.hs
@@ -124,7 +124,53 @@ test =
         let result = ToolResult False "" (Just "error occurred")
         case Aeson.decode (Aeson.encode result) of
           Nothing -> Test.assertFailure "Failed to decode ToolResult"
-          Just decoded -> toolResultError decoded Test.@=? Just "error occurred"
+          Just decoded -> toolResultError decoded Test.@=? Just "error occurred",
+      Test.unit "readFileTool handles missing files" <| do
+        let args = Aeson.object ["path" .= ("/nonexistent/path/to/file.txt" :: Text)]
+        result <- Engine.toolExecute readFileTool args
+        case Aeson.fromJSON result of
+          Aeson.Success (tr :: ToolResult) -> do
+            toolResultSuccess tr Test.@=? False
+            isJust (toolResultError tr) Test.@=? True
+          Aeson.Error e -> Test.assertFailure e,
+      Test.unit "editFileTool handles no-match case" <| do
+        let args =
+              Aeson.object
+                [ "path" .= ("/nonexistent/file.txt" :: Text),
+                  "old_str" .= ("needle" :: Text),
+                  "new_str" .= ("replacement" :: Text)
+                ]
+        result <- Engine.toolExecute editFileTool args
+        case Aeson.fromJSON result of
+          Aeson.Success (tr :: ToolResult) -> toolResultSuccess tr Test.@=? False
+          Aeson.Error e -> Test.assertFailure e,
+      Test.unit "runBashTool captures exit codes" <| do
+        let args = Aeson.object ["command" .= ("exit 42" :: Text)]
+        result <- Engine.toolExecute runBashTool args
+        case Aeson.fromJSON result of
+          Aeson.Success (tr :: ToolResult) -> do
+            toolResultSuccess tr Test.@=? False
+            toolResultError tr Test.@=? Just "Exit code: 42"
+          Aeson.Error e -> Test.assertFailure e,
+      Test.unit "runBashTool captures stdout" <| do
+        let args = Aeson.object ["command" .= ("echo hello" :: Text)]
+        result <- Engine.toolExecute runBashTool args
+        case Aeson.fromJSON result of
+          Aeson.Success (tr :: ToolResult) -> do
+            toolResultSuccess tr Test.@=? True
+            ("hello" `Text.isInfixOf` toolResultOutput tr) Test.@=? True
+          Aeson.Error e -> Test.assertFailure e,
+      Test.unit "searchCodebaseTool returns structured results" <| do
+        let args =
+              Aeson.object
+                [ "pattern" .= ("module" :: Text),
+                  "path" .= ("." :: Text),
+                  "max_results" .= (5 :: Int)
+                ]
+        result <- Engine.toolExecute searchCodebaseTool args
+        case Aeson.fromJSON result of
+          Aeson.Success (tr :: ToolResult) -> toolResultSuccess tr Test.@=? True
+          Aeson.Error e -> Test.assertFailure e
     ]
 
 data ToolResult = ToolResult