From 0f8baeeb0d25a7015a2d313d683863e3c69645e6 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Thu, 21 Nov 2024 18:19:23 +0100 Subject: [PATCH] test: Add database deletion test for qdrant Added testing of database deletion for qdrant Test COG-488 --- cognee/tests/test_qdrant.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/cognee/tests/test_qdrant.py b/cognee/tests/test_qdrant.py index 680399e60..4c2462c3b 100644 --- a/cognee/tests/test_qdrant.py +++ b/cognee/tests/test_qdrant.py @@ -59,9 +59,16 @@ async def main(): print(f"{result}\n") history = await cognee.get_search_history() - 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) + qdrant_client = get_vector_engine().get_qdrant_client() + collections_response = await qdrant_client.get_collections() + assert len(collections_response.collections) == 0, "QDrant vector database is not empty" + if __name__ == "__main__": import asyncio asyncio.run(main())