From 6a057711a2f0ed84f28940d2fc3080db3abaec0f Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 10 Dec 2025 11:46:37 +0100
Subject: [PATCH] feat: adds missing tests to context extension retriever
 integration test

---
 ...st_graph_completion_retriever_context_extension.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/cognee/tests/integration/retrieval/test_graph_completion_retriever_context_extension.py b/cognee/tests/integration/retrieval/test_graph_completion_retriever_context_extension.py
index 0d5413ff8..c87de16ef 100644
--- a/cognee/tests/integration/retrieval/test_graph_completion_retriever_context_extension.py
+++ b/cognee/tests/integration/retrieval/test_graph_completion_retriever_context_extension.py
@@ -213,3 +213,14 @@ async def test_get_graph_completion_extension_context_on_empty_graph(setup_test_
     assert all(isinstance(item, str) and item.strip() for item in answer), (
         "Answer must contain only non-empty strings"
     )
+
+
+@pytest.mark.asyncio
+async def test_graph_completion_extension_get_triplets_empty(setup_test_environment_empty):
+    """Integration test: verify GraphCompletionContextExtensionRetriever get_triplets handles empty graph."""
+    retriever = GraphCompletionContextExtensionRetriever()
+
+    triplets = await retriever.get_triplets("Who works at Figma?")
+
+    assert isinstance(triplets, list), "Triplets should be a list"
+    assert len(triplets) == 0, "Should return empty list on empty graph"