c889a57b6b
Test Suites / Build CI Environment (push) Has been cancelled
Test Suites / Basic Tests (push) Has been cancelled
Test Suites / End-to-End Tests (push) Has been cancelled
Test Suites / CLI Tests (push) Has been cancelled
Test Suites / Slow End-to-End Tests (push) Has been cancelled
Test Suites / Graph Database Tests (push) Has been cancelled
Test Suites / Vector DB Tests (push) Has been cancelled
Test Suites / Temporal Graph Test (push) Has been cancelled
Test Suites / Search Test on Different DBs (push) Has been cancelled
Test Suites / Example Tests (push) Has been cancelled
Test Suites / Notebook Tests (push) Has been cancelled
Test Suites / OS and Python Tests Ubuntu (push) Has been cancelled
Test Suites / OS and Python Tests Extended (push) Has been cancelled
Test Suites / LLM Test Suite (push) Has been cancelled
Test Suites / S3 File Storage Test (push) Has been cancelled
Test Suites / Run Integration Tests (push) Has been cancelled
Test Suites / MCP Tests (push) Has been cancelled
Test Suites / Docker Compose Test (push) Has been cancelled
Test Suites / Docker CI test (push) Has been cancelled
Test Suites / Relational DB Migration Tests (push) Has been cancelled
Test Suites / Distributed Cognee Test (push) Has been cancelled
Test Suites / DB Examples Tests (push) Has been cancelled
Test Suites / Test Completion Status (push) Has been cancelled
Test Suites / Claude Code Review (push) Has been cancelled
Test Suites / basic checks (push) Has been cancelled
build | Build and Push Cognee MCP Docker Image to dockerhub / docker-build-and-push (push) Has been cancelled
Scorecard supply-chain security / Scorecard analysis (push) Has been cancelled
build | Build and Push Docker Image to dockerhub / docker-build-and-push (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges Core Functionality (3.11) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges Core Functionality (3.12) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges with Different Graph Databases (kuzu, kuzu) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges with Different Graph Databases (neo4j, neo4j) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges Examples (push) Has been cancelled
Weighted Edges Tests / Code Quality for Weighted Edges (push) Has been cancelled
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
import pathlib
|
|
|
|
import pytest
|
|
|
|
import cognee
|
|
from cognee.base_config import get_base_config
|
|
from cognee.infrastructure.databases.cache.config import get_cache_config
|
|
from cognee.infrastructure.databases.cache.get_cache_engine import (
|
|
close_cache_engine,
|
|
create_cache_engine,
|
|
get_cache_engine,
|
|
)
|
|
from cognee.modules.data.deletion.prune_data import prune_data
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_prune_data_closes_cached_fs_cache(tmp_path: pathlib.Path):
|
|
base_config = get_base_config()
|
|
cache_config = get_cache_config()
|
|
previous_data_root_directory = base_config.data_root_directory
|
|
previous_cache_backend = cache_config.cache_backend
|
|
previous_caching = cache_config.caching
|
|
|
|
try:
|
|
await close_cache_engine()
|
|
cognee.config.data_root_directory(str(tmp_path))
|
|
cache_config.cache_backend = "fs"
|
|
cache_config.caching = True
|
|
|
|
cache_engine = get_cache_engine()
|
|
assert cache_engine is not None
|
|
assert (tmp_path / ".cognee_fs_cache" / "sessions_db").exists()
|
|
|
|
await prune_data()
|
|
|
|
assert not tmp_path.exists()
|
|
assert create_cache_engine.cache_info().currsize == 0
|
|
finally:
|
|
await close_cache_engine()
|
|
cognee.config.data_root_directory(previous_data_root_directory)
|
|
cache_config.cache_backend = previous_cache_backend
|
|
cache_config.caching = previous_caching
|