def test_worker_prerun_guid_exists(monkeypatch, mocker: MockerFixture,
two_unique_uuid4):
"""
Tests that GUID is set to the GUID if a GUID exists in the task object.
"""
mock_task = mocker.Mock()
mock_task.request = {'Correlation-ID': '704ae5472cae4f8daa8f2cc5a5a8mock'}
mocked_settings = deepcopy(django_settings.DJANGO_GUID)
mocked_settings['INTEGRATIONS'] = [CeleryIntegration(log_parent=False)]
with override_settings(DJANGO_GUID=mocked_settings):
settings = Settings()
monkeypatch.setattr('django_guid.integrations.celery.signals.settings',
settings)
worker_prerun(mock_task)
assert get_guid() == '704ae5472cae4f8daa8f2cc5a5a8mock'
def test_worker_prerun_guid_does_not_exist(monkeypatch, mocker: MockerFixture,
mock_uuid):
"""
Tests that a GUID is set if it does not exist
"""
mock_task = mocker.Mock()
mock_task.request = {'Correlation-ID': None}
mocked_settings = deepcopy(django_settings.DJANGO_GUID)
mocked_settings['INTEGRATIONS'] = [CeleryIntegration(log_parent=False)]
with override_settings(DJANGO_GUID=mocked_settings):
settings = Settings()
monkeypatch.setattr('django_guid.integrations.celery.signals.settings',
settings)
worker_prerun(mock_task)
assert get_guid() == '704ae5472cae4f8daa8f2cc5a5a8mock'
def test_worker_prerun_guid_log_parent_with_origin(monkeypatch,
mocker: MockerFixture,
mock_uuid_two_unique):
"""
Tests that depth works when there is an origin
"""
from django_guid.integrations.celery.signals import parent_header
mock_task = mocker.Mock()
mock_task.request = {
'Correlation-ID': None,
parent_header: '1234'
} # No origin
mocked_settings = deepcopy(django_settings.DJANGO_GUID)
mocked_settings['INTEGRATIONS'] = [CeleryIntegration(log_parent=True)]
with override_settings(DJANGO_GUID=mocked_settings):
settings = Settings()
monkeypatch.setattr('django_guid.integrations.celery.signals.settings',
settings)
worker_prerun(mock_task)
assert get_guid() == '704ae5472cae4f8daa8f2cc5a5a8mock'
assert celery_current.get() == 'c494886651cd4baaa8654e4d24a8mock'
assert celery_parent.get() == '1234'