def test_mesos_executor_driver_init(self):
os.environ['MESOS_SLAVE_PID'] = str(self.slave.pid)
executor = Executor()
driver = PesosExecutorDriver(executor, context=self.context)
assert driver.start() == mesos_pb2.DRIVER_RUNNING
assert driver.stop() == mesos_pb2.DRIVER_STOPPED
assert driver.join() == mesos_pb2.DRIVER_STOPPED
def test_mesos_executor_driver_init(self):
os.environ['MESOS_SLAVE_PID'] = str(self.slave.pid)
executor = Executor()
driver = PesosExecutorDriver(executor, context=self.context)
assert driver.start() == mesos_pb2.DRIVER_RUNNING
assert driver.stop() == mesos_pb2.DRIVER_STOPPED
assert driver.join() == mesos_pb2.DRIVER_STOPPED
def test_mesos_executor_driver_init(): context = Context() context.start() # spawn slave slave = MockSlave() context.spawn(slave) os.environ['MESOS_SLAVE_PID'] = str(slave.pid) executor = Executor() driver = PesosExecutorDriver(executor, context=context) assert driver.start() == mesos.DRIVER_RUNNING assert driver.stop() == mesos.DRIVER_STOPPED assert driver.join() == mesos.DRIVER_STOPPED context.stop()
def test_mesos_executor_register(self):
os.environ['MESOS_SLAVE_PID'] = str(self.slave.pid)
registered_event, registered_call, registered_side_effect = self.mock_method()
executor = mock.create_autospec(Executor, spec_set=True)
executor.registered = mock.Mock(side_effect=registered_side_effect)
driver = PesosExecutorDriver(executor, context=self.context)
assert driver.start() == mesos_pb2.DRIVER_RUNNING
driver.executor_process.connected.wait(timeout=MAX_TIMEOUT)
assert driver.executor_process.connected.is_set()
registered_event.wait(timeout=MAX_TIMEOUT)
assert registered_event.is_set()
assert registered_call.mock_calls == [
mock.call(driver, self.executor_info, self.framework_info, self.slave.slave_info)]
assert driver.stop() == mesos_pb2.DRIVER_STOPPED
def test_mesos_executor_register():
context = Context()
context.start()
# spawn slave
slave = MockSlave()
context.spawn(slave)
os.environ['MESOS_SLAVE_PID'] = str(slave.pid)
executor = mock.MagicMock()
executor.registered = mock.MagicMock()
driver = PesosExecutorDriver(executor, context=context)
assert driver.start() == mesos.DRIVER_RUNNING
command_info = mesos.CommandInfo(value='wat')
framework_id = mesos.FrameworkID(value='fake_framework_id')
executor_id = mesos.ExecutorID(value='fake_executor_id')
executor_info = mesos.ExecutorInfo(
executor_id=executor_id,
framework_id=framework_id,
command=command_info
)
framework_info = mesos.FrameworkInfo(user='******', name='fake_framework_name')
slave.send_registered(
driver.executor_process.pid,
executor_info,
framework_id,
framework_info
)
driver.executor_process.connected.wait(timeout=1)
assert driver.executor_process.connected.is_set()
executor.registered.assert_called_with(driver, executor_info, framework_info, slave.slave_info)
assert driver.stop() == mesos.DRIVER_STOPPED
context.stop()
def test_mesos_executor_register(self):
os.environ['MESOS_SLAVE_PID'] = str(self.slave.pid)
registered_event, registered_call, registered_side_effect = self.mock_method(
)
executor = mock.create_autospec(Executor, spec_set=True)
executor.registered = mock.Mock(side_effect=registered_side_effect)
driver = PesosExecutorDriver(executor, context=self.context)
assert driver.start() == mesos_pb2.DRIVER_RUNNING
driver.executor_process.connected.wait(timeout=MAX_TIMEOUT)
assert driver.executor_process.connected.is_set()
registered_event.wait(timeout=MAX_TIMEOUT)
assert registered_event.is_set()
assert registered_call.mock_calls == [
mock.call(driver, self.executor_info, self.framework_info,
self.slave.slave_info)
]
assert driver.stop() == mesos_pb2.DRIVER_STOPPED
