def test_error_dag_containerized(dagster_docker_image, environments_path): # pylint: disable=redefined-outer-name
pipeline_name = 'demo_error_pipeline'
handle = ExecutionTargetHandle.for_pipeline_module('test_pipelines.repo',
pipeline_name)
environment_yaml = [
os.path.join(environments_path, 'env_s3.yaml'),
]
environment_dict = load_yaml_from_glob_list(environment_yaml)
run_id = make_new_run_id()
execution_date = timezone.utcnow()
dag, tasks = make_airflow_dag_containerized_for_handle(
handle, pipeline_name, dagster_docker_image, environment_dict)
with pytest.raises(AirflowException) as exc_info:
execute_tasks_in_dag(dag, tasks, run_id, execution_date)
assert 'Exception: Unusual error' in str(exc_info.value)
def test_error_dag_containerized():
pipeline_name = 'demo_error_pipeline'
handle = ExecutionTargetHandle.for_pipeline_module(
'dagster_airflow_tests.test_project.dagster_airflow_demo',
pipeline_name)
environment_yaml = [
script_relative_path('test_project/env_s3.yaml'),
]
environment_dict = load_yaml_from_glob_list(environment_yaml)
run_id = str(uuid.uuid4())
execution_date = timezone.utcnow()
dag, tasks = make_airflow_dag_containerized_for_handle(
handle, pipeline_name, IMAGE, environment_dict)
with pytest.raises(AirflowException) as exc_info:
execute_tasks_in_dag(dag, tasks, run_id, execution_date)
assert 'Exception: Unusual error' in str(exc_info.value)