style: apply ruff auto-fix

- Auto-formatted code with ruff format
- Applied ruff linting fixes with --fix

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
github-actions[bot]
2026-02-14 00:05:57 +00:00
parent 331b3a3a1a
commit fc515ebb70
2 changed files with 37 additions and 25 deletions
+6 -4
View File
@@ -371,7 +371,6 @@ class TestAzureTestConnectionIntegration:
data = response.json() data = response.json()
assert "status" in data assert "status" in data
@patch("app.api.azure.settings") @patch("app.api.azure.settings")
@patch("app.api.azure.logger") @patch("app.api.azure.logger")
@pytest.mark.asyncio @pytest.mark.asyncio
@@ -388,7 +387,7 @@ class TestAzureTestConnectionIntegration:
# Verify warning was logged # Verify warning was logged
mock_logger.warning.assert_called_once() mock_logger.warning.assert_called_once()
assert "configuration is incomplete" in mock_logger.warning.call_args[0][0].lower() assert "configuration is incomplete" in mock_logger.warning.call_args[0][0].lower()
assert result["status"] == "error" assert result["status"] == "error"
@patch("app.api.azure.DocumentIntelligenceAdministrationClient") @patch("app.api.azure.DocumentIntelligenceAdministrationClient")
@@ -518,6 +517,7 @@ class TestAzureTestConnectionIntegration:
async def test_azure_connection_logs_outer_exception(self, mock_logger, mock_settings): async def test_azure_connection_logs_outer_exception(self, mock_logger, mock_settings):
"""Test that exceptions in outer try block are logged with exception().""" """Test that exceptions in outer try block are logged with exception()."""
from unittest.mock import PropertyMock from unittest.mock import PropertyMock
from app.api.azure import test_azure_connection from app.api.azure import test_azure_connection
# Trigger an exception in the outer try block # Trigger an exception in the outer try block
@@ -531,7 +531,7 @@ class TestAzureTestConnectionIntegration:
# Should catch the exception and return error # Should catch the exception and return error
assert result["status"] == "error" assert result["status"] == "error"
assert "unexpected error" in result["message"].lower() assert "unexpected error" in result["message"].lower()
# Verify exception was logged with logger.exception # Verify exception was logged with logger.exception
mock_logger.exception.assert_called_once() mock_logger.exception.assert_called_once()
@@ -545,6 +545,7 @@ class TestAzureTestConnectionIntegration:
): ):
"""Test that warning is logged when operations parsing fails.""" """Test that warning is logged when operations parsing fails."""
from unittest.mock import PropertyMock from unittest.mock import PropertyMock
from app.api.azure import test_azure_connection from app.api.azure import test_azure_connection
mock_settings.azure_endpoint = "https://test.cognitiveservices.azure.com/" mock_settings.azure_endpoint = "https://test.cognitiveservices.azure.com/"
@@ -575,12 +576,13 @@ class TestAzureTestConnectionIntegration:
async def test_azure_connection_outer_exception_handler(self, mock_settings): async def test_azure_connection_outer_exception_handler(self, mock_settings):
"""Test the outer exception handler catches unexpected errors.""" """Test the outer exception handler catches unexpected errors."""
from unittest.mock import PropertyMock from unittest.mock import PropertyMock
from app.api.azure import test_azure_connection from app.api.azure import test_azure_connection
# Create a mock that raises exception when azure_endpoint is accessed using PropertyMock # Create a mock that raises exception when azure_endpoint is accessed using PropertyMock
type(mock_settings).azure_endpoint = PropertyMock(side_effect=RuntimeError("Outer error")) type(mock_settings).azure_endpoint = PropertyMock(side_effect=RuntimeError("Outer error"))
type(mock_settings).azure_ai_key = PropertyMock(return_value="test-key") type(mock_settings).azure_ai_key = PropertyMock(return_value="test-key")
mock_request = Mock() mock_request = Mock()
result = await test_azure_connection(mock_request) result = await test_azure_connection(mock_request)
+31 -21
View File
@@ -280,7 +280,7 @@ class TestTaskLogCollector:
collector = TaskLogCollector() collector = TaskLogCollector()
# Don't set a formatter to trigger an edge case # Don't set a formatter to trigger an edge case
logger = logging.getLogger("test_exception") logger = logging.getLogger("test_exception")
logger.addHandler(collector) logger.addHandler(collector)
logger.setLevel(logging.DEBUG) logger.setLevel(logging.DEBUG)
@@ -292,7 +292,7 @@ class TestTaskLogCollector:
# Should handle gracefully # Should handle gracefully
except Exception: except Exception:
pytest.fail("Collector should handle exceptions gracefully") pytest.fail("Collector should handle exceptions gracefully")
logger.removeHandler(collector) logger.removeHandler(collector)
@@ -313,16 +313,18 @@ class TestLogTaskProgressWithFileProcessingStep:
# Setup mocks # Setup mocks
mock_db = MagicMock() mock_db = MagicMock()
mock_session_local.return_value.__enter__.return_value = mock_db mock_session_local.return_value.__enter__.return_value = mock_db
# No existing step record # No existing step record
mock_db.query.return_value.filter.return_value.first.return_value = None mock_db.query.return_value.filter.return_value.first.return_value = None
# Mock datetime # Mock datetime
from datetime import datetime as dt, timezone from datetime import datetime as dt
from datetime import timezone
mock_now = dt(2024, 1, 1, 12, 0, 0, tzinfo=timezone.utc) mock_now = dt(2024, 1, 1, 12, 0, 0, tzinfo=timezone.utc)
mock_datetime.now.return_value = mock_now mock_datetime.now.return_value = mock_now
mock_datetime.timezone = timezone mock_datetime.timezone = timezone
# Mock FileProcessingStep creation # Mock FileProcessingStep creation
mock_step = Mock() mock_step = Mock()
mock_file_step.return_value = mock_step mock_file_step.return_value = mock_step
@@ -356,14 +358,16 @@ class TestLogTaskProgressWithFileProcessingStep:
mock_db = MagicMock() mock_db = MagicMock()
mock_session_local.return_value.__enter__.return_value = mock_db mock_session_local.return_value.__enter__.return_value = mock_db
mock_db.query.return_value.filter.return_value.first.return_value = None mock_db.query.return_value.filter.return_value.first.return_value = None
from datetime import datetime as dt, timezone from datetime import datetime as dt
from datetime import timezone
mock_now = dt(2024, 1, 1, 12, 0, 0, tzinfo=timezone.utc) mock_now = dt(2024, 1, 1, 12, 0, 0, tzinfo=timezone.utc)
mock_datetime.now.return_value = mock_now mock_datetime.now.return_value = mock_now
mock_datetime.timezone = timezone mock_datetime.timezone = timezone
mock_step = Mock() mock_step = Mock()
mock_file_step.return_value = mock_step mock_file_step.return_value = mock_step
@@ -393,14 +397,16 @@ class TestLogTaskProgressWithFileProcessingStep:
mock_db = MagicMock() mock_db = MagicMock()
mock_session_local.return_value.__enter__.return_value = mock_db mock_session_local.return_value.__enter__.return_value = mock_db
mock_db.query.return_value.filter.return_value.first.return_value = None mock_db.query.return_value.filter.return_value.first.return_value = None
from datetime import datetime as dt, timezone from datetime import datetime as dt
from datetime import timezone
mock_now = dt(2024, 1, 1, 12, 0, 0, tzinfo=timezone.utc) mock_now = dt(2024, 1, 1, 12, 0, 0, tzinfo=timezone.utc)
mock_datetime.now.return_value = mock_now mock_datetime.now.return_value = mock_now
mock_datetime.timezone = timezone mock_datetime.timezone = timezone
mock_step = Mock() mock_step = Mock()
mock_file_step.return_value = mock_step mock_file_step.return_value = mock_step
@@ -428,13 +434,15 @@ class TestLogTaskProgressWithFileProcessingStep:
mock_db = MagicMock() mock_db = MagicMock()
mock_session_local.return_value.__enter__.return_value = mock_db mock_session_local.return_value.__enter__.return_value = mock_db
# Existing step without started_at # Existing step without started_at
mock_existing_step = Mock() mock_existing_step = Mock()
mock_existing_step.started_at = None mock_existing_step.started_at = None
mock_db.query.return_value.filter.return_value.first.return_value = mock_existing_step mock_db.query.return_value.filter.return_value.first.return_value = mock_existing_step
from datetime import datetime as dt, timezone from datetime import datetime as dt
from datetime import timezone
mock_now = dt(2024, 1, 1, 12, 0, 0, tzinfo=timezone.utc) mock_now = dt(2024, 1, 1, 12, 0, 0, tzinfo=timezone.utc)
mock_datetime.now.return_value = mock_now mock_datetime.now.return_value = mock_now
mock_datetime.timezone = timezone mock_datetime.timezone = timezone
@@ -462,12 +470,14 @@ class TestLogTaskProgressWithFileProcessingStep:
mock_db = MagicMock() mock_db = MagicMock()
mock_session_local.return_value.__enter__.return_value = mock_db mock_session_local.return_value.__enter__.return_value = mock_db
mock_existing_step = Mock() mock_existing_step = Mock()
mock_existing_step.started_at = None mock_existing_step.started_at = None
mock_db.query.return_value.filter.return_value.first.return_value = mock_existing_step mock_db.query.return_value.filter.return_value.first.return_value = mock_existing_step
from datetime import datetime as dt, timezone from datetime import datetime as dt
from datetime import timezone
mock_now = dt(2024, 1, 1, 12, 0, 0, tzinfo=timezone.utc) mock_now = dt(2024, 1, 1, 12, 0, 0, tzinfo=timezone.utc)
mock_datetime.now.return_value = mock_now mock_datetime.now.return_value = mock_now
mock_datetime.timezone = timezone mock_datetime.timezone = timezone
@@ -511,7 +521,7 @@ class TestLogTaskProgressWithFileProcessingStep:
# Verify collector was used # Verify collector was used
mock_ensure.assert_called_once() mock_ensure.assert_called_once()
mock_collector.drain.assert_called_once_with("task-with-logs") mock_collector.drain.assert_called_once_with("task-with-logs")
# Verify detail was set from collected logs # Verify detail was set from collected logs
call_kwargs = mock_processing_log.call_args[1] call_kwargs = mock_processing_log.call_args[1]
assert call_kwargs["detail"] == "Buffered log line 1\nBuffered log line 2" assert call_kwargs["detail"] == "Buffered log line 1\nBuffered log line 2"