fix pre-commit
This commit is contained in:
@@ -46,7 +46,9 @@ class CeleryModuleTest(SimpleTestCase):
|
||||
)
|
||||
app_mock.autodiscover_tasks.assert_called_once_with()
|
||||
self.assertEqual(module.app, app_mock)
|
||||
self.assertIn("parse-industrial-production-daily", module.app.conf.beat_schedule)
|
||||
self.assertIn(
|
||||
"parse-industrial-production-daily", module.app.conf.beat_schedule
|
||||
)
|
||||
self.assertIn("parse-manufactures-daily", module.app.conf.beat_schedule)
|
||||
self.assertIn("parse-industrial-products-daily", module.app.conf.beat_schedule)
|
||||
self.assertIn("parse-inspections-weekly", module.app.conf.beat_schedule)
|
||||
@@ -63,12 +65,16 @@ class CeleryModuleTest(SimpleTestCase):
|
||||
apply_async=apply_async_mock
|
||||
)
|
||||
|
||||
with patch.object(module, "settings", SimpleNamespace(
|
||||
CELERY_STARTUP_REFRESH_ENABLED=True,
|
||||
CELERY_STARTUP_REFRESH_LOCK_KEY="startup-lock",
|
||||
CELERY_STARTUP_REFRESH_LOCK_TTL_SECONDS=120,
|
||||
CELERY_STARTUP_REFRESH_DELAY_SECONDS=45,
|
||||
)), patch.object(module.cache, "add", return_value=True) as add_mock, patch.dict(
|
||||
with patch.object(
|
||||
module,
|
||||
"settings",
|
||||
SimpleNamespace(
|
||||
CELERY_STARTUP_REFRESH_ENABLED=True,
|
||||
CELERY_STARTUP_REFRESH_LOCK_KEY="startup-lock",
|
||||
CELERY_STARTUP_REFRESH_LOCK_TTL_SECONDS=120,
|
||||
CELERY_STARTUP_REFRESH_DELAY_SECONDS=45,
|
||||
),
|
||||
), patch.object(module.cache, "add", return_value=True) as add_mock, patch.dict(
|
||||
sys.modules, {"apps.parsers.tasks": fake_tasks_module}
|
||||
):
|
||||
module._queue_startup_sources_refresh()
|
||||
@@ -88,12 +94,16 @@ class CeleryModuleTest(SimpleTestCase):
|
||||
apply_async=apply_async_mock
|
||||
)
|
||||
|
||||
with patch.object(module, "settings", SimpleNamespace(
|
||||
CELERY_STARTUP_REFRESH_ENABLED=True,
|
||||
CELERY_STARTUP_REFRESH_LOCK_KEY="startup-lock",
|
||||
CELERY_STARTUP_REFRESH_LOCK_TTL_SECONDS=120,
|
||||
CELERY_STARTUP_REFRESH_DELAY_SECONDS=45,
|
||||
)), patch.object(module.cache, "add", return_value=False), patch.dict(
|
||||
with patch.object(
|
||||
module,
|
||||
"settings",
|
||||
SimpleNamespace(
|
||||
CELERY_STARTUP_REFRESH_ENABLED=True,
|
||||
CELERY_STARTUP_REFRESH_LOCK_KEY="startup-lock",
|
||||
CELERY_STARTUP_REFRESH_LOCK_TTL_SECONDS=120,
|
||||
CELERY_STARTUP_REFRESH_DELAY_SECONDS=45,
|
||||
),
|
||||
), patch.object(module.cache, "add", return_value=False), patch.dict(
|
||||
sys.modules, {"apps.parsers.tasks": fake_tasks_module}
|
||||
):
|
||||
module._queue_startup_sources_refresh()
|
||||
|
||||
Reference in New Issue
Block a user