Fix system tests failing on importable checks

Change-Id: I8af21344a836c232ff505c1442b960253eec31b7
apache · Apr 14, 2022 · 3bffe04 · 3bffe04
1 parent 5bb09f6
commit 3bffe04
Show file tree

Hide file tree

Showing 7 changed files with 14 additions and 14 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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"