GH-929 Ensure AWS Functioininvoker handles Mono<Void> return the same way as imperative Consumer

Resolves #929
This commit is contained in:
Oleg Zhurakousky
2022-10-19 10:51:55 +02:00
parent e6985724d9
commit d613d9a4c1
2 changed files with 34 additions and 10 deletions

View File

@@ -44,6 +44,7 @@ import com.fasterxml.jackson.databind.ObjectMapper;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
import org.springframework.cloud.function.json.JsonMapper;
@@ -1024,6 +1025,21 @@ public class FunctionInvokerTests {
assertThat(result.get("body")).isEqualTo("\"OK\"");
}
@SuppressWarnings("rawtypes")
@Test
public void testApiGatewayWithMonoVoidAsReturn() throws Exception {
System.setProperty("MAIN_CLASS", ApiGatewayConfiguration.class.getName());
System.setProperty("spring.cloud.function.definition", "reactiveWithVoidReturn");
FunctionInvoker invoker = new FunctionInvoker();
InputStream targetStream = new ByteArrayInputStream(this.apiGatewayEvent.getBytes());
ByteArrayOutputStream output = new ByteArrayOutputStream();
invoker.handleRequest(targetStream, output, null);
Map result = mapper.readValue(output.toByteArray(), Map.class);
assertThat(result.get("body")).isEqualTo("\"OK\"");
}
@Test
public void testWithDefaultRoutingFailure() throws Exception {
System.setProperty("MAIN_CLASS", SampleConfiguration.class.getName());
@@ -1315,6 +1331,11 @@ public class FunctionInvokerTests {
return v -> v.toUpperCase();
}
@Bean
public Function<Mono<String>, Mono<Void>> reactiveWithVoidReturn() {
return v -> Mono.empty();
}
@Bean
public Function<Person, String> uppercasePojo() {
return v -> {