diff --git a/slack_sdk/http_retry/__init__.py b/slack_sdk/http_retry/__init__.py index b7116dad9..17b04acfa 100644 --- a/slack_sdk/http_retry/__init__.py +++ b/slack_sdk/http_retry/__init__.py @@ -4,6 +4,7 @@ from .builtin_handlers import ( ConnectionErrorRetryHandler, RateLimitErrorRetryHandler, + ServerErrorRetryHandler, ) from .interval_calculator import RetryIntervalCalculator from .builtin_interval_calculators import ( @@ -17,6 +18,7 @@ connect_error_retry_handler = ConnectionErrorRetryHandler() rate_limit_error_retry_handler = RateLimitErrorRetryHandler() +server_error_retry_handler = ServerErrorRetryHandler() def default_retry_handlers() -> List[RetryHandler]: @@ -27,6 +29,7 @@ def all_builtin_retry_handlers() -> List[RetryHandler]: return [ connect_error_retry_handler, rate_limit_error_retry_handler, + server_error_retry_handler, ] @@ -34,6 +37,7 @@ def all_builtin_retry_handlers() -> List[RetryHandler]: "RetryHandler", "ConnectionErrorRetryHandler", "RateLimitErrorRetryHandler", + "ServerErrorRetryHandler", "RetryIntervalCalculator", "FixedValueRetryIntervalCalculator", "BackoffRetryIntervalCalculator", diff --git a/slack_sdk/http_retry/async_handler.py b/slack_sdk/http_retry/async_handler.py index ed9f6115a..e16ba3286 100644 --- a/slack_sdk/http_retry/async_handler.py +++ b/slack_sdk/http_retry/async_handler.py @@ -80,6 +80,7 @@ async def prepare_for_next_attempt_async( __all__ = [ + "AsyncRetryHandler", "RetryState", "HttpRequest", "HttpResponse", diff --git a/tests/slack_sdk/http_retry/test_builtins.py b/tests/slack_sdk/http_retry/test_builtins.py index 49394bb30..4918e63a8 100644 --- a/tests/slack_sdk/http_retry/test_builtins.py +++ b/tests/slack_sdk/http_retry/test_builtins.py @@ -17,11 +17,11 @@ def test_default_ones(self): self.assertEqual(1, len(list)) list = all_builtin_retry_handlers() - self.assertEqual(2, len(list)) + self.assertEqual(3, len(list)) list.clear() self.assertEqual(0, len(list)) list = all_builtin_retry_handlers() - self.assertEqual(2, len(list)) + self.assertEqual(3, len(list)) def test_fixed_value_retry_interval_calculator(self): for fixed_value in [0.1, 0.2]: