oncall-engine/engine/common/tests/test_task_queue_assignment.py
Yulya Artyukhina e472b03e1d
Remove deprecated alerting integration tasks (#2944)
# What this PR does
These celery tasks have not been used for more than one week (since
[v1.3.25](https://github.com/grafana/oncall/releases/tag/v1.3.25) which
released an improvement for Grafana Alerting integration)

## Checklist

- [ ] Unit, integration, and e2e (if applicable) tests updated
- [x] Documentation added (or `pr:no public docs` PR label added if not
required)
- [x] `CHANGELOG.md` updated (or `pr:no changelog` PR label added if not
required)
2023-09-01 13:11:02 +00:00

33 lines
1.2 KiB
Python

from celery import current_app
from settings.celery_task_routes import CELERY_TASK_ROUTES
"""
If a task has a legitimate reason to not have a queue assignment it can
be added here (In development, in process of deprecation, etc.) if possible
we should avoid @shared_dedicated_queue_retry_task or @shared_task and
remove entirely if it is not needed.
"""
COMMON_IGNORED_TASKS = {
"common.oncall_gateway.tasks.create_slack_connector_async_v2",
}
def check_celery_task_route_mapping(task_ids, ignored_prefixes, additional_ignored_tasks=None):
tasks = set(k for k in current_app.tasks.keys() if not k.startswith(ignored_prefixes))
tasks -= set(COMMON_IGNORED_TASKS)
if additional_ignored_tasks:
tasks -= set(additional_ignored_tasks)
tasks -= set(task_ids)
if tasks:
print(f"Unassigned queue for celery task {tasks}")
assert len(tasks) == 0
def test_celery_task_route_mapping():
"""
If this test does not pass make sure you have added any newly added
@shared_dedicated_queue_retry_task or @shared_task to CELERY_TASK_ROUTES
in engine/settings/celery_task_routes.py
"""
check_celery_task_route_mapping(CELERY_TASK_ROUTES.keys(), ("extensions", "celery"))