Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Commit 346cce5

Browse files
committed
Update celery health check test
1 parent eb793c7 commit 346cce5

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

core/tests/test_tasks.py

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,12 +90,25 @@ def test_can_async_task(self):
9090
"b": 2,
9191
"raise_": False,
9292
"wait": 3,
93-
"allow_prod_db": False,
93+
"allow_prod_db": True,
9494
}
9595
)
9696
self.assertTrue(ok)
9797
self.assertEqual(result.get(), 3)
9898

99+
def test_async_check_quits_when_allow_prod_db_is_false(self):
100+
ok, result = health_check.submit_task(
101+
kwargs={
102+
"a": 1,
103+
"b": 2,
104+
"raise_": False,
105+
"wait": 3,
106+
"allow_prod_db": False,
107+
}
108+
)
109+
self.assertTrue(ok)
110+
self.assertIsNone(result.get(), None)
111+
99112

100113
class TestSendMailTask(TestCase):
101114
def test_can_send_to_standalone_email(self):

0 commit comments

Comments
 (0)