feat: update GenerationContext with directed codegen related changes

awslabs · Mar 27, 2024 · 61b676c · 61b676c
1 parent 454ad2b
commit 61b676c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/...degen/src/test/kotlin/software/amazon/smithy/aws/swift/codegen/PresignerGeneratorTests.kt b/...degen/src/test/kotlin/software/amazon/smithy/aws/swift/codegen/PresignerGeneratorTests.kt
@@ -246,7 +246,7 @@ extension PutObjectInput {
         val context = TestContextGenerator.initContextFrom(smithyFile, serviceShapeId, RestJson1Trait.ID)
         val presigner = PresignerGenerator()
         val generator = AWSRestJson1ProtocolGenerator()
-        val codegenContext = GenerationContext(context.ctx.model, context.ctx.symbolProvider, context.ctx.settings, generator)
+        val codegenContext = GenerationContext(context.ctx.model, context.ctx.symbolProvider, context.ctx.settings, context.manifest, generator)
         val protocolGenerationContext = ProtocolGenerator.GenerationContext(context.ctx.settings, context.ctx.model, context.ctx.service, context.ctx.symbolProvider, listOf(), RestJson1Trait.ID, context.ctx.delegator)
         codegenContext.protocolGenerator?.initializeMiddleware(context.ctx)
         presigner.writeAdditionalFiles(codegenContext, protocolGenerationContext, context.ctx.delegator)

diff --git a/...software/amazon/smithy/aws/swift/codegen/customizations/PresignableUrlIntegrationTests.kt b/...software/amazon/smithy/aws/swift/codegen/customizations/PresignableUrlIntegrationTests.kt
@@ -67,7 +67,7 @@ public struct PutObjectPresignedURLMiddleware: ClientRuntime.Middleware {
         val presigner = PresignableUrlIntegration()
         val generator = AWSRestJson1ProtocolGenerator()
 
-        val codegenContext = GenerationContext(context.ctx.model, context.ctx.symbolProvider, context.ctx.settings, generator)
+        val codegenContext = GenerationContext(context.ctx.model, context.ctx.symbolProvider, context.ctx.settings, context.manifest, generator)
         val protocolGenerationContext = ProtocolGenerator.GenerationContext(context.ctx.settings, context.ctx.model, context.ctx.service, context.ctx.symbolProvider, listOf(), RestJson1Trait.ID, context.ctx.delegator)
         codegenContext.protocolGenerator?.initializeMiddleware(context.ctx)
         presigner.writeAdditionalFiles(codegenContext, protocolGenerationContext, context.ctx.delegator)