Made resolver a parameter

smithy-lang · Oct 12, 2023 · 8e562e7 · 8e562e7
1 parent 9227b7a
commit 8e562e7
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/.../software/amazon/smithy/kotlin/codegen/rendering/protocol/HttpBindingProtocolGenerator.kt b/.../software/amazon/smithy/kotlin/codegen/rendering/protocol/HttpBindingProtocolGenerator.kt
@@ -266,11 +266,15 @@ abstract class HttpBindingProtocolGenerator : ProtocolGenerator {
             writer.write("val payload = #T(context, input)", opBodySerializerFn)
             writer.write("builder.body = #T.fromBytes(payload)", RuntimeTypes.Http.HttpBody)
         }
-        renderContentTypeHeader(ctx, op, writer)
+        renderContentTypeHeader(ctx, op, writer, resolver)
     }
 
-    protected open fun renderContentTypeHeader(ctx: ProtocolGenerator.GenerationContext, op: OperationShape, writer: KotlinWriter) {
-        val resolver = getProtocolHttpBindingResolver(ctx.model, ctx.service)
+    protected open fun renderContentTypeHeader(
+        ctx: ProtocolGenerator.GenerationContext,
+        op: OperationShape,
+        writer: KotlinWriter,
+        resolver: HttpBindingResolver = getProtocolHttpBindingResolver(ctx.model, ctx.service),
+    ) {
         resolver.determineRequestContentType(op)?.let { contentType ->
             writer.withBlock("if (builder.body !is HttpBody.Empty) {", "}") {
                 write("builder.headers.setMissing(\"Content-Type\", #S)", contentType)