diff --git a/cognee/.cognee_system/databases/cognee_db b/cognee/.cognee_system/databases/cognee_db deleted file mode 100644 index 60455ad29..000000000 Binary files a/cognee/.cognee_system/databases/cognee_db and /dev/null differ diff --git a/cognee/tests/integration/run_toy_tasks/conftest.py b/cognee/tests/integration/run_toy_tasks/conftest.py new file mode 100644 index 000000000..e0234162b --- /dev/null +++ b/cognee/tests/integration/run_toy_tasks/conftest.py @@ -0,0 +1,10 @@ + +import os +import pytest + + +@pytest.fixture(autouse=True, scope="session") +def copy_cognee_db_to_target_location(): + + os.system + yield diff --git a/cognee/tests/integration/run_toy_tasks/data/cognee_db b/cognee/tests/integration/run_toy_tasks/data/cognee_db new file mode 100644 index 000000000..e69de29bb diff --git a/cognee/tests/unit/run_tasks/test_run_tasks.py b/cognee/tests/integration/run_toy_tasks/test_run_tasks.py similarity index 94% rename from cognee/tests/unit/run_tasks/test_run_tasks.py rename to cognee/tests/integration/run_toy_tasks/test_run_tasks.py index e76b8440b..2b510a306 100644 --- a/cognee/tests/unit/run_tasks/test_run_tasks.py +++ b/cognee/tests/integration/run_toy_tasks/test_run_tasks.py @@ -5,7 +5,6 @@ from unittest.mock import patch from cognee.modules.pipelines.operations.run_tasks import run_tasks from cognee.modules.pipelines.tasks.Task import Task from cognee.modules.users.methods.get_default_user import get_default_user -from cognee.tests.unit.utils.get_mock_user import get_mock_user async def run_and_check_tasks(): diff --git a/cognee/tests/unit/run_tasks/test_run_tasks_from_queue.py b/cognee/tests/integration/run_toy_tasks/test_run_tasks_from_queue.py similarity index 100% rename from cognee/tests/unit/run_tasks/test_run_tasks_from_queue.py rename to cognee/tests/integration/run_toy_tasks/test_run_tasks_from_queue.py diff --git a/cognee/tests/unit/run_tasks/conftest.py b/cognee/tests/unit/run_tasks/conftest.py index 2cb56e933..3c804227e 100644 --- a/cognee/tests/unit/run_tasks/conftest.py +++ b/cognee/tests/unit/run_tasks/conftest.py @@ -1,28 +1,7 @@ -import sys -import warnings -from unittest.mock import patch +import os import pytest -from cognee.modules.users.methods.get_default_user import get_default_user -from cognee.tests.unit.utils.get_mock_user import get_mock_user - - -def apply_with_to_item(module_items, fn, fn_str): - for i, (name, module) in enumerate(module_items): - if hasattr(module, fn_str) and not "test" in name: - with patch(name, fn): - if len(module_items[(i + 1) :]) > 0: - apply_with_to_item(module_items[(i + 1) :], fn, fn_str) - - @pytest.fixture(autouse=True, scope="session") -def set_get_mock_user_wrapper(): - - def get_mock_user_wrapper(): - return get_mock_user(None, None) - - module_items = list(sys.modules.items()) - apply_with_to_item(module_items, get_mock_user_wrapper(), "get_default_user") - - yield +def copy_cognee_db_to_target_location(): + os.system("mv cognee/tests/integration/run_toy_tasks/data/cognee_db cognee/.cognee_system/databases") diff --git a/cognee/tests/unit/utils/get_mock_user.py b/cognee/tests/unit/utils/get_mock_user.py deleted file mode 100644 index 76ce62cd0..000000000 --- a/cognee/tests/unit/utils/get_mock_user.py +++ /dev/null @@ -1,45 +0,0 @@ -import warnings -from typing import List, Optional -from unittest.mock import Mock, create_autospec -from uuid import UUID, uuid4 - -from cognee.modules.users.models import User - - -def get_mock_user( - user_id: Optional[UUID] = None, - groups: Optional[List["Group"]] = None, - **additional_attributes -) -> Mock: - """ - Creates a mock User instance with configurable attributes. - - Args: - user_id: Optional UUID for the user. Generates random UUID if not provided. - groups: Optional list of group mocks to associate with the user. - **additional_attributes: Any additional attributes to set on the mock user. - - Returns: - Mock: A configured mock User instance. - """ - warnings.warn("\n\n\n---------------USING MOCK USER--------------------\n\n\n") - - # Generate a random UUID if none provided - user_id = user_id or uuid4() - - # Create base mock - mock_user = create_autospec(User, instance=True) - - # Configure basic attributes - mock_user.id = user_id - mock_user.__tablename__ = "users" - mock_user.groups = groups or [] - - # Set polymorphic identity - mock_user.__mapper_args__ = {"polymorphic_identity": "user"} - - # Add any additional attributes - for attr_name, attr_value in additional_attributes.items(): - setattr(mock_user, attr_name, attr_value) - - return mock_user