Python KillManager Examples

Example #1

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

Example #2

File: test_kill_manager.py Project: AltanAlpay/aurora

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

Example #3

 def setUp(self):
     self.kill_manager = KillManager()

Example #4

File: test_kill_manager.py Project: AltanAlpay/aurora

 def setUp(self):
   self.kill_manager = KillManager()