Fix protocol tests

smithy-lang · May 9, 2024 · c6273ac · c6273ac
1 parent 9543c67
commit c6273ac
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/...-swift-codegen/src/main/kotlin/software/amazon/smithy/swift/codegen/ClientRuntimeTypes.kt b/...-swift-codegen/src/main/kotlin/software/amazon/smithy/swift/codegen/ClientRuntimeTypes.kt
@@ -120,7 +120,6 @@ object ClientRuntimeTypes {
         val SDKLogLevel = runtimeSymbol("SDKLogLevel")
         val ClientLogMode = runtimeSymbol("ClientLogMode")
         val IdempotencyTokenGenerator = runtimeSymbol("IdempotencyTokenGenerator")
-        val DefaultRetryStrategy = runtimeSymbol("DefaultRetryStrategy")
         val DefaultRetryErrorInfoProvider = runtimeSymbol("DefaultRetryErrorInfoProvider")
         val DefaultSDKRuntimeConfiguration = runtimeSymbol("DefaultSDKRuntimeConfiguration")
         val DateFormatter = runtimeSymbol("DateFormatter")

diff --git a/...in/kotlin/software/amazon/smithy/swift/codegen/integration/middlewares/RetryMiddleware.kt b/...in/kotlin/software/amazon/smithy/swift/codegen/integration/middlewares/RetryMiddleware.kt
@@ -33,7 +33,8 @@ class RetryMiddleware(
 
     override fun render(ctx: ProtocolGenerator.GenerationContext, writer: SwiftWriter, op: OperationShape, operationStackName: String) {
         if (ctx.settings.useInterceptors) {
-            writer.write("builder.retryStrategy(\$N(options: config.retryStrategyOptions))", ClientRuntimeTypes.Core.DefaultRetryStrategy)
+            writer.addImport(SwiftDependency.SMITHY_RETRIES.target)
+            writer.write("builder.retryStrategy(\$N(options: config.retryStrategyOptions))", SmithyRetriesTypes.DefaultRetryStrategy)
             writer.write("builder.retryErrorInfoProvider(\$N.errorInfo(for:))", retryErrorInfoProviderSymbol)
         } else {
             val output = MiddlewareShapeUtils.outputSymbol(symbolProvider, model, op)