diff --git a/core/tests/routes/memory/test_memory_by_user.py b/core/tests/routes/memory/test_memory_by_user.py index d42a9711..dff37eee 100644 --- a/core/tests/routes/memory/test_memory_by_user.py +++ b/core/tests/routes/memory/test_memory_by_user.py @@ -3,32 +3,32 @@ # episodic memories are saved having the correct user def test_episodic_memory_by_user(client): - # send websocket message from user A + # send websocket message from user C send_websocket_message({ "text": "I am user C", }, client, user_id="C") # episodic recall (no user) params = { - "text": "I am user C" + "text": "I am user" } - response = client.get(f"/memory/recall/", params=params, headers={"user_id": "C"}) + response = client.get(f"/memory/recall/", params=params) json = response.json() assert response.status_code == 200 episodic_memories = json["vectors"]["collections"]["episodic"] - assert len(episodic_memories) == 1 + assert len(episodic_memories) == 0 # episodic recall (memories from non existing user) params = { - "text": "I am user A" + "text": "I am user not existing" } - response = client.get(f"/memory/recall/", params=params, headers={"user_id": "A"}) + response = client.get(f"/memory/recall/", params=params, headers={"user_id": "not_existing"}) json = response.json() assert response.status_code == 200 episodic_memories = json["vectors"]["collections"]["episodic"] assert len(episodic_memories) == 0 - # episodic recall (memories from user A) + # episodic recall (memories from user C) params = { "text": "I am user C" }