address reviewer feedback

Author: shollyman

tests/unit/test__job_helpers.py

@@ -601,7 +601,6 @@ def test_query_and_wait_sets_job_creation_mode():
 
 def test_query_and_wait_sets_location():
     client = mock.create_autospec(Client)
-    client.default_job_creation_mode = None
     client._call_api.return_value = {
         "jobReference": {
             "projectId": "response-project",
@@ -639,6 +638,7 @@ def test_query_and_wait_sets_location():
                 "useInt64Timestamp": True,
             },
             "requestId": mock.ANY,
+            "jobCreationMode": mock.ANY,
         },
         timeout=None,
     )

tests/unit/test_client.py

@@ -213,6 +213,17 @@ def test_ctor_w_client_options_universe(self):
         )
         self.assertEqual(client._connection.API_BASE_URL, "https://bigquery.foo.com")
 
+    def test_ctor_w_job_creation_mode(self):
+        creds = _make_credentials()
+        http = object()
+        client = self._make_one(
+            project=self.PROJECT,
+            credentials=creds,
+            _http=http,
+            default_job_creation_mode="foo",
+        )
+        self.assertEqual(client.default_job_creation_mode, "foo")
+
     def test_ctor_w_location(self):
         from google.cloud.bigquery._http import Connection