Fix BigQuery system test (#18373)

apache · Sep 20, 2021 · 18d7e1e · 18d7e1e
1 parent ef0ff3f
commit 18d7e1e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/airflow/providers/google/cloud/example_dags/example_bigquery_operations.py b/airflow/providers/google/cloud/example_dags/example_bigquery_operations.py
@@ -149,6 +149,7 @@
                 "sourceFormat": "CSV",
                 "compression": "NONE",
                 "csvOptions": {"skipLeadingRows": 1},
+                "sourceUris": [DATA_SAMPLE_GCS_URL],
             },
         },
         bucket=DATA_SAMPLE_GCS_BUCKET_NAME,

diff --git a/airflow/providers/google/cloud/example_dags/example_bigquery_queries.py b/airflow/providers/google/cloud/example_dags/example_bigquery_queries.py
@@ -155,7 +155,11 @@
                 "query": {
                     "query": f"SELECT * FROM {DATASET_NAME}.{TABLE_1}",
                     "useLegacySql": False,
-                    "destinationTable": f"{DATASET_NAME}.{TABLE_2}",
+                    "destinationTable": {
+                        'projectId': PROJECT_ID,
+                        'datasetId': DATASET_NAME,
+                        'tableId': TABLE_2,
+                    },
                 }
             },
             location=location,