gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: Update unit tests for require auth

Browse source
This commit is contained in:
Igor Ilic 2025-10-30 16:52:04 +01:00
parent 50977db946
commit 9d8430cfb0
1 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
cognee/tests/unit/api/test_conditional_authentication_endpoints.py
View file

@ -1,3 +1,4 @@
import os
import pytest import pytest
from unittest.mock import patch, AsyncMock, MagicMock from unittest.mock import patch, AsyncMock, MagicMock
from uuid import uuid4 from uuid import uuid4
@ -5,8 +6,6 @@ from fastapi.testclient import TestClient
from types import SimpleNamespace from types import SimpleNamespace
import importlib import importlib
from cognee.api.client import app
# Fixtures for reuse across test classes # Fixtures for reuse across test classes
@pytest.fixture @pytest.fixture
@ -32,6 +31,10 @@ def mock_authenticated_user():
) )
# To turn off authentication we need to set the environment variable before importing the module
# Also both require_authentication and backend access control must be false
os.environ["REQUIRE_AUTHENTICATION"] = "false"
os.environ["ENABLE_BACKEND_ACCESS_CONTROL"] = "false"
gau_mod = importlib.import_module("cognee.modules.users.methods.get_authenticated_user") gau_mod = importlib.import_module("cognee.modules.users.methods.get_authenticated_user")
@ -40,6 +43,8 @@ class TestConditionalAuthenticationEndpoints:
@pytest.fixture @pytest.fixture
def client(self): def client(self):
from cognee.api.client import app
"""Create a test client.""" """Create a test client."""
return TestClient(app) return TestClient(app)
@ -133,6 +138,8 @@ class TestConditionalAuthenticationBehavior:
@pytest.fixture @pytest.fixture
def client(self): def client(self):
from cognee.api.client import app
return TestClient(app) return TestClient(app)
@pytest.mark.parametrize( @pytest.mark.parametrize(
@ -209,6 +216,8 @@ class TestConditionalAuthenticationErrorHandling:
@pytest.fixture @pytest.fixture
def client(self): def client(self):
from cognee.api.client import app
return TestClient(app) return TestClient(app)
@patch.object(gau_mod, "get_default_user", new_callable=AsyncMock) @patch.object(gau_mod, "get_default_user", new_callable=AsyncMock)
@ -232,7 +241,7 @@ class TestConditionalAuthenticationErrorHandling:
# The exact error message may vary depending on the actual database connection # The exact error message may vary depending on the actual database connection
# The important thing is that we get a 500 error when user creation fails # The important thing is that we get a 500 error when user creation fails
def test_current_environment_configuration(self): def test_current_environment_configuration(self, client):
"""Test that current environment configuration is working properly.""" """Test that current environment configuration is working properly."""
# This tests the actual module state without trying to change it # This tests the actual module state without trying to change it
from cognee.modules.users.methods.get_authenticated_user import ( from cognee.modules.users.methods.get_authenticated_user import (