From c8571ad3c7a86b30adf72e51121b45567bcd777e Mon Sep 17 00:00:00 2001 From: hajdul88 <52442977+hajdul88@users.noreply.github.com> Date: Fri, 12 Dec 2025 11:12:02 +0100 Subject: [PATCH] fixes enable backend access control on case asserts --- cognee/tests/test_search_db.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/cognee/tests/test_search_db.py b/cognee/tests/test_search_db.py index e279be3bf..336c77944 100644 --- a/cognee/tests/test_search_db.py +++ b/cognee/tests/test_search_db.py @@ -342,8 +342,20 @@ async def test_search_db(): ]: assert isinstance(search_results, list), f"{name}: should return a list" assert search_results, f"{name}: should not be empty" - assert isinstance(search_results[0], dict), f"{name}: expected dict payloads" - assert str(search_results[0].get("text", "")).strip(), f"{name}: missing non-empty 'text'" + + first = search_results[0] + assert isinstance(first, dict), f"{name}: expected dict entries, got {type(first).__name__}" + + payloads = search_results + if "search_result" in first and "text" not in first: + payloads = (first.get("search_result") or [None])[0] + + assert isinstance(payloads, list), ( + f"{name}: expected list payloads, got {type(payloads).__name__}" + ) + assert payloads, f"{name}: expected non-empty payload list" + assert isinstance(payloads[0], dict), f"{name}: expected dict payloads" + assert str(payloads[0].get("text", "")).strip(), f"{name}: missing non-empty 'text'" graph_engine = await get_graph_engine() graph = await graph_engine.get_graph_data()