def test_start__stop(self):
worker = self.worker
worker.blueprint.shutdown_complete.set()
worker.steps = [MockStep(StartStopStep(self)) for _ in range(4)]
worker.blueprint.state = RUN
worker.blueprint.started = 4
for w in worker.steps:
w.start = Mock()
w.close = Mock()
w.stop = Mock()
worker.start()
for w in worker.steps:
w.start.assert_called()
worker.consumer = Mock()
worker.stop(exitcode=3)
for stopstep in worker.steps:
stopstep.close.assert_called()
stopstep.stop.assert_called()
# Doesn't close pool if no pool.
worker.start()
worker.pool = None
worker.stop()
# test that stop of None is not attempted
worker.steps[-1] = None
worker.start()
worker.stop()
def test_start__stop(self):
worker = self.worker
worker.blueprint.shutdown_complete.set()
worker.steps = [MockStep(StartStopStep(self)) for _ in range(4)]
worker.blueprint.state = RUN
worker.blueprint.started = 4
for w in worker.steps:
w.start = Mock()
w.close = Mock()
w.stop = Mock()
worker.start()
for w in worker.steps:
w.start.assert_called()
worker.consumer = Mock()
worker.stop(exitcode=3)
for stopstep in worker.steps:
stopstep.close.assert_called()
stopstep.stop.assert_called()
# Doesn't close pool if no pool.
worker.start()
worker.pool = None
worker.stop()
# test that stop of None is not attempted
worker.steps[-1] = None
worker.start()
worker.stop()