def test_mesos_executor_driver_abort(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.abort() == mesos_pb2.DRIVER_ABORTED
driver.executor_process.aborted.wait(timeout=MAX_TIMEOUT)
assert driver.executor_process.aborted.is_set()
assert driver.join() == mesos_pb2.DRIVER_ABORTED
def test_mesos_executor_driver_abort(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.abort() == mesos_pb2.DRIVER_ABORTED
driver.executor_process.aborted.wait(timeout=MAX_TIMEOUT)
assert driver.executor_process.aborted.is_set()
assert driver.join() == mesos_pb2.DRIVER_ABORTED
def test_mesos_executor_driver_run_and_abort(self):
os.environ['MESOS_SLAVE_PID'] = str(self.slave.pid)
executor = Executor()
driver = PesosExecutorDriver(executor, context=self.context)
join_event = threading.Event()
def runner():
assert driver.run() == mesos_pb2.DRIVER_ABORTED
join_event.set()
threading.Thread(target=runner).start()
assert not join_event.is_set()
driver.started.wait(timeout=MAX_TIMEOUT)
assert driver.started.is_set()
assert driver.abort() == mesos_pb2.DRIVER_ABORTED
join_event.wait(timeout=MAX_TIMEOUT)
assert join_event.is_set()
def test_mesos_executor_driver_run_and_abort(self):
os.environ['MESOS_SLAVE_PID'] = str(self.slave.pid)
executor = Executor()
driver = PesosExecutorDriver(executor, context=self.context)
join_event = threading.Event()
def runner():
assert driver.run() == mesos_pb2.DRIVER_ABORTED
join_event.set()
threading.Thread(target=runner).start()
assert not join_event.is_set()
driver.started.wait(timeout=MAX_TIMEOUT)
assert driver.started.is_set()
assert driver.abort() == mesos_pb2.DRIVER_ABORTED
join_event.wait(timeout=MAX_TIMEOUT)
assert join_event.is_set()
