feat: add set_default_query_job_config to client

googleapis · Feb 28, 2023 · e9b7fc3 · e9b7fc3
1 parent cd0aaa1
commit e9b7fc3
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/google/cloud/bigquery/client.py b/google/cloud/bigquery/client.py
@@ -891,6 +891,12 @@ def set_iam_policy(
 
         return Policy.from_api_repr(response)
 
+    def set_default_query_job_config(
+        self,
+        default_query_job_config: QueryJobConfig
+    ) -> None:
+        self._default_query_job_config = copy.deepcopy(default_query_job_config)
+
     def test_iam_permissions(
         self,
         table: Union[Table, TableReference, TableListItem, str],

diff --git a/tests/unit/test_client.py b/tests/unit/test_client.py
@@ -1760,6 +1760,19 @@ def test_test_iam_permissions_w_invalid_table(self):
         with self.assertRaises(ValueError):
             client.test_iam_permissions(table_resource_string, PERMISSIONS)
 
+    def test_set_default_query_job_config(self):
+        from google.cloud.bigquery import QueryJobConfig
+
+        creds = _make_credentials()
+        http = object()
+        client = self._make_one(project=self.PROJECT, credentials=creds, _http=http)
+        self.assertIsNone(client._default_query_job_config)
+
+        job_config = QueryJobConfig()
+        job_config.dry_run = True
+        client.set_default_query_job_config(job_config)
+        self.assertIsInstance(client._default_query_job_config, QueryJobConfig)
+
     def test_update_dataset_w_invalid_field(self):
         from google.cloud.bigquery.dataset import Dataset