Tiny fix for system test example_datacatalog_search_catalog.py (apach…

…e#35701)
NayaraWakewski · Nov 17, 2023 · 02165c5 · 02165c5
1 parent c47f516
commit 02165c5
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/system/providers/google/datacatalog/example_datacatalog_search_catalog.py b/tests/system/providers/google/datacatalog/example_datacatalog_search_catalog.py
@@ -47,11 +47,11 @@
 
 BUCKET_NAME = f"bucket_{DAG_ID}_{ENV_ID}"
 LOCATION = "us-central1"
-ENTRY_GROUP_ID = f"id_{DAG_ID}_{ENV_ID}"
+ENTRY_GROUP_ID = f"id_{DAG_ID}_{ENV_ID}".replace("-", "_")
 ENTRY_GROUP_NAME = f"name {DAG_ID} {ENV_ID}"
 ENTRY_ID = "python_files"
 ENTRY_NAME = "Wizard"
-TEMPLATE_ID = f"template_id_search_{ENV_ID}"
+TEMPLATE_ID = f"template_id_search_{ENV_ID}".replace("-", "_")
 TAG_TEMPLATE_DISPLAY_NAME = f"Data Catalog {DAG_ID} {ENV_ID}"
 FIELD_NAME_1 = "first"
 
@@ -147,7 +147,9 @@
     # Search
     # [START howto_operator_gcp_datacatalog_search_catalog]
     search_catalog = CloudDataCatalogSearchCatalogOperator(
-        task_id="search_catalog", scope={"include_project_ids": [PROJECT_ID]}, query=f"projectid:{PROJECT_ID}"
+        task_id="search_catalog",
+        scope={"include_project_ids": [PROJECT_ID]},
+        query=f"name:{ENTRY_GROUP_NAME}",
     )
     # [END howto_operator_gcp_datacatalog_search_catalog]