def test_mesos_executor_driver_framework_message(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
# wait until registered
driver.executor_process.connected.wait(timeout=MAX_TIMEOUT)
assert driver.executor_process.connected.is_set()
# send and wait for framework message
assert driver.send_framework_message(
b'beep boop beep') == mesos_pb2.DRIVER_RUNNING
self.slave.framework_message_event.wait(timeout=MAX_TIMEOUT)
assert self.slave.framework_message_event.is_set()
framework_message = internal.ExecutorToFrameworkMessage(
slave_id=self.slave.slave_id,
framework_id=mesos_pb2.FrameworkID(value=self.FRAMEWORK_ID),
executor_id=mesos_pb2.ExecutorID(value=self.EXECUTOR_ID),
data=b'beep boop beep',
)
assert len(self.slave.framework_messages) == 1
assert self.slave.framework_messages[0][
0] == driver.executor_process.pid
assert self.slave.framework_messages[0][1] == framework_message, (
'%s != %s' %
(self.slave.framework_messages[0][1], framework_message))
def test_mesos_executor_driver_framework_message(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
# wait until registered
driver.executor_process.connected.wait(timeout=MAX_TIMEOUT)
assert driver.executor_process.connected.is_set()
# send and wait for framework message
assert driver.send_framework_message(b'beep boop beep') == mesos_pb2.DRIVER_RUNNING
self.slave.framework_message_event.wait(timeout=MAX_TIMEOUT)
assert self.slave.framework_message_event.is_set()
framework_message = internal.ExecutorToFrameworkMessage(
slave_id=self.slave.slave_id,
framework_id=mesos_pb2.FrameworkID(value=self.FRAMEWORK_ID),
executor_id=mesos_pb2.ExecutorID(value=self.EXECUTOR_ID),
data=b'beep boop beep',
)
assert len(self.slave.framework_messages) == 1
assert self.slave.framework_messages[0][0] == driver.executor_process.pid
assert self.slave.framework_messages[0][1] == framework_message, (
'%s != %s' % (self.slave.framework_messages[0][1], framework_message))
