michalslowikowski00 commented on a change in pull request #8631:
URL: https://github.com/apache/airflow/pull/8631#discussion_r423011020
##########
File path: tests/providers/google/cloud/hooks/test_bigquery.py
##########
@@ -376,9 +360,8 @@ def test_run_load_with_non_csv_as_src_fmt(self, fmt,
mock_get_service, mock_rwc)
except ValueError:
self.fail("run_load() raised ValueError unexpectedly!")
-
@mock.patch("airflow.providers.google.cloud.hooks.bigquery.BigQueryHook.get_service")
-
@mock.patch("airflow.providers.google.cloud.hooks.bigquery.BigQueryHook.run_with_configuration")
- def test_run_extract(self, run_with_config, mock_get_service):
+
@mock.patch("airflow.providers.google.cloud.hooks.bigquery.BigQueryHook.insert_job")
+ def test_run_extract(self, mock_insert):
source_project_dataset_table = "{}.{}.{}".format(PROJECT_ID,
DATASET_ID, TABLE_ID)
destination_cloud_storage_uris = ["gs://bucket/file.csv"]
Review comment:
```suggestion
destination_cloud_storage_uri = ["gs://bucket/file.csv"]
```
----------------------------------------------------------------
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]
