Skip to content

Commit c1838c0

Browse files
committed
Update long-polling integration tests
1 parent 7a16b1b commit c1838c0

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

aws/sdk/integration-tests/sqs/tests/long-polling.rs

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ use aws_smithy_runtime_api::client::interceptors::Intercept;
1515
use aws_smithy_runtime_api::client::runtime_components::RuntimeComponents;
1616
use aws_smithy_types::body::SdkBody;
1717
use aws_smithy_types::config_bag::ConfigBag;
18-
use aws_smithy_types::retry::RetryConfig;
18+
use aws_smithy_types::retry::{RetryConfig, RetrySpec};
1919
use std::time::{Duration, SystemTime};
2020

2121
fn 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

Comments
 (0)