class test_Hub(AppCase):
def setup(self):
self.w = Mock(name='w')
self.hub = Hub(self.w)
self.w.hub = Mock(name='w.hub')
@patch('celery.worker.components.set_event_loop')
@patch('celery.worker.components.get_event_loop')
def test_create(self, get_event_loop, set_event_loop):
self.hub._patch_thread_primitives = Mock(name='ptp')
self.assertIs(self.hub.create(self.w), self.hub)
self.hub._patch_thread_primitives.assert_called_with(self.w)
def test_start(self):
self.hub.start(self.w)
def test_stop(self):
self.hub.stop(self.w)
self.w.hub.close.assert_called_with()
def test_terminate(self):
self.hub.terminate(self.w)
self.w.hub.close.assert_called_with()
class test_Hub(AppCase):
def setup(self):
self.w = Mock(name='w')
self.hub = Hub(self.w)
self.w.hub = Mock(name='w.hub')
@patch('celery.worker.components.set_event_loop')
@patch('celery.worker.components.get_event_loop')
def test_create(self, get_event_loop, set_event_loop):
self.hub._patch_thread_primitives = Mock(name='ptp')
self.assertIs(self.hub.create(self.w), self.hub)
self.hub._patch_thread_primitives.assert_called_with(self.w)
def test_start(self):
self.hub.start(self.w)
def test_stop(self):
self.hub.stop(self.w)
self.w.hub.close.assert_called_with()
def test_terminate(self):
self.hub.terminate(self.w)
self.w.hub.close.assert_called_with()
def setup(self):
self.w = Mock(name='w')
self.hub = Hub(self.w)
self.w.hub = Mock(name='w.hub')
def setup(self):
self.w = Mock(name='w')
self.hub = Hub(self.w)
self.w.hub = Mock(name='w.hub')
