mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-06-15 17:25:26 -04:00
catalini82
9d7a3d66c0
Co-authored-by: Cata <cata@bigjohn.local> Co-authored-by: Alexandre Teixeira <111787685+alteixeira20@users.noreply.github.com>
29 lines
982 B
Python
29 lines
982 B
Python
from unittest.mock import MagicMock
|
|
|
|
import routes.memory_routes as memory_routes
|
|
from src.memory import MemoryManager
|
|
|
|
|
|
def test_memory_search_returns_only_callers_memories(monkeypatch, tmp_path):
|
|
manager = MemoryManager(str(tmp_path))
|
|
alice_memory = manager.add_entry("Project codename is Odyssey", owner="alice")
|
|
bob_memory = manager.add_entry("Project codename is Odyssey", owner="bob")
|
|
manager.save([alice_memory, bob_memory])
|
|
|
|
monkeypatch.setattr(memory_routes, "get_current_user", lambda request: "bob")
|
|
router = memory_routes.setup_memory_routes(manager, MagicMock())
|
|
search = next(
|
|
route.endpoint
|
|
for route in router.routes
|
|
if route.path == "/api/memory/search" and "POST" in route.methods
|
|
)
|
|
|
|
result = search(
|
|
request=None,
|
|
query="Project codename is Odyssey",
|
|
session_id=None,
|
|
category=None,
|
|
)
|
|
|
|
assert [memory["id"] for memory in result["memories"]] == [bob_memory["id"]]
|