Skip to content

Commit 9f5b1c8

Browse files
committed
fix codegen client test
1 parent 969b00a commit 9f5b1c8

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

codegen-client/src/test/kotlin/software/amazon/smithy/rust/codegen/client/smithy/generators/ConfigOverrideRuntimePluginGeneratorTest.kt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -182,6 +182,7 @@ internal class ConfigOverrideRuntimePluginGeneratorTest {
182182
"ShouldAttempt" to
183183
RuntimeType.smithyRuntimeApi(runtimeConfig)
184184
.resolve("client::retries::ShouldAttempt"),
185+
"TokenBucket" to RuntimeType.smithyRuntime(runtimeConfig).resolve("client::retries::TokenBucket"),
185186
)
186187
rustCrate.testModule {
187188
unitTest("test_operation_overrides_retry_config") {
@@ -199,6 +200,7 @@ internal class ConfigOverrideRuntimePluginGeneratorTest {
199200
200201
let mut layer = #{Layer}::new("test");
201202
layer.store_put(#{RequestAttempts}::new(1));
203+
layer.store_put(#{TokenBucket}::default());
202204
203205
let mut cfg = #{ConfigBag}::of_layers(vec![layer]);
204206
let client_config_layer = client_config.config;

0 commit comments

Comments
 (0)