|
@@ -161,14 +161,10 @@ class test_TaskPool(Case):
|
|
|
pool.apply_async(lambda x: x, (2, ), {})
|
|
|
|
|
|
def test_terminate_job(self):
|
|
|
-
|
|
|
- @patch('celery.concurrency.processes._kill')
|
|
|
- def _do_test(_kill):
|
|
|
- pool = TaskPool(10)
|
|
|
- pool.terminate_job(1341)
|
|
|
- _kill.assert_called_with(1341, signal.SIGTERM)
|
|
|
-
|
|
|
- _do_test()
|
|
|
+ pool = TaskPool(10)
|
|
|
+ pool._pool = Mock()
|
|
|
+ pool.terminate_job(1341)
|
|
|
+ pool._pool.terminate_job.assert_called_with(1341, None)
|
|
|
|
|
|
def test_grow_shrink(self):
|
|
|
pool = TaskPool(10)
|