def test_parse_arguments_for_init_with_both_params_if_environment_is_set(
monkeypatch):
temp_home = tempfile.mkdtemp()
temp_airflow_home = path.join(temp_home, "original", "airflow")
temp_airflow_cfg = path.join(temp_airflow_home, "airflow.cfg")
monkeypatch.setenv("AIRFLOW_HOME", temp_airflow_home)
monkeypatch.setenv("AIRFLOW_CONFIG", temp_airflow_cfg)
monkeypatch.setattr(path, "expanduser",
lambda x: x.replace("~", temp_home))
control_airflow_home = path.join(temp_home, "not_default_anymore",
"airflow")
control_airflow_cfg = path.join(control_airflow_home, "airflow.cfg")
input_args = [
"init", "--home", control_airflow_home, "--config", control_airflow_cfg
]
result_args = parse_arguments(input_args, temp_home)
rmtree(temp_home)
assert result_args.home == control_airflow_home, \
"Failed to parse --home"
assert result_args.config == control_airflow_cfg, \
"Failed to parse --config"
def test_parse_arguments_for_init_with_relative_path_for_both_params_if_environment_is_not_set(
monkeypatch):
temp_home = tempfile.mkdtemp()
monkeypatch.delenv("AIRFLOW_HOME", raising=False)
monkeypatch.delenv("AIRFLOW_CONFIG", raising=False)
monkeypatch.setattr(path, "expanduser",
lambda x: x.replace("~", temp_home))
input_airflow_home = "./not_default_anymore/airflow"
input_airflow_cfg = "./not_default_anymore/airflow/airflow.cfg"
control_airflow_home = get_absolute_path(input_airflow_home, temp_home)
control_airflow_cfg = get_absolute_path(input_airflow_cfg, temp_home)
input_args = [
"init", "--home", input_airflow_home, "--config", input_airflow_cfg
]
result_args = parse_arguments(input_args, temp_home)
rmtree(temp_home)
assert result_args.home == control_airflow_home, \
"Failed to parse --home"
assert result_args.config == control_airflow_cfg, \
"Failed to parse --config"
def test_load_test_suite(args, control_ids):
parsed_args = parse_arguments(args, DATA_FOLDER)
try:
suite_data = load_test_suite(parsed_args)
selected_ids = [test_data["id"] for _, test_data in suite_data.items()]
except Exception as err:
assert False, f"Failed to run test, {err}"
finally:
rmtree(parsed_args.tmp)
assert selected_ids == control_ids, \
"Failed to select proper tests from --suite based on --range"
def test_parse_arguments_for_init_with_defaults_if_environment_is_set(
monkeypatch):
temp_home = tempfile.mkdtemp()
temp_airflow_home = path.join(temp_home, "original", "airflow")
temp_airflow_cfg = path.join(temp_airflow_home, "airflow.cfg")
monkeypatch.setenv("AIRFLOW_HOME", temp_airflow_home)
monkeypatch.setenv("AIRFLOW_CONFIG", temp_airflow_cfg)
monkeypatch.setattr(path, "expanduser",
lambda x: x.replace("~", temp_home))
input_args = ["init"]
result_args = parse_arguments(input_args, temp_home)
rmtree(temp_home)
assert result_args.home == temp_airflow_home, \
"Failed to set default for --home. \
Check if get_airflow_home or get_airflow_config \
from airflow.configuration were not changed"
assert result_args.config == temp_airflow_cfg, \
"Failed to set default for --config \
def test_parse_arguments_for_making_list_from_range_in_test_subparser(
args, control):
result_args = parse_arguments(args, DATA_FOLDER)
rmtree(result_args.tmp)
assert result_args.range == control