From 6e4ffcca7a9e1136fb99fd178467d5b250384b63 Mon Sep 17 00:00:00 2001
From: arsam <me@asabbagh.com>
Date: Fri, 18 Sep 2026 23:06:45 +0330
Subject: [PATCH] Import tests: tests/test_index_router

Import the existing local project snapshot. This commit organizes existing files; it does not represent newly developed functionality.
---
 tests/test_index_router.py | 69 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 69 insertions(+)
 create mode 100644 tests/test_index_router.py

diff --git a/tests/test_index_router.py b/tests/test_index_router.py
new file mode 100644
index 0000000..0d29f91
--- /dev/null
+++ b/tests/test_index_router.py
@@ -0,0 +1,69 @@
+from fastapi import FastAPI
+from fastapi.testclient import TestClient
+
+
+def test_ingest_and_search_private_index(tmp_path, monkeypatch):
+    monkeypatch.setattr("axon.services.index.settings.index_db_path", str(tmp_path / "index.sqlite3"))
+
+    from axon.routers.index import router
+
+    app = FastAPI()
+    app.include_router(router)
+    client = TestClient(app)
+
+    ingest = client.post(
+        "/ingest",
+        json={
+            "documents": [
+                {
+                    "id": "doc1",
+                    "title": "Private Axon Notes",
+                    "content": "Axon supports private retrieval infrastructure.",
+                    "url": "private://doc1",
+                }
+            ]
+        },
+    )
+    assert ingest.status_code == 200
+    assert ingest.json()["indexed"] == 1
+
+    search = client.post("/index/search", json={"query": "private retrieval"})
+    assert search.status_code == 200
+    data = search.json()
+    assert data["results"][0]["id"] == "doc1"
+
+
+def test_search_can_include_private_index(monkeypatch):
+    from axon.models import SearchResult
+    from axon.services.cache import InMemoryCache
+
+    cache = InMemoryCache()
+
+    async def async_cache():
+        return cache
+
+    async def empty(*args, **kwargs):
+        return []
+
+    async def private(*args, **kwargs):
+        return [SearchResult(title="Private", url="private://1", content="secret axon data", score=0.9)]
+
+    async def rerank(query, contents):
+        return [(url, text, 0.9) for url, text in contents]
+
+    monkeypatch.setattr("axon.routers.search.get_cache", async_cache)
+    monkeypatch.setattr("axon.routers.search.search_brave", empty)
+    monkeypatch.setattr("axon.routers.search.search_ddg", empty)
+    monkeypatch.setattr("axon.routers.search.search_serper", empty)
+    monkeypatch.setattr("axon.routers.search.search_index_as_results", private)
+    monkeypatch.setattr("axon.routers.search.rerank_results", rerank)
+
+    from axon.routers.search import router
+
+    app = FastAPI()
+    app.include_router(router)
+    client = TestClient(app)
+
+    resp = client.post("/search", json={"query": "secret"})
+    assert resp.status_code == 200
+    assert resp.json()["results"][0]["url"] == "private://1"