From 3bffe04e28ae43e9f9251243f139a37f3d6f1945 Mon Sep 17 00:00:00 2001 From: Bartlomiej Hirsz Date: Thu, 14 Apr 2022 11:59:57 +0000 Subject: [PATCH] Fix system tests failing on importable checks Change-Id: I8af21344a836c232ff505c1442b960253eec31b7 --- tests/system/providers/google/gcs/example_gcs_acl.py | 4 ++-- tests/system/providers/google/gcs/example_gcs_copy_delete.py | 4 ++-- tests/system/providers/google/gcs/example_gcs_sensor.py | 4 ++-- tests/system/providers/google/gcs/example_gcs_to_gcs.py | 4 ++-- tests/system/providers/google/gcs/example_gcs_transform.py | 4 ++-- .../providers/google/gcs/example_gcs_transform_timespan.py | 4 ++-- .../providers/google/gcs/example_gcs_upload_download.py | 4 ++-- 7 files changed, 14 insertions(+), 14 deletions(-) diff --git a/tests/system/providers/google/gcs/example_gcs_acl.py b/tests/system/providers/google/gcs/example_gcs_acl.py index 7180defd95dd9..83783abbfd497 100644 --- a/tests/system/providers/google/gcs/example_gcs_acl.py +++ b/tests/system/providers/google/gcs/example_gcs_acl.py @@ -33,8 +33,8 @@ from airflow.providers.google.cloud.transfers.local_to_gcs import LocalFilesystemToGCSOperator from airflow.utils.trigger_rule import TriggerRule -ENV_ID = os.environ["SYSTEM_TESTS_ENV_ID"] -PROJECT_ID = os.environ["SYSTEM_TESTS_GCP_PROJECT"] +ENV_ID = os.environ.get("SYSTEM_TESTS_ENV_ID") +PROJECT_ID = os.environ.get("SYSTEM_TESTS_GCP_PROJECT") DAG_ID = "gcs_acl" diff --git a/tests/system/providers/google/gcs/example_gcs_copy_delete.py b/tests/system/providers/google/gcs/example_gcs_copy_delete.py index 519ded377ec88..b00c947def291 100644 --- a/tests/system/providers/google/gcs/example_gcs_copy_delete.py +++ b/tests/system/providers/google/gcs/example_gcs_copy_delete.py @@ -37,8 +37,8 @@ from airflow.providers.google.cloud.transfers.local_to_gcs import LocalFilesystemToGCSOperator from airflow.utils.trigger_rule import TriggerRule -ENV_ID = os.environ["SYSTEM_TESTS_ENV_ID"] -PROJECT_ID = os.environ["SYSTEM_TESTS_GCP_PROJECT"] +ENV_ID = os.environ.get("SYSTEM_TESTS_ENV_ID") +PROJECT_ID = os.environ.get("SYSTEM_TESTS_GCP_PROJECT") DAG_ID = "gcs_copy_delete" diff --git a/tests/system/providers/google/gcs/example_gcs_sensor.py b/tests/system/providers/google/gcs/example_gcs_sensor.py index d7a19be1303bf..90c6016db9b71 100644 --- a/tests/system/providers/google/gcs/example_gcs_sensor.py +++ b/tests/system/providers/google/gcs/example_gcs_sensor.py @@ -36,8 +36,8 @@ from airflow.providers.google.cloud.transfers.local_to_gcs import LocalFilesystemToGCSOperator from airflow.utils.trigger_rule import TriggerRule -ENV_ID = os.environ["SYSTEM_TESTS_ENV_ID"] -PROJECT_ID = os.environ["SYSTEM_TESTS_GCP_PROJECT"] +ENV_ID = os.environ.get("SYSTEM_TESTS_ENV_ID") +PROJECT_ID = os.environ.get("SYSTEM_TESTS_GCP_PROJECT") DAG_ID = "gcs_sensor" diff --git a/tests/system/providers/google/gcs/example_gcs_to_gcs.py b/tests/system/providers/google/gcs/example_gcs_to_gcs.py index 45f339b7ec35d..f067c29ef74d9 100644 --- a/tests/system/providers/google/gcs/example_gcs_to_gcs.py +++ b/tests/system/providers/google/gcs/example_gcs_to_gcs.py @@ -35,8 +35,8 @@ from airflow.providers.google.cloud.transfers.local_to_gcs import LocalFilesystemToGCSOperator from airflow.utils.trigger_rule import TriggerRule -ENV_ID = os.environ["SYSTEM_TESTS_ENV_ID"] -PROJECT_ID = os.environ["SYSTEM_TESTS_GCP_PROJECT"] +ENV_ID = os.environ.get("SYSTEM_TESTS_ENV_ID") +PROJECT_ID = os.environ.get("SYSTEM_TESTS_GCP_PROJECT") DAG_ID = "gcs_to_gcs" diff --git a/tests/system/providers/google/gcs/example_gcs_transform.py b/tests/system/providers/google/gcs/example_gcs_transform.py index 7fc429af82b21..75bef5113aaf3 100644 --- a/tests/system/providers/google/gcs/example_gcs_transform.py +++ b/tests/system/providers/google/gcs/example_gcs_transform.py @@ -32,8 +32,8 @@ from airflow.providers.google.cloud.transfers.local_to_gcs import LocalFilesystemToGCSOperator from airflow.utils.trigger_rule import TriggerRule -ENV_ID = os.environ["SYSTEM_TESTS_ENV_ID"] -PROJECT_ID = os.environ["SYSTEM_TESTS_GCP_PROJECT"] +ENV_ID = os.environ.get("SYSTEM_TESTS_ENV_ID") +PROJECT_ID = os.environ.get("SYSTEM_TESTS_GCP_PROJECT") DAG_ID = "gcs_transform" diff --git a/tests/system/providers/google/gcs/example_gcs_transform_timespan.py b/tests/system/providers/google/gcs/example_gcs_transform_timespan.py index 1bada6b2946ed..40f2eeafa059c 100644 --- a/tests/system/providers/google/gcs/example_gcs_transform_timespan.py +++ b/tests/system/providers/google/gcs/example_gcs_transform_timespan.py @@ -33,8 +33,8 @@ from airflow.providers.google.cloud.transfers.local_to_gcs import LocalFilesystemToGCSOperator from airflow.utils.trigger_rule import TriggerRule -ENV_ID = os.environ["SYSTEM_TESTS_ENV_ID"] -PROJECT_ID = os.environ["SYSTEM_TESTS_GCP_PROJECT"] +ENV_ID = os.environ.get("SYSTEM_TESTS_ENV_ID") +PROJECT_ID = os.environ.get("SYSTEM_TESTS_GCP_PROJECT") DAG_ID = "gcs_transform_timespan" diff --git a/tests/system/providers/google/gcs/example_gcs_upload_download.py b/tests/system/providers/google/gcs/example_gcs_upload_download.py index 5ba7ae25b3567..98934aaea43e4 100644 --- a/tests/system/providers/google/gcs/example_gcs_upload_download.py +++ b/tests/system/providers/google/gcs/example_gcs_upload_download.py @@ -29,8 +29,8 @@ from airflow.providers.google.cloud.transfers.local_to_gcs import LocalFilesystemToGCSOperator from airflow.utils.trigger_rule import TriggerRule -ENV_ID = os.environ["SYSTEM_TESTS_ENV_ID"] -PROJECT_ID = os.environ["SYSTEM_TESTS_GCP_PROJECT"] +ENV_ID = os.environ.get("SYSTEM_TESTS_ENV_ID") +PROJECT_ID = os.environ.get("SYSTEM_TESTS_GCP_PROJECT") DAG_ID = "gcs_upload_download"