@@ -15,7 +15,7 @@ use aws_smithy_runtime_api::client::interceptors::Intercept;
1515use aws_smithy_runtime_api:: client:: runtime_components:: RuntimeComponents ;
1616use aws_smithy_types:: body:: SdkBody ;
1717use aws_smithy_types:: config_bag:: ConfigBag ;
18- use aws_smithy_types:: retry:: RetryConfig ;
18+ use aws_smithy_types:: retry:: { RetryConfig , RetrySpec } ;
1919use std:: time:: { Duration , SystemTime } ;
2020
2121fn req ( ) -> http_1x:: Request < SdkBody > {
@@ -74,6 +74,7 @@ async fn long_polling_backoff_after_transient_error_when_token_bucket_empty() {
7474
7575 let config = Config :: builder ( )
7676 . with_test_defaults_v2 ( )
77+ . retry_config ( RetryConfig :: standard ( ) . with_retry_spec ( RetrySpec :: v2_1 ( ) ) )
7778 . stalled_stream_protection ( StalledStreamProtectionConfig :: disabled ( ) )
7879 . interceptor ( StaticBackoffInterceptor )
7980 . time_source ( SharedTimeSource :: new ( time_source) )
@@ -108,6 +109,7 @@ async fn long_polling_backoff_after_throttling_error_when_token_bucket_empty() {
108109
109110 let config = Config :: builder ( )
110111 . with_test_defaults_v2 ( )
112+ . retry_config ( RetryConfig :: standard ( ) . with_retry_spec ( RetrySpec :: v2_1 ( ) ) )
111113 . stalled_stream_protection ( StalledStreamProtectionConfig :: disabled ( ) )
112114 . interceptor ( StaticBackoffInterceptor )
113115 . time_source ( SharedTimeSource :: new ( time_source) )
@@ -144,6 +146,7 @@ async fn long_polling_max_attempts_exceeded_must_not_delay() {
144146
145147 let config = Config :: builder ( )
146148 . with_test_defaults_v2 ( )
149+ . retry_config ( RetryConfig :: standard ( ) . with_retry_spec ( RetrySpec :: v2_1 ( ) ) )
147150 . stalled_stream_protection ( StalledStreamProtectionConfig :: disabled ( ) )
148151 . interceptor ( StaticBackoffInterceptor )
149152 . time_source ( SharedTimeSource :: new ( time_source) )
@@ -188,6 +191,7 @@ async fn long_polling_success_must_not_delay() {
188191
189192 let config = Config :: builder ( )
190193 . with_test_defaults_v2 ( )
194+ . retry_config ( RetryConfig :: standard ( ) . with_retry_spec ( RetrySpec :: v2_1 ( ) ) )
191195 . stalled_stream_protection ( StalledStreamProtectionConfig :: disabled ( ) )
192196 . interceptor ( StaticBackoffInterceptor )
193197 . time_source ( SharedTimeSource :: new ( time_source) )
@@ -230,6 +234,7 @@ async fn long_polling_non_retryable_errors_must_not_delay() {
230234
231235 let config = Config :: builder ( )
232236 . with_test_defaults_v2 ( )
237+ . retry_config ( RetryConfig :: standard ( ) . with_retry_spec ( RetrySpec :: v2_1 ( ) ) )
233238 . stalled_stream_protection ( StalledStreamProtectionConfig :: disabled ( ) )
234239 . interceptor ( StaticBackoffInterceptor )
235240 . time_source ( SharedTimeSource :: new ( time_source) )
0 commit comments