1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
|
import operator
from unittest.mock import Mock
from celery import signals
from celery.concurrency import solo
from celery.utils.functional import noop
class test_solo_TaskPool:
def test_on_start(self):
x = solo.TaskPool()
x.on_start()
def test_on_apply(self):
x = solo.TaskPool()
x.on_start()
x.on_apply(operator.add, (2, 2), {}, noop, noop)
def test_info(self):
x = solo.TaskPool()
x.on_start()
assert x.info
def test_on_worker_process_init_called(self):
"""Upon the initialization of a new solo worker pool a worker_process_init
signal should be emitted"""
on_worker_process_init = Mock()
signals.worker_process_init.connect(on_worker_process_init)
solo.TaskPool()
assert on_worker_process_init.call_count == 1
|