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

ruff format

Browse source
This commit is contained in:
Daulet Amirkhanov 2025-09-03 16:51:30 +01:00
parent 6fe2771421
commit b9dad5f01d
2 changed files with 18 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -31,9 +31,8 @@ def mock_authenticated_user():
tenant_id=uuid4(),
)
gau_mod = importlib.import_module(
"cognee.modules.users.methods.get_authenticated_user"
)
gau_mod = importlib.import_module("cognee.modules.users.methods.get_authenticated_user")
class TestConditionalAuthenticationEndpoints:
@ -76,7 +75,7 @@ class TestConditionalAuthenticationEndpoints:
assert "CookieAuth" in security_schemes
@patch("cognee.api.v1.add.add")
@patch.object(gau_mod, 'get_default_user', new_callable=AsyncMock)
@patch.object(gau_mod, "get_default_user", new_callable=AsyncMock)
@patch(
"cognee.api.client.REQUIRE_AUTHENTICATION",
False,
@ -101,7 +100,7 @@ class TestConditionalAuthenticationEndpoints:
# Core test: authentication is not required (should not get 401)
assert response.status_code != 401
@patch.object(gau_mod, 'get_default_user', new_callable=AsyncMock)
@patch.object(gau_mod, "get_default_user", new_callable=AsyncMock)
@patch(
"cognee.api.client.REQUIRE_AUTHENTICATION",
False,
@ -143,7 +142,7 @@ class TestConditionalAuthenticationBehavior:
("/api/v1/datasets", "GET"),
],
)
@patch.object(gau_mod, 'get_default_user', new_callable=AsyncMock)
@patch.object(gau_mod, "get_default_user", new_callable=AsyncMock)
def test_get_endpoints_work_without_auth(
self, mock_get_default, client, endpoint, method, mock_default_user
):
@ -170,14 +169,11 @@ class TestConditionalAuthenticationBehavior:
except Exception:
pass # If response is not JSON, that's fine
gsm_mod = importlib.import_module("cognee.modules.settings.get_settings")
gsm_mod = importlib.import_module(
"cognee.modules.settings.get_settings"
)
@patch.object(gsm_mod, 'get_vectordb_config')
@patch.object(gsm_mod, 'get_llm_config')
@patch.object(gau_mod, 'get_default_user', new_callable=AsyncMock)
@patch.object(gsm_mod, "get_vectordb_config")
@patch.object(gsm_mod, "get_llm_config")
@patch.object(gau_mod, "get_default_user", new_callable=AsyncMock)
def test_settings_endpoint_integration(
self, mock_get_default, mock_llm_config, mock_vector_config, client, mock_default_user
):
@ -215,7 +211,7 @@ class TestConditionalAuthenticationErrorHandling:
def client(self):
return TestClient(app)
@patch.object(gau_mod, 'get_default_user', new_callable=AsyncMock)
@patch.object(gau_mod, "get_default_user", new_callable=AsyncMock)
def test_get_default_user_fails(self, mock_get_default, client):
"""Test behavior when get_default_user fails (with current environment)."""
mock_get_default.side_effect = Exception("Database connection failed")

17
cognee/tests/unit/modules/users/test_conditional_authentication.py
View file

@ -10,16 +10,14 @@ import importlib
from cognee.modules.users.models import User
gau_mod = importlib.import_module(
"cognee.modules.users.methods.get_authenticated_user"
)
gau_mod = importlib.import_module("cognee.modules.users.methods.get_authenticated_user")
class TestConditionalAuthentication:
"""Test cases for conditional authentication functionality."""
@pytest.mark.asyncio
@patch.object(gau_mod, 'get_default_user', new_callable=AsyncMock)
@patch.object(gau_mod, "get_default_user", new_callable=AsyncMock)
async def test_require_authentication_false_no_token_returns_default_user(
self, mock_get_default
):
@ -37,7 +35,7 @@ class TestConditionalAuthentication:
mock_get_default.assert_called_once()
@pytest.mark.asyncio
@patch.object(gau_mod, 'get_default_user', new_callable=AsyncMock)
@patch.object(gau_mod, "get_default_user", new_callable=AsyncMock)
async def test_require_authentication_false_with_valid_user_returns_user(
self, mock_get_default
):
@ -59,7 +57,7 @@ class TestConditionalAuthentication:
mock_get_default.assert_not_called()
@pytest.mark.asyncio
@patch.object(gau_mod, 'get_default_user', new_callable=AsyncMock)
@patch.object(gau_mod, "get_default_user", new_callable=AsyncMock)
async def test_require_authentication_true_with_user_returns_user(self, mock_get_default):
"""Test that when REQUIRE_AUTHENTICATION=true and user present, returns user."""
mock_authenticated_user = User(
@ -128,6 +126,7 @@ class TestConditionalAuthenticationEnvironmentVariables:
from cognee.modules.users.methods.get_authenticated_user import (
REQUIRE_AUTHENTICATION,
)
importlib.invalidate_caches()
assert not REQUIRE_AUTHENTICATION
@ -195,7 +194,7 @@ class TestConditionalAuthenticationEdgeCases:
"""Test edge cases and error scenarios."""
@pytest.mark.asyncio
@patch.object(gau_mod, 'get_default_user', new_callable=AsyncMock)
@patch.object(gau_mod, "get_default_user", new_callable=AsyncMock)
async def test_get_default_user_raises_exception(self, mock_get_default):
"""Test behavior when get_default_user raises an exception."""
mock_get_default.side_effect = Exception("Database error")
@ -205,7 +204,7 @@ class TestConditionalAuthenticationEdgeCases:
await gau_mod.get_authenticated_user(user=None)
@pytest.mark.asyncio
@patch.object(gau_mod, 'get_default_user', new_callable=AsyncMock)
@patch.object(gau_mod, "get_default_user", new_callable=AsyncMock)
async def test_user_type_consistency(self, mock_get_default):
"""Test that the function always returns the same type."""
mock_user = User(
@ -242,7 +241,7 @@ class TestConditionalAuthenticationEdgeCases:
class TestAuthenticationScenarios:
"""Test specific authentication scenarios that could occur in FastAPI Users."""
@patch.object(gau_mod, 'get_default_user', new_callable=AsyncMock)
@patch.object(gau_mod, "get_default_user", new_callable=AsyncMock)
async def test_fallback_to_default_user_scenarios(self, mock_get_default):
"""
Test fallback to default user for all scenarios where FastAPI Users returns None: