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()