mirror of
https://github.com/quay/quay.git
synced 2025-11-17 23:02:34 +03:00
22290c9812
* Updated raven dependency to be compatible with python 3.12 --------- Co-authored-by: shudeshp <shudeshp@redhat.com>
101 lines
3.3 KiB
Python
101 lines
3.3 KiB
Python
from unittest.mock import MagicMock, patch
|
|
|
|
from workers.worker import Worker
|
|
|
|
|
|
def test_worker_name_defined_outside_sentry_condition():
|
|
"""Test that worker_name is defined regardless of Sentry configuration."""
|
|
with patch("workers.worker.app") as mock_app:
|
|
mock_app.config.get.return_value = "FakeSentry" # Disable Sentry
|
|
|
|
worker = Worker()
|
|
|
|
# The worker_name should be available even when Sentry is disabled
|
|
# We can't directly access it since it's local to __init__, but we can
|
|
# verify that the Worker initializes without errors
|
|
assert worker is not None
|
|
|
|
|
|
def test_sentry_not_initialized_when_disabled():
|
|
"""Test that Sentry is not initialized when disabled."""
|
|
with patch("workers.worker.app") as mock_app, patch(
|
|
"workers.worker.sentry_sdk"
|
|
) as mock_sentry_sdk:
|
|
|
|
# Configure Sentry to be disabled
|
|
mock_app.config.get.side_effect = lambda key, default=None: {
|
|
"EXCEPTION_LOG_TYPE": "FakeSentry",
|
|
}.get(key, default)
|
|
|
|
worker = Worker()
|
|
|
|
# Verify Sentry was not initialized
|
|
mock_sentry_sdk.init.assert_not_called()
|
|
mock_sentry_sdk.set_tag.assert_not_called()
|
|
|
|
|
|
def test_sentry_not_initialized_when_no_dsn():
|
|
"""Test that Sentry is not initialized when DSN is empty."""
|
|
with patch("workers.worker.app") as mock_app, patch(
|
|
"workers.worker.sentry_sdk"
|
|
) as mock_sentry_sdk:
|
|
|
|
mock_app.config.get.side_effect = lambda key, default=None: {
|
|
"EXCEPTION_LOG_TYPE": "Sentry",
|
|
"SENTRY_DSN": "",
|
|
}.get(key, default)
|
|
|
|
worker = Worker()
|
|
|
|
# Verify Sentry was not initialized
|
|
mock_sentry_sdk.init.assert_not_called()
|
|
mock_sentry_sdk.set_tag.assert_not_called()
|
|
|
|
|
|
def test_exception_capture_in_operation():
|
|
"""Test that exceptions are captured by Sentry in operations."""
|
|
with patch("workers.worker.app") as mock_app, patch(
|
|
"workers.worker.sentry_sdk"
|
|
) as mock_sentry_sdk, patch("workers.worker.UseThenDisconnect") as mock_use_then_disconnect:
|
|
|
|
mock_app.config.get.side_effect = lambda key, default=None: {
|
|
"EXCEPTION_LOG_TYPE": "Sentry",
|
|
"SENTRY_DSN": "https://test@sentry.io/123",
|
|
}.get(key, default)
|
|
|
|
worker = Worker()
|
|
|
|
def failing_operation():
|
|
raise ValueError("Test exception")
|
|
|
|
worker.add_operation(failing_operation, 60)
|
|
|
|
wrapped_operation = worker._operations[0][0]
|
|
|
|
wrapped_operation()
|
|
|
|
mock_sentry_sdk.capture_exception.assert_called_once()
|
|
|
|
|
|
def test_default_sentry_config_values():
|
|
"""Test that default Sentry configuration values are used when not specified."""
|
|
with patch("workers.worker.app") as mock_app, patch(
|
|
"workers.worker.sentry_sdk"
|
|
) as mock_sentry_sdk, patch("socket.gethostname") as mock_hostname:
|
|
|
|
mock_app.config.get.side_effect = lambda key, default=None: {
|
|
"EXCEPTION_LOG_TYPE": "Sentry",
|
|
"SENTRY_DSN": "https://test@sentry.io/123",
|
|
}.get(key, default)
|
|
|
|
mock_hostname.return_value = "test-host"
|
|
|
|
worker = Worker()
|
|
|
|
mock_sentry_sdk.init.assert_called_once_with(
|
|
dsn="https://test@sentry.io/123",
|
|
environment="production", # default
|
|
traces_sample_rate=0.1, # default
|
|
profiles_sample_rate=0.1, # default
|
|
)
|