pengjunjie/langflow
1
0
Fork 1
mirror of https://github.com/langflow-ai/langflow.git synced 2026-07-24 05:39:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update test_security_cors.py

Browse Source
This commit is contained in:
Eric Hare
2026-04-09 12:59:13 -07:00
parent 387f5f997f
commit 04b3b9be49
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/backend/tests/unit/test_security_cors.py
View File

@ -280,6 +280,7 @@ class TestRefreshTokenSecurity:
"""Test that valid refresh tokens work correctly."""
from uuid import uuid4
from langflow.services.auth.service import AuthService
from langflow.services.auth.utils import create_refresh_token
mock_db = AsyncMock()
@ -288,7 +289,14 @@ class TestRefreshTokenSecurity:
user_id = uuid4()
mock_user.id = user_id
with patch("langflow.services.auth.service.jwt.decode") as mock_decode:
# Create a langflow AuthService instance (not lfx) with mocked settings
mock_settings_service = MagicMock()
auth_service = AuthService(mock_settings_service)
with (
patch("langflow.services.auth.utils._auth_service", return_value=auth_service),
patch("langflow.services.auth.service.jwt.decode") as mock_decode,
):
mock_decode.return_value = {"sub": str(user_id), "type": "refresh"} # Correct type
with patch("langflow.services.auth.utils.get_jwt_verification_key") as mock_verification_key:
@ -297,9 +305,7 @@ class TestRefreshTokenSecurity:
with patch("langflow.services.auth.service.get_user_by_id", new_callable=AsyncMock) as mock_get_user:
mock_get_user.return_value = mock_user
with patch(
"langflow.services.auth.service.AuthService.create_user_tokens", new_callable=AsyncMock
) as mock_create_tokens:
with patch.object(auth_service, "create_user_tokens", new_callable=AsyncMock) as mock_create_tokens:
expected_access = "new-access-token"
expected_refresh = "new-refresh-token"
mock_create_tokens.return_value = {