feat(mcp): implement Prompts capability, Resources, and clean up daemon architecture

server/src/handlers/tasks.rs

@@ -487,3 +487,132 @@ impl McpTool for ListMilestonesHandler {
         Ok(data)
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use tempfile::tempdir;
+    use serde_json::json;
+
+    #[tokio::test]
+    async fn test_add_task_and_list() {
+        let dir = tempdir().unwrap();
+        let state = Arc::new(MemoryState::new(dir.path().to_str().unwrap()));
+        
+        let add_handler = AddTaskHandler;
+        let args = json!({
+            "title": "Fix the hyperdrive",
+            "description": "It's making a strange noise",
+            "acceptance_criteria": ["Stop the noise", "Reach lightspeed"],
+        });
+        
+        let res = add_handler.execute(args, state.clone()).await.unwrap();
+        assert!(res.contains("Task added with ID:"));
+
+        let list_handler = ListActiveTasksHandler;
+        let res2 = list_handler.execute(json!({}), state.clone()).await.unwrap();
+        assert!(res2.contains("Fix the hyperdrive"));
+    }
+
+    #[tokio::test]
+    async fn test_update_task_status() {
+        let dir = tempdir().unwrap();
+        let state = Arc::new(MemoryState::new(dir.path().to_str().unwrap()));
+        
+        let add_handler = AddTaskHandler;
+        let res = add_handler.execute(json!({"title": "Test", "description": "test"}), state.clone()).await.unwrap();
+        
+        let id_start = res.find("ID: ").unwrap() + 4;
+        let task_id = res[id_start..].trim();
+
+        let update_handler = UpdateTaskStatusHandler;
+        let args = json!({
+            "id": task_id,
+            "status": "done"
+        });
+        let res3 = update_handler.execute(args, state.clone()).await.unwrap();
+        assert_eq!(res3, "Task status updated.");
+        
+        let list_handler = ListActiveTasksHandler;
+        let res4 = list_handler.execute(json!({}), state.clone()).await.unwrap();
+        assert!(!res4.contains(task_id));
+    }
+
+    #[tokio::test]
+    async fn test_milestones_and_criteria() {
+        let dir = tempdir().unwrap();
+        let state = Arc::new(MemoryState::new(dir.path().to_str().unwrap()));
+        
+        // Add Milestone
+        let add_milestone = AddMilestoneHandler;
+        let args_ms = json!({
+            "name": "v1.0",
+            "title": "Release 1.0",
+            "description": "First release",
+            "target_date": 1700000000,
+            "end_date": 1700000000,
+            "namespace": "global"
+        });
+        let res1 = add_milestone.execute(args_ms, state.clone()).await.unwrap();
+        assert!(res1.contains("Milestone added"));
+        
+        // Fetch milestone ID from state directly to update
+        let ms_id = state.milestones.read_with(|ms| ms[0].id.clone());
+
+        // Update Milestone
+        let update_ms = UpdateMilestoneHandler;
+        let args_ums = json!({
+            "id": ms_id,
+            "status": "completed"
+        });
+        let res2 = update_ms.execute(args_ums, state.clone()).await.unwrap();
+        assert_eq!(res2, "Milestone updated");
+
+        // List Milestones
+        let list_ms = ListMilestonesHandler;
+        let res3 = list_ms.execute(json!({"namespace": "global"}), state.clone()).await.unwrap();
+        assert!(res3.contains("completed"));
+        assert!(res3.contains("Release 1.0"));
+
+        // Task Acceptance Criteria
+        let add_task = AddTaskHandler;
+        let res_task = add_task.execute(json!({"title": "Test", "description": "desc"}), state.clone()).await.unwrap();
+        let task_id = res_task[res_task.find("ID: ").unwrap() + 4..].trim();
+
+        let set_ac = SetAcceptanceCriteriaHandler;
+        let args_ac = json!({
+            "task_id": task_id,
+            "task_title": "Test",
+            "criteria": ["Do X", "Do Y"]
+        });
+        let res4 = set_ac.execute(args_ac, state.clone()).await.unwrap();
+        assert_eq!(res4, "Acceptance criteria set successfully.");
+
+        let verify_ac = VerifyAcceptanceCriteriaHandler;
+        let args_vac = json!({
+            "task_id": task_id,
+            "criteria": "Do X",
+            "proof": "I did X"
+        });
+        let res5 = verify_ac.execute(args_vac, state.clone()).await.unwrap();
+        assert!(res5.contains("Acceptance criteria verified"));
+    }
+
+    #[tokio::test]
+    async fn test_delete_task_with_children() {
+        let dir = tempfile::tempdir().unwrap();
+        let state = Arc::new(MemoryState::new(dir.path().to_str().unwrap()));
+
+        let add_task = AddTaskHandler;
+        let parent = add_task.execute(json!({"title": "Parent", "description": "p"}), state.clone()).await.unwrap();
+        let parent_id = parent[parent.find("ID: ").unwrap() + 4..].trim().to_string();
+
+        let child = add_task.execute(json!({"title": "Child", "description": "c", "parent_id": parent_id}), state.clone()).await.unwrap();
+        let _child_id = child[child.find("ID: ").unwrap() + 4..].trim().to_string();
+
+        let del_task = DeleteTaskHandler;
+        let res_del = del_task.execute(json!({"id": parent_id}), state.clone()).await.unwrap();
+        assert!(res_del.contains("Deleted task and its children (2 total)."));
+    }
+}
+