Beispiel #1
0
 def test_status_is_terminal(self):
     for terminal_status in ExecutorBase.TERMINAL_STATES:
         assert ExecutorBase.status_is_terminal(terminal_status)
     assert not ExecutorBase.status_is_terminal('RUNNING')
     assert not ExecutorBase.status_is_terminal('BASSCANNON')
Beispiel #2
0
 def setUp(self):
     self.executor_base = ExecutorBase()
Beispiel #3
0
class TestExecutorBase(TestCase):
    def setUp(self):
        self.executor_base = ExecutorBase()

    def test_status_is_terminal(self):
        for terminal_status in ExecutorBase.TERMINAL_STATES:
            assert ExecutorBase.status_is_terminal(terminal_status)
        assert not ExecutorBase.status_is_terminal('RUNNING')
        assert not ExecutorBase.status_is_terminal('BASSCANNON')

    @mock.patch('twitter.common.log.info', spec=log.info)
    def test_log(self, mock_info):
        test_message = 'testing'
        self.executor_base.log(test_message)
        mock_info.assert_called_once_with('Executor [None]: %s' % test_message)

    def test_registered(self):
        driver = ExecutorDriver()
        executor_info = mesos_pb2.ExecutorInfo()
        framework_info = mesos_pb2.FrameworkInfo()
        slave_info = mesos_pb2.SlaveInfo()

        self.executor_base.registered(driver, executor_info, framework_info,
                                      slave_info)
        assert self.executor_base._driver == driver
        assert self.executor_base._executor_info == executor_info
        assert self.executor_base._framework_info == framework_info
        assert self.executor_base._slave_info == slave_info

    def test_reregistered(self):
        driver = ExecutorDriver()
        slave_info = mesos_pb2.SlaveInfo()
        self.executor_base.reregistered(driver, slave_info)

    def test_disconnected(self):
        driver = ExecutorDriver()
        self.executor_base.disconnected(driver)

    @mock.patch('mesos.interface.mesos_pb2.TaskStatus',
                spec=mesos_pb2.TaskStatus)
    def test_send_update(self, MockTaskStatus):
        driver = mock.Mock(ExecutorDriver)
        task_id = 'task_id'
        state = mesos_pb2.TASK_RUNNING
        message = 'test_message'
        self.executor_base.send_update(driver, task_id, state, message)
        driver.sendStatusUpdate.assert_called_once_with(
            MockTaskStatus.return_value)
        assert MockTaskStatus.return_value.state == state
        assert MockTaskStatus.return_value.task_id.value == task_id
        assert MockTaskStatus.return_value.message == message

    def test_frameworkMessage(self):
        driver = ExecutorDriver()
        self.executor_base.frameworkMessage(driver, 'test_message')

    def test_error(self):
        driver = ExecutorDriver()
        self.executor_base.error(driver, 'message')
 def test_status_is_terminal(self):
   for terminal_status in ExecutorBase.TERMINAL_STATES:
     assert ExecutorBase.status_is_terminal(terminal_status)
   assert not ExecutorBase.status_is_terminal('RUNNING')
   assert not ExecutorBase.status_is_terminal('BASSCANNON')
 def setUp(self):
   self.executor_base = ExecutorBase()
class TestExecutorBase(TestCase):
  def setUp(self):
    self.executor_base = ExecutorBase()

  def test_status_is_terminal(self):
    for terminal_status in ExecutorBase.TERMINAL_STATES:
      assert ExecutorBase.status_is_terminal(terminal_status)
    assert not ExecutorBase.status_is_terminal('RUNNING')
    assert not ExecutorBase.status_is_terminal('BASSCANNON')

  @mock.patch('twitter.common.log.info', spec=log.info)
  def test_log(self, mock_info):
    test_message = 'testing'
    self.executor_base.log(test_message)
    mock_info.assert_called_once_with('Executor [None]: %s' % test_message)

  def test_registered(self):
    driver = ExecutorDriver()
    executor_info = mesos_pb2.ExecutorInfo()
    framework_info = mesos_pb2.FrameworkInfo()
    slave_info = mesos_pb2.SlaveInfo()

    self.executor_base.registered(driver, executor_info, framework_info, slave_info)
    assert self.executor_base._driver == driver
    assert self.executor_base._executor_info == executor_info
    assert self.executor_base._framework_info == framework_info
    assert self.executor_base._slave_info == slave_info

  def test_reregistered(self):
    driver = ExecutorDriver()
    slave_info = mesos_pb2.SlaveInfo()
    self.executor_base.reregistered(driver, slave_info)

  def test_disconnected(self):
    driver = ExecutorDriver()
    self.executor_base.disconnected(driver)

  @mock.patch('mesos.interface.mesos_pb2.TaskStatus', spec=mesos_pb2.TaskStatus)
  def test_send_update(self, MockTaskStatus):
    driver = mock.Mock(ExecutorDriver)
    task_id = 'task_id'
    state = mesos_pb2.TASK_RUNNING
    message = 'test_message'
    self.executor_base.send_update(driver, task_id, state, message)
    driver.sendStatusUpdate.assert_called_once_with(MockTaskStatus.return_value)
    assert MockTaskStatus.return_value.state == state
    assert MockTaskStatus.return_value.task_id.value == task_id
    assert MockTaskStatus.return_value.message == message

  def test_frameworkMessage(self):
    driver = ExecutorDriver()
    self.executor_base.frameworkMessage(driver, 'test_message')

  def test_error(self):
    driver = ExecutorDriver()
    self.executor_base.error(driver, 'message')