Migrate Google analytics example to new design AIP-47 (#25006)

related: #22447, #22430
apache · Jul 16, 2022 · 9b7e4a7 · 9b7e4a7
1 parent 84b6799
commit 9b7e4a7
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 64 deletions.
diff --git a/docs/apache-airflow-providers-google/operators/marketing_platform/analytics.rst b/docs/apache-airflow-providers-google/operators/marketing_platform/analytics.rst
@@ -35,7 +35,7 @@ List the Accounts
 To list accounts from Analytics you can use the
 :class:`~airflow.providers.google.marketing_platform.operators.analytics.GoogleAnalyticsListAccountsOperator`.
 
-.. exampleinclude:: /../../airflow/providers/google/marketing_platform/example_dags/example_analytics.py
+.. exampleinclude:: /../../tests/system/providers/google/marketing_platform/example_analytics.py
  :language: python
  :dedent: 4
  :start-after: [START howto_marketing_platform_list_accounts_operator]
@@ -53,7 +53,7 @@ Returns a web property-Google Ads link to which the user has access.
 To list web property-Google Ads link you can use the
 :class:`~airflow.providers.google.marketing_platform.operators.analytics.GoogleAnalyticsGetAdsLinkOperator`.
 
-.. exampleinclude:: /../../airflow/providers/google/marketing_platform/example_dags/example_analytics.py
+.. exampleinclude:: /../../tests/system/providers/google/marketing_platform/example_analytics.py
  :language: python
  :dedent: 4
  :start-after: [START howto_marketing_platform_get_ads_link_operator]
@@ -71,7 +71,7 @@ Operator returns a list of entity Google Ads links.
 To list Google Ads links you can use the
 :class:`~airflow.providers.google.marketing_platform.operators.analytics.GoogleAnalyticsRetrieveAdsLinksListOperator`.
 
-.. exampleinclude:: /../../airflow/providers/google/marketing_platform/example_dags/example_analytics.py
+.. exampleinclude:: /../../tests/system/providers/google/marketing_platform/example_analytics.py
  :language: python
  :dedent: 4
  :start-after: [START howto_marketing_platform_retrieve_ads_links_list_operator]

diff --git a/tests/providers/google/marketing_platform/operators/test_analytics_system.py b/tests/providers/google/marketing_platform/operators/test_analytics_system.py
diff --git a/...latform/example_dags/example_analytics.py → ...e/marketing_platform/example_analytics.py b/...latform/example_dags/example_analytics.py → ...e/marketing_platform/example_analytics.py
@@ -30,6 +30,9 @@
  GoogleAnalyticsRetrieveAdsLinksListOperator,
 )
 
+ENV_ID = os.environ.get("SYSTEM_TESTS_ENV_ID")
+DAG_ID = "example_google_analytics"
+
 ACCOUNT_ID = os.environ.get("GA_ACCOUNT_ID", "123456789")
 
 BUCKET = os.environ.get("GMP_ANALYTICS_BUCKET", "test-airflow-analytics-bucket")
@@ -39,10 +42,11 @@
 DATA_ID = "kjdDu3_tQa6n8Q1kXFtSmg"
 
 with models.DAG(
- "example_google_analytics",
+ DAG_ID,
  schedule_interval='@once', # Override to match your needs,
  start_date=datetime(2021, 1, 1),
  catchup=False,
+ tags=["example", "analytics"],
 ) as dag:
  # [START howto_marketing_platform_list_accounts_operator]
  list_account = GoogleAnalyticsListAccountsOperator(task_id="list_account")
@@ -86,3 +90,9 @@
  )
 
  upload >> [delete, transform]
+
+
+from tests.system.utils import get_test_run # noqa: E402
+
+# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
+test_run = get_test_run(dag)