fix codegen client test

smithy-lang · Jan 9, 2025 · 9f5b1c8 · 9f5b1c8
1 parent 969b00a
commit 9f5b1c8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.../smithy/rust/codegen/client/smithy/generators/ConfigOverrideRuntimePluginGeneratorTest.kt b/.../smithy/rust/codegen/client/smithy/generators/ConfigOverrideRuntimePluginGeneratorTest.kt
@@ -182,6 +182,7 @@ internal class ConfigOverrideRuntimePluginGeneratorTest {
                     "ShouldAttempt" to
                         RuntimeType.smithyRuntimeApi(runtimeConfig)
                             .resolve("client::retries::ShouldAttempt"),
+                    "TokenBucket" to RuntimeType.smithyRuntime(runtimeConfig).resolve("client::retries::TokenBucket"),
                 )
             rustCrate.testModule {
                 unitTest("test_operation_overrides_retry_config") {
@@ -199,6 +200,7 @@ internal class ConfigOverrideRuntimePluginGeneratorTest {
 
                         let mut layer = #{Layer}::new("test");
                         layer.store_put(#{RequestAttempts}::new(1));
+                        layer.store_put(#{TokenBucket}::default());
 
                         let mut cfg = #{ConfigBag}::of_layers(vec![layer]);
                         let client_config_layer = client_config.config;