def test_nested_pipeline_jinja_arg(self):
results = run_and_get(nested_pipeline_dag, "my_xcom_pipeline")
first_result = read_xcom_result_value(results, "result_1")
second_result = read_xcom_result_value(results, "result_2")
assert "SEPARATOR" in first_result
assert "SEPARATOR" not in second_result
def test_jinja_arg_to_task(self):
results = run_and_get(jinja_to_task_dag, "bool_to_string")
parsed_result = read_xcom_result_value(results)
assert parsed_result == "False"
def test_simple_pipeline_jinja_arg(self):
results = run_and_get(simple_jinja_dag, "my_second_pipeline")
parsed_result = read_xcom_result_value(results)
assert "SEPARATOR" in parsed_result
def test_dag_config_run(self):
result = run_and_get(dag_with_config, task_id="t_A")
actual = read_xcom_result_value(result)
assert actual == "from_configfrom_config"
def test_dag_pipeline_run(self):
result = run_and_get(dag_with_pipeline, task_id="t_A")
actual = read_xcom_result_value(result)
assert actual == "from_pipeline_ctorfrom_pipeline_ctor"
def test_simple_2_run(self):
assert len(dag_simple_run_wiring.tasks) == 2
result = run_and_get(dag_simple_run_wiring, task_id="t_B")
assert read_xcom_result_value(result) == "checkcheckt_B"
def test_simple_run(self):
actual = run_and_get(dag_simple_build, "t_A")
assert read_xcom_result_value(actual) == "checkcheck"