Improve output type conversion handling
- Re-enable, clean and improve special handling for collection/array output type - Add tests to validate and demonstrate the differences in special handling of collection of Messages ve collection of non-Messages Resolves #464
This commit is contained in:
@@ -26,6 +26,7 @@ import java.util.Map;
|
||||
import java.util.function.Consumer;
|
||||
import java.util.function.Function;
|
||||
import java.util.function.Supplier;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
import org.junit.Ignore;
|
||||
import org.junit.Test;
|
||||
@@ -50,6 +51,7 @@ import org.springframework.messaging.converter.MessageConverter;
|
||||
import org.springframework.messaging.support.GenericMessage;
|
||||
import org.springframework.messaging.support.MessageBuilder;
|
||||
import org.springframework.stereotype.Component;
|
||||
import org.springframework.util.CollectionUtils;
|
||||
import org.springframework.util.MimeType;
|
||||
import org.springframework.util.ReflectionUtils;
|
||||
|
||||
@@ -336,6 +338,7 @@ public class BeanFactoryAwareFunctionRegistryTests {
|
||||
assertThat(composed).isFalse();
|
||||
}
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
@Test
|
||||
public void byteArrayNoSpecialHandling() throws Exception {
|
||||
FunctionCatalog catalog = this.configureCatalog(ByteArrayFunction.class);
|
||||
@@ -345,6 +348,21 @@ public class BeanFactoryAwareFunctionRegistryTests {
|
||||
assertThat(result.getPayload()).isEqualTo("\"b2xsZWg=\"".getBytes());
|
||||
}
|
||||
|
||||
@SuppressWarnings("rawtypes")
|
||||
@Test
|
||||
public void testMultipleValuesInOutputHandling() throws Exception {
|
||||
FunctionCatalog catalog = this.configureCatalog(CollectionOutConfiguration.class);
|
||||
FunctionInvocationWrapper function = catalog.lookup("parseToList", "application/json");
|
||||
assertThat(function).isNotNull();
|
||||
Object result = (Message) function.apply(MessageBuilder.withPayload("1, 2, 3".getBytes()).setHeader(MessageHeaders.CONTENT_TYPE, "text/plain").build());
|
||||
assertThat(result instanceof Message).isTrue();
|
||||
|
||||
function = catalog.lookup("parseToListOfMessages", "application/json");
|
||||
assertThat(function).isNotNull();
|
||||
result = function.apply(MessageBuilder.withPayload("1, 2, 3".getBytes()).setHeader(MessageHeaders.CONTENT_TYPE, "text/plain").build());
|
||||
assertThat(result instanceof Message).isFalse();
|
||||
}
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
@Test
|
||||
public void testSerializationWithCompatibleWildcardSubtypeAcceptHeader() {
|
||||
@@ -370,6 +388,25 @@ public class BeanFactoryAwareFunctionRegistryTests {
|
||||
assertThat(dateResult.getHeaders().get("accept")).isNull();
|
||||
}
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
@EnableAutoConfiguration
|
||||
public static class CollectionOutConfiguration {
|
||||
|
||||
@Bean
|
||||
public Function<String, List<String>> parseToList() {
|
||||
return v -> CollectionUtils.arrayToList(v.split(","));
|
||||
}
|
||||
|
||||
@Bean
|
||||
public Function<String, List<Message<String>>> parseToListOfMessages() {
|
||||
return v -> {
|
||||
List<Message<String>> list = (List<Message<String>>) CollectionUtils.arrayToList(v.split(",")).stream()
|
||||
.map(value -> MessageBuilder.withPayload(value).build()).collect(Collectors.toList());
|
||||
return list;
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
@EnableAutoConfiguration
|
||||
public static class NegotiatingMessageConverterConfiguration {
|
||||
|
||||
|
||||
Reference in New Issue
Block a user