def test_start__terminate(self):
worker = self.worker
worker.blueprint.shutdown_complete.set()
worker.blueprint.started = 5
worker.blueprint.state = RUN
worker.steps = [MockStep() for _ in range(5)]
worker.start()
for w in worker.steps[:3]:
w.start.assert_called()
assert worker.blueprint.started == len(worker.steps)
assert worker.blueprint.state == RUN
worker.terminate()
for step in worker.steps:
step.terminate.assert_called()
worker.blueprint.state = TERMINATE
worker.terminate()
def test_start__terminate(self):
worker = self.worker
worker.blueprint.shutdown_complete.set()
worker.blueprint.started = 5
worker.blueprint.state = RUN
worker.steps = [MockStep() for _ in range(5)]
worker.start()
for w in worker.steps[:3]:
w.start.assert_called()
assert worker.blueprint.started == len(worker.steps)
assert worker.blueprint.state == RUN
worker.terminate()
for step in worker.steps:
step.terminate.assert_called()
worker.blueprint.state = TERMINATE
worker.terminate()