def get_executor_under_test(dotted_path): """ Create and return a MockExecutor """ from airflow.executors.executor_loader import ExecutorLoader if dotted_path == "MockExecutor": try: # Run against master and 1.10.x releases from tests.test_utils.mock_executor import MockExecutor as Executor except ImportError: from tests.executors.test_executor import TestExecutor as Executor else: Executor = ExecutorLoader.load_executor(dotted_path) # Change this to try other executors class ShortCircuitExecutor(ShortCircuitExecutorMixin, Executor): """ Placeholder class that implements the inheritance hierarchy """ scheduler_job = None return ShortCircuitExecutor
def get_executor_under_test(dotted_path): """ Create and return a MockExecutor """ from airflow.executors.executor_loader import ExecutorLoader if dotted_path == "MockExecutor": from tests.test_utils.mock_executor import MockExecutor as executor else: executor = ExecutorLoader.load_executor(dotted_path) executor_cls = type(executor) # Change this to try other executors class ShortCircuitExecutor(ShortCircuitExecutorMixin, executor_cls): """ Placeholder class that implements the inheritance hierarchy """ scheduler_job = None return ShortCircuitExecutor