Update error in system tests that expect error for wrong location (googleapis#7286)

tswast · web-flow · commit 5e2b474cb20b · 2019-02-04T14:53:30.000-08:00
We used to get BadRequest for querying with the wrong location. Now we
get NotFound. I update the test to just look for the base class of
GoogleAPICallError, since it seems we aren't guarnteed the actual status
code.
diff --git a/bigquery/tests/system.py b/bigquery/tests/system.py
@@ -48,6 +48,7 @@
 from google.api_core.exceptions import BadRequest
 from google.api_core.exceptions import Conflict
 from google.api_core.exceptions import Forbidden
+from google.api_core.exceptions import GoogleAPICallError
 from google.api_core.exceptions import NotFound
 from google.api_core.exceptions import InternalServerError
 from google.api_core.exceptions import ServiceUnavailable
@@ -765,17 +766,17 @@ def test_load_table_from_file_w_explicit_location(self):
         self.assertTrue(eu_query.done)
 
         # Cannot query from US.
-        with self.assertRaises(BadRequest):
+        with self.assertRaises(GoogleAPICallError):
             list(client.query(query_string, location="US", job_config=query_config))
 
         # Cannot copy from US.
-        with self.assertRaises(BadRequest):
+        with self.assertRaises(GoogleAPICallError):
             client.copy_table(
                 table_ref, dataset.table("letters2_us"), location="US"
             ).result()
 
         # Cannot extract from US.
-        with self.assertRaises(BadRequest):
+        with self.assertRaises(GoogleAPICallError):
             client.extract_table(
                 table_ref, "gs://{}/letters-us.csv".format(bucket_name), location="US"
             ).result()
