mirror of
https://github.com/quay/quay.git
synced 2026-01-26 06:21:37 +03:00
a06cc6fa43
* chore(pre-commit): match black version with requirements-dev * run `make black` against repo * ci: switch to black 24.4.2 * fix: py312 * fix: flake8 errors * fix: flake8 conflicts * chore: add git blame ignore revs file
112 lines
3.6 KiB
Python
112 lines
3.6 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()
|
|
|
|
call_args = mock_sentry_sdk.init.call_args
|
|
assert call_args is not None
|
|
|
|
kwargs = call_args.kwargs
|
|
assert kwargs["dsn"] == "https://test@sentry.io/123"
|
|
assert kwargs["environment"] == "production" # default
|
|
assert kwargs["traces_sample_rate"] == 0.1 # default
|
|
assert kwargs["profiles_sample_rate"] == 0.1 # default
|
|
|
|
# Verify the SDK was called exactly once
|
|
assert mock_sentry_sdk.init.call_count == 1
|