diff --git a/spring-cloud-function-adapters/spring-cloud-function-adapter-aws/src/test/java/org/springframework/cloud/function/adapter/aws/FunctionInvokerTests.java b/spring-cloud-function-adapters/spring-cloud-function-adapter-aws/src/test/java/org/springframework/cloud/function/adapter/aws/FunctionInvokerTests.java index 2158addcf..9dba9ec95 100644 --- a/spring-cloud-function-adapters/spring-cloud-function-adapter-aws/src/test/java/org/springframework/cloud/function/adapter/aws/FunctionInvokerTests.java +++ b/spring-cloud-function-adapters/spring-cloud-function-adapter-aws/src/test/java/org/springframework/cloud/function/adapter/aws/FunctionInvokerTests.java @@ -52,7 +52,6 @@ import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.messaging.Message; import org.springframework.messaging.converter.AbstractMessageConverter; -import org.springframework.messaging.support.MessageBuilder; import org.springframework.util.MimeType; import static org.assertj.core.api.Assertions.assertThat; @@ -765,24 +764,6 @@ public class FunctionInvokerTests { assertThat(result.get("body")).isEqualTo("\"Hello from ELB\""); } - @Test - public void testLBEventReturningMessage() throws Exception { - System.setProperty("MAIN_CLASS", LBConfiguration.class.getName()); - System.setProperty("spring.cloud.function.definition", "inputOutputLBEventAsMessage"); - FunctionInvoker invoker = new FunctionInvoker(); - - InputStream targetStream = new ByteArrayInputStream(this.sampleLBEvent.getBytes()); - ByteArrayOutputStream output = new ByteArrayOutputStream(); - invoker.handleRequest(targetStream, output, null); - - Map result = mapper.readValue(output.toByteArray(), Map.class); - System.out.println("RESULT: " + result); - assertThat(result.get("body")).isEqualTo("\"Hello\""); - assertThat((boolean) result.get("isBase64Encoded")).isFalse(); - assertThat(((Map) result.get("headers")).get("foo")).isEqualTo("bar"); - assertThat(result.get("statusCode")).isEqualTo(200); - } - @Test public void testLBEventAsMessage() throws Exception { System.setProperty("MAIN_CLASS", LBConfiguration.class.getName()); @@ -1317,14 +1298,6 @@ public class FunctionInvokerTests { }; } - @Bean - public Function> inputOutputLBEventAsMessage() { - return v -> { - Message message = MessageBuilder.withPayload("\"Hello\"".getBytes(StandardCharsets.UTF_8)).setHeader("foo", "bar").build(); - return message; - }; - } - @Bean public Function inputOutputLBEvent() { return v -> { diff --git a/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/catalog/SimpleFunctionRegistry.java b/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/catalog/SimpleFunctionRegistry.java index 3107884ca..17b7c66dd 100644 --- a/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/catalog/SimpleFunctionRegistry.java +++ b/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/catalog/SimpleFunctionRegistry.java @@ -1136,7 +1136,7 @@ public class SimpleFunctionRegistry implements FunctionRegistry { } if (convertedOutput instanceof Message) { - if (((Message) convertedOutput).getPayload() instanceof byte[]) { + if (((Message) convertedOutput).getPayload() instanceof byte[] && ObjectUtils.isEmpty(contentType)) { return convertedOutput; } else if (isExtractPayload((Message) convertedOutput, type)) {