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

server/src/api/rest.rs

@@ -107,3 +107,59 @@ pub async fn gate_set_handler(
 pub async fn health_handler() -> &'static str {
     "OK"
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::router::MemoryHandler;
+    use crate::state::MemoryState;
+    use std::sync::atomic::AtomicUsize;
+    use std::sync::RwLock;
+    use tempfile::tempdir;
+
+    #[tokio::test]
+    async fn test_gate_handlers() {
+        let dir = tempdir().unwrap();
+        let state = Arc::new(MemoryState::new(dir.path().to_str().unwrap()));
+        let app_state = Arc::new(AppState {
+            handler: Arc::new(MemoryHandler::new(state.clone())),
+            clients: RwLock::new(HashMap::new()),
+            next_id: AtomicUsize::new(1),
+        });
+
+        // Set a gate to authorized
+        let set_req = GateSetReq {
+            action: "push".to_string(),
+            target: "main".to_string(),
+            namespace: Some("global".to_string()),
+            params: HashMap::new(),
+            authorize: Some(true),
+            block: None,
+            reason: None,
+        };
+        let res_set = gate_set_handler(State(app_state.clone()), Json(set_req)).await.unwrap();
+        assert_eq!(res_set.into_response().status(), axum::http::StatusCode::OK);
+
+        // Verify the gate (and consume it)
+        let verify_req = GateVerifyReq {
+            action: "push".to_string(),
+            target: "main".to_string(),
+            namespace: Some("global".to_string()),
+            params: HashMap::new(),
+            consume: true,
+        };
+        let res_verify = gate_verify_handler(State(app_state.clone()), Query(verify_req)).await.unwrap();
+        assert_eq!(res_verify.into_response().status(), axum::http::StatusCode::OK);
+
+        // Verify again should fail since it was consumed
+        let verify_req2 = GateVerifyReq {
+            action: "push".to_string(),
+            target: "main".to_string(),
+            namespace: Some("global".to_string()),
+            params: HashMap::new(),
+            consume: false,
+        };
+        let res_verify2 = gate_verify_handler(State(app_state.clone()), Query(verify_req2)).await;
+        assert!(res_verify2.is_err());
+    }
+}