1. 1#![allow(clippy::unwrap_used, clippy::expect_used, clippy::panic)]
2. 2
3. 3//! `/search` reads exactly one Agent's memory (AGENTS.md invariant 37),
4. 4//! resolved the way `/memory` resolves it: `?agent=`, defaulting to `vak`.
5. 5
6. 6use axum::{
7. 7 Router,
8. 8 body::Body,
9. 9 http::{Request, StatusCode},
10. 10};
11. 11use http_body_util::BodyExt;
12. 12use serde_json::{Value, json};
13. 13use tower::ServiceExt;
14. 14
15. 15async fn call(app: &Router, method: &str, path: &str, body: Value) -> (StatusCode, Value) {
16. 16 let response = app
17. 17 .clone()
18. 18 .oneshot(
19. 19 Request::builder()
20. 20 .method(method)
21. 21 .uri(path)
22. 22 .header("content-type", "application/json")
23. 23 .body(Body::from(body.to_string()))
24. 24 .unwrap(),
25. 25 )
26. 26 .await
27. 27 .unwrap();
28. 28 let status = response.status();
29. 29 let bytes = response.into_body().collect().await.unwrap().to_bytes();
30. 30 (
31. 31 status,
32. 32 serde_json::from_slice(&bytes).unwrap_or(Value::Null),
33. 33 )
34. 34}
35. 35
36. 36#[tokio::test]
37. 37async fn search_reads_the_resolved_agents_own_memory() {
38. 38 vak_config::paths::isolate_home_for_tests();
39. 39 let temp = tempfile::tempdir().unwrap();
40. 40 let cwd = temp.path().join("workspace");
41. 41 std::fs::create_dir_all(&cwd).unwrap();
42. 42 let core = vak_core::Core::new_with_trust(cwd, true).unwrap();
43. 43 core.set_sessions_home(temp.path().join("sessions-home"));
44. 44 let app = vak_server::router(core);
45. 45
46. 46 let (status, body) = call(
47. 47 &app,
48. 48 "POST",
49. 49 "/memory",
50. 50 json!({"text": "the rollback window opens at noon", "kind": "fact", "tag": "rollback"}),
51. 51 )
52. 52 .await;
53. 53 assert_eq!(status, StatusCode::CREATED, "{body}");
54. 54
55. 55 let (status, found) = call(&app, "GET", "/search?q=rollback%20window", json!({})).await;
56. 56 assert_eq!(status, StatusCode::OK, "{found}");
57. 57 assert!(
58. 58 found.to_string().contains("rollback window opens at noon"),
59. 59 "the default Agent finds its own note: {found}"
60. 60 );
61. 61
62. 62 let (status, scoped) = call(
63. 63 &app,
64. 64 "GET",
65. 65 "/search?q=rollback%20window&agent=vak",
66. 66 json!({}),
67. 67 )
68. 68 .await;
69. 69 assert_eq!(status, StatusCode::OK);
70. 70 assert_eq!(found, scoped, "no agent means `vak`, exactly like /memory");
71. 71}
72. 72