From 54daa6986cc3d1a884696958ac2fc83a1cbd0c24 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Thu, 21 Nov 2024 16:39:38 +0100 Subject: [PATCH] test: Add test for deletion of neo4j graph database Added test to verify all nodes and edges from neo4j database have been cleaned Test COG-488 --- cognee/tests/test_neo4j.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/cognee/tests/test_neo4j.py b/cognee/tests/test_neo4j.py index 756b29cc4..02f3eaccd 100644 --- a/cognee/tests/test_neo4j.py +++ b/cognee/tests/test_neo4j.py @@ -61,6 +61,15 @@ async def main(): assert len(history) == 6, "Search history is not correct." + await cognee.prune.prune_data() + assert not os.path.isdir(data_directory_path), "Local data files are not deleted" + + await cognee.prune.prune_system(metadata=True) + from cognee.infrastructure.databases.graph import get_graph_engine + graph_engine = await get_graph_engine() + nodes, edges = await graph_engine.get_graph_data() + assert len(nodes) == 0 and len(edges) == 0, "Neo4j graph database is not empty" + if __name__ == "__main__": import asyncio asyncio.run(main())