diff --git a/codegen-server/src/main/kotlin/software/amazon/smithy/rust/codegen/server/smithy/generators/protocol/ServerProtocolTestGenerator.kt b/codegen-server/src/main/kotlin/software/amazon/smithy/rust/codegen/server/smithy/generators/protocol/ServerProtocolTestGenerator.kt index 6448a3f6bc04222ed04fa9c017aeac0a7b88c82d..98a02ef6c462dfcecffd64e113a64822f1dafb57 100644 --- a/codegen-server/src/main/kotlin/software/amazon/smithy/rust/codegen/server/smithy/generators/protocol/ServerProtocolTestGenerator.kt +++ b/codegen-server/src/main/kotlin/software/amazon/smithy/rust/codegen/server/smithy/generators/protocol/ServerProtocolTestGenerator.kt @@ -650,7 +650,6 @@ class ServerProtocolTestGenerator( FailingTest(RestJson, "RestJsonFooErrorWithDunderType", TestType.Response), FailingTest(RestJson, "RestJsonFooErrorWithDunderTypeAndNamespace", TestType.Response), FailingTest(RestJson, "RestJsonFooErrorWithDunderTypeUriAndNamespace", TestType.Response), - FailingTest(RestJson, "RestJsonHttpResponseCode", TestType.Response), FailingTest(RestJson, "RestJsonNoInputAndNoOutput", TestType.Response), FailingTest(RestJson, "RestJsonStreamingTraitsRequireLengthWithBlob", TestType.Response), FailingTest(RestJson, "RestJsonHttpWithEmptyBlobPayload", TestType.Request), diff --git a/codegen-server/src/main/kotlin/software/amazon/smithy/rust/codegen/server/smithy/protocols/ServerHttpProtocolGenerator.kt b/codegen-server/src/main/kotlin/software/amazon/smithy/rust/codegen/server/smithy/protocols/ServerHttpProtocolGenerator.kt index f41c64fc2bba2d654cb5eb8d898579fabc718299..89becf201c392a9529aeb469aa8cb4d1b5c276fa 100644 --- a/codegen-server/src/main/kotlin/software/amazon/smithy/rust/codegen/server/smithy/protocols/ServerHttpProtocolGenerator.kt +++ b/codegen-server/src/main/kotlin/software/amazon/smithy/rust/codegen/server/smithy/protocols/ServerHttpProtocolGenerator.kt @@ -465,7 +465,7 @@ private class ServerHttpProtocolImplGenerator( Attribute.AllowUnusedMut.render(this) rustTemplate("let mut builder = #{http}::Response::builder();", *codegenScope) serverRenderResponseHeaders(operationShape) - bindings.find { it.location == HttpLocation.RESPONSE_CODE }?.let { serverRenderResponseCodeBinding(it) } + bindings.find { it.location == HttpLocation.RESPONSE_CODE }?.let { serverRenderResponseCodeBinding(it)(this) } operationShape.outputShape(model).findStreamingMember(model)?.let { val memberName = symbolProvider.toMemberName(it)