test: Update tests with regards to auto scaling changes

.github/workflows/e2e_tests.yml

@@ -104,6 +104,7 @@ jobs:
       - name: Run default basic pipeline with telemetry on
         env:
           ENV: 'local'
+          ENABLE_BACKEND_ACCESS_CONTROL: 'false'
           LLM_MODEL: ${{ secrets.LLM_MODEL }}
           LLM_ENDPOINT: ${{ secrets.LLM_ENDPOINT }}
           LLM_API_KEY: ${{ secrets.LLM_API_KEY }}

cognee/tests/test_dataset_database_handler.py

@@ -70,12 +70,6 @@ async def main():
     )
     cognee.config.system_root_directory(cognee_directory_path)
 
-    # Create a clean slate for cognee -- reset data and system state
-    print("Resetting cognee data...")
-    await cognee.prune.prune_data()
-    await cognee.prune.prune_system(metadata=True)
-    print("Data reset complete.\n")
-
     # Add custom dataset database handler
     from cognee.infrastructure.databases.dataset_database_handler.use_dataset_database_handler import (
         use_dataset_database_handler,
@@ -86,6 +80,12 @@ async def main():
     )
     use_dataset_database_handler("custom_kuzu_handler", KuzuTestDatasetDatabaseHandler, "kuzu")
 
+    # Create a clean slate for cognee -- reset data and system state
+    print("Resetting cognee data...")
+    await cognee.prune.prune_data()
+    await cognee.prune.prune_system(metadata=True)
+    print("Data reset complete.\n")
+
     # cognee knowledge graph will be created based on this text
     text = """
     Natural language processing (NLP) is an interdisciplinary