refactor: Implement unified search abstraction, MemoryState refactoring, error handling, and unit test expansion

server/src/handlers/logs.rs

@@ -74,3 +74,52 @@ impl McpTool for GetRecentLogsHandler {
         Ok(result)
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use tempfile::tempdir;
+    use std::sync::Arc;
+    use serde_json::json;
+
+    #[tokio::test]
+    async fn test_watch_process_logs() {
+        let dir = tempdir().unwrap();
+        let state = Arc::new(MemoryState::new(dir.path().to_str().unwrap()));
+        let handler = WatchProcessLogsHandler;
+
+        let log_file = dir.path().join("test.log");
+        std::fs::write(&log_file, "line1\nline2").unwrap();
+
+        let args = json!({
+            "file_path": log_file.to_str().unwrap()
+        });
+
+        let result = handler.execute(args, state)
+            .await
+            .map_err(|e| format!("Failed to watch logs: {}", e))
+            .unwrap();
+        assert!(result.contains("Started watching logs"));
+    }
+
+    #[tokio::test]
+    async fn test_get_recent_logs() {
+        let dir = tempdir().unwrap();
+        let state = Arc::new(MemoryState::new(dir.path().to_str().unwrap()));
+        let handler = GetRecentLogsHandler;
+
+        let log_file = dir.path().join("test_recent.log");
+        std::fs::write(&log_file, "line1\nline2\nline3").unwrap();
+
+        let args = json!({
+            "file_path": log_file.to_str().unwrap()
+        });
+
+        let result = handler.execute(args, state)
+            .await
+            .map_err(|e| format!("Failed to get recent logs: {}", e))
+            .unwrap();
+        assert!(result.contains("line1"));
+        assert!(result.contains("line3"));
+    }
+}