class TestKillManager(TestCase): def setUp(self): self.kill_manager = KillManager() def test_status(self): reason = 'reason' assert self.kill_manager.status is None self.kill_manager.kill(reason) result = self.kill_manager.status assert result.reason == reason assert result.status == mesos_pb2.TaskState.Value('TASK_KILLED') def test_name(self): assert self.kill_manager.name() == 'kill_manager' def test_kill(self): reason = 'reason' assert self.kill_manager._killed is False self.kill_manager.kill(reason) assert self.kill_manager._reason == reason assert self.kill_manager._killed is True
def setUp(self): self.kill_manager = KillManager()