Bug: revises Exception type (googleapis#1994)

chalmerlowe · web-flow · commit e7e0fcf9d5db · 2024-08-14T12:30:51.000-04:00
* revises Exception type

* updates error choices
diff --git a/google/cloud/bigquery/job/base.py b/google/cloud/bigquery/job/base.py
@@ -49,7 +49,7 @@
     "notImplemented": http.client.NOT_IMPLEMENTED,
     "policyViolation": http.client.FORBIDDEN,
     "quotaExceeded": http.client.FORBIDDEN,
-    "rateLimitExceeded": http.client.FORBIDDEN,
+    "rateLimitExceeded": http.client.TOO_MANY_REQUESTS,
     "resourceInUse": http.client.BAD_REQUEST,
     "resourcesExceeded": http.client.BAD_REQUEST,
     "responseTooLarge": http.client.FORBIDDEN,
diff --git a/tests/unit/test_job_retry.py b/tests/unit/test_job_retry.py
@@ -442,7 +442,7 @@ def api_request(method, path, query_params=None, data=None, **kw):
 
     orig_job_id = job.job_id
     job_retry = dict(job_retry=None) if job_retry_on_query == "Result" else {}
-    with pytest.raises(google.api_core.exceptions.Forbidden):
+    with pytest.raises(google.api_core.exceptions.TooManyRequests):
         job.result(**job_retry)
 
     assert job.job_id == orig_job_id
