mik-laj commented on a change in pull request #6761: [AIRFLOW-6204] Add GCP system tests helper URL: https://github.com/apache/airflow/pull/6761#discussion_r362333009
########## File path: tests/gcp/operators/test_cloud_memorystore_system.py ########## @@ -18,31 +18,35 @@ # under the License. """System tests for Google Cloud Memorystore operators""" -from tests.gcp.operators.test_cloud_memorystore_system_helper import GCPCloudMemorystoreTestHelper +import pytest + +from airflow.gcp.example_dags.example_cloud_memorystore import BUCKET_NAME from tests.gcp.utils.gcp_authenticator import GCP_MEMORYSTORE # TODO: Update it -from tests.test_utils.gcp_system_helpers import GCP_DAG_FOLDER, provide_gcp_context, skip_gcp_system -from tests.test_utils.system_tests_class import SystemTest +from tests.test_utils.gcp_system_helpers import GCP_DAG_FOLDER, GcpSystemTest, provide_gcp_context + +command = GcpSystemTest.commands_registry() + + +@command +def create_bucket(): + GcpSystemTest.create_gcs_bucket(BUCKET_NAME, location="europe-north1") -@skip_gcp_system(GCP_MEMORYSTORE, require_local_executor=True) -class CloudBuildExampleDagsSystemTest(SystemTest): - """ - System tests for Google Cloud Memorystore operators +@command +def delete_bucket(): + GcpSystemTest.delete_gcs_bucket(BUCKET_NAME) - It use a real service. - """ - helper = GCPCloudMemorystoreTestHelper() - @provide_gcp_context(GCP_MEMORYSTORE) - def setUp(self): - super().setUp() - self.helper.create_bucket() [email protected] +def helper(): + create_bucket() + yield + delete_bucket() - @provide_gcp_context(GCP_MEMORYSTORE) - def test_run_example_dag(self): - self.run_dag('gcp_cloud_memorystore', GCP_DAG_FOLDER) - @provide_gcp_context(GCP_MEMORYSTORE) - def tearDown(self): - self.helper.delete_bucket() - super().tearDown() +@command [email protected](GCP_MEMORYSTORE) [email protected]("helper") +def test_run_example_dag(): Review comment: I think this discussion should be on the mailing list. In a moment I will create a new thread to discuss the topic of using new pytest features. Many people expressed their support for the question but did not comment on how to migrate the old convention to the new one. We don't have to use all the pytest features because the old code still works fine. ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected] With regards, Apache Git Services
