def test_scale_celery_task(self, mock_scale_by_runs, mock_scale_by_tasks):
with self.settings(CELERY_SCALE_BY_RUN=True):
scale_celery_task(16000)
mock_scale_by_runs.assert_called_once_with(16000)
with self.settings(CELERY_SCALE_BY_RUN=False):
scale_celery_task(16000)
celery_tasks = get_celery_tasks_scale_by_task()
mock_scale_by_tasks.assert_called_once_with(celery_tasks, 16000)
import os
import celery
import django
from django.core.cache import cache
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "eventkit_cloud.settings.prod")
django.setup()
from eventkit_cloud.celery import app
from eventkit_cloud.tasks.scheduled_tasks import scale_by_runs, scale_celery_task
if __name__ == '__main__':
scale_celery_task(30000)