Skip to content
This repository was archived by the owner on Sep 2, 2025. It is now read-only.

Commit 79fbd90

Browse files
jx2leemikealfare
andauthored
add test case when raise ServiceUnavailable in is_retryable (#1224)
* raise ServiceUnavailable and test that it is retry-able --------- Co-authored-by: Mike Alfare <[email protected]>
1 parent d4be89a commit 79fbd90

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed
Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
kind: Features
2+
body: add is_retryable test case when raise ServiceUnavailable
3+
time: 2024-05-05T01:18:38.737882+09:00
4+
custom:
5+
Author: jx2lee
6+
Issue: "682"

tests/unit/test_bigquery_connection_manager.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,12 +84,14 @@ def test_is_retryable(self):
8484
rate_limit_error = exceptions.Forbidden(
8585
"code broke", errors=[{"reason": "rateLimitExceeded"}]
8686
)
87+
service_unavailable_error = exceptions.ServiceUnavailable("service is unavailable")
8788

8889
self.assertTrue(_is_retryable(internal_server_error))
8990
self.assertTrue(_is_retryable(bad_request_error))
9091
self.assertTrue(_is_retryable(connection_error))
9192
self.assertFalse(_is_retryable(client_error))
9293
self.assertTrue(_is_retryable(rate_limit_error))
94+
self.assertTrue(_is_retryable(service_unavailable_error))
9395

9496
def test_drop_dataset(self):
9597
mock_table = Mock()

0 commit comments

Comments
 (0)