GH-1025 Ensure DELETE can only be mapped to a Consumer

This commit is contained in:
Oleg Zhurakousky
2023-05-31 21:48:57 +02:00
parent f6410af8e1
commit 863934709d
3 changed files with 24 additions and 20 deletions

View File

@@ -40,6 +40,7 @@ import org.springframework.http.ResponseEntity.BodyBuilder;
import org.springframework.messaging.Message; import org.springframework.messaging.Message;
import org.springframework.messaging.support.MessageBuilder; import org.springframework.messaging.support.MessageBuilder;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import org.springframework.util.Assert;
import org.springframework.util.CollectionUtils; import org.springframework.util.CollectionUtils;
import org.springframework.util.MultiValueMap; import org.springframework.util.MultiValueMap;
import org.springframework.util.StringUtils; import org.springframework.util.StringUtils;
@@ -161,6 +162,7 @@ public class FunctionController {
public void delete(WebRequest request, @RequestBody(required = false) String body) { public void delete(WebRequest request, @RequestBody(required = false) String body) {
FunctionWrapper wrapper = wrapper(request); FunctionWrapper wrapper = wrapper(request);
if (FunctionWebRequestProcessingHelper.isValidFunction("DELETE", wrapper.getFunction().getFunctionDefinition(), this.functionHttpProperties)) { if (FunctionWebRequestProcessingHelper.isValidFunction("DELETE", wrapper.getFunction().getFunctionDefinition(), this.functionHttpProperties)) {
Assert.isTrue(wrapper.getFunction().isConsumer(), "'DELETE' can only be mapped to Consumer");
FunctionWebRequestProcessingHelper.processRequest(wrapper, wrapper.getArgument(), false); FunctionWebRequestProcessingHelper.processRequest(wrapper, wrapper.getArgument(), false);
} }
else { else {

View File

@@ -16,24 +16,18 @@
package org.springframework.cloud.function.web.mvc; package org.springframework.cloud.function.web.mvc;
import static org.assertj.core.api.Assertions.assertThat;
import java.net.URI; import java.net.URI;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.function.Consumer; import java.util.function.Consumer;
import java.util.function.Function; import java.util.function.Function;
import java.util.stream.Collectors;
import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Disabled;
import org.junit.jupiter.api.Test; import org.junit.jupiter.api.Test;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.SpringApplication; import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.EnableAutoConfiguration; import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
import org.springframework.boot.test.context.SpringBootTest; import org.springframework.boot.test.context.SpringBootTest;
@@ -45,27 +39,19 @@ import org.springframework.cloud.function.web.RestApplication;
import org.springframework.cloud.function.web.mvc.HttpDeleteIntegrationTests.ApplicationConfiguration; import org.springframework.cloud.function.web.mvc.HttpDeleteIntegrationTests.ApplicationConfiguration;
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Bean;
import org.springframework.http.HttpStatus; import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.RequestEntity; import org.springframework.http.RequestEntity;
import org.springframework.http.ResponseEntity; import org.springframework.http.ResponseEntity;
import org.springframework.messaging.Message; import org.springframework.messaging.Message;
import org.springframework.messaging.support.MessageBuilder;
import org.springframework.test.context.ContextConfiguration; import org.springframework.test.context.ContextConfiguration;
import org.springframework.util.LinkedMultiValueMap; import org.springframework.util.MimeType;
import org.springframework.util.MultiValueMap;
import org.springframework.util.StringUtils;
import static org.assertj.core.api.Assertions.assertThat;
/** /**
* @author Oleg Zhurakousky * @author Oleg Zhurakousky
*/ */
@SpringBootTest(webEnvironment = WebEnvironment.RANDOM_PORT, properties = "spring.main.web-application-type=servlet") @SpringBootTest(webEnvironment = WebEnvironment.RANDOM_PORT, properties = "spring.main.web-application-type=servlet")
@ContextConfiguration(classes = { RestApplication.class, ApplicationConfiguration.class }) @ContextConfiguration(classes = {ApplicationConfiguration.class})
public class HttpDeleteIntegrationTests { public class HttpDeleteIntegrationTests {
private static final MediaType EVENT_STREAM = MediaType.TEXT_EVENT_STREAM;
@LocalServerPort @LocalServerPort
private int port; private int port;
@@ -98,6 +84,14 @@ public class HttpDeleteIntegrationTests {
assertThat(result.getStatusCode()).isEqualTo(HttpStatus.NO_CONTENT); assertThat(result.getStatusCode()).isEqualTo(HttpStatus.NO_CONTENT);
} }
@Test
public void testDeleteWithFunction() throws Exception {
ResponseEntity<Void> result = this.rest.exchange(
RequestEntity.delete(new URI("/deleteFunction"))
.build(), Void.class);
assertThat(result.getStatusCode()).isEqualTo(HttpStatus.INTERNAL_SERVER_ERROR);
}
@EnableAutoConfiguration @EnableAutoConfiguration
@TestConfiguration @TestConfiguration
public static class ApplicationConfiguration { public static class ApplicationConfiguration {
@@ -109,6 +103,16 @@ public class HttpDeleteIntegrationTests {
args); args);
} }
@Bean
public Function<String, String> deleteFunction() {
return v -> {
assertThat(v).isEqualTo("123");
System.out.println("Deleting: " + v);
return null;
};
}
@Bean @Bean
public Consumer<String> deleteConsumer() { public Consumer<String> deleteConsumer() {
return v -> { return v -> {

View File

@@ -64,8 +64,6 @@ import static org.assertj.core.api.Assertions.assertThat;
@ContextConfiguration(classes = { RestApplication.class, ApplicationConfiguration.class }) @ContextConfiguration(classes = { RestApplication.class, ApplicationConfiguration.class })
public class HttpPostIntegrationTests { public class HttpPostIntegrationTests {
private static final MediaType EVENT_STREAM = MediaType.TEXT_EVENT_STREAM;
@LocalServerPort @LocalServerPort
private int port; private int port;