SpringCloud/spring-cloud-function
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

GH-816 Enhance test validating collection output

Browse Source
This commit is contained in:
Oleg Zhurakousky
2022-02-23 15:08:18 +01:00
parent 3785a7c0b0
commit e347e20e80
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
spring-cloud-function-context/src/test/java/org/springframework/cloud/function/context/catalog/BeanFactoryAwareFunctionRegistryTests.java
View File

@@ -418,18 +418,25 @@ public class BeanFactoryAwareFunctionRegistryTests {
assertThat(result.getPayload()).isEqualTo("\"b2xsZWg=\"".getBytes()); assertThat(result.getPayload()).isEqualTo("\"b2xsZWg=\"".getBytes());
} }
@SuppressWarnings({ "rawtypes", "unchecked" })
@Test @Test
public void testMultipleValuesInOutputHandling() throws Exception { public void testMultipleValuesInOutputHandling() throws Exception {
FunctionCatalog catalog = this.configureCatalog(CollectionOutConfiguration.class); FunctionCatalog catalog = this.configureCatalog(CollectionOutConfiguration.class);
FunctionInvocationWrapper function = catalog.lookup("parseToList", "application/json"); FunctionInvocationWrapper function = catalog.lookup("parseToList", "application/json");
assertThat(function).isNotNull(); assertThat(function).isNotNull();
Object result = function.apply(MessageBuilder.withPayload("1, 2, 3".getBytes()).setHeader(MessageHeaders.CONTENT_TYPE, "text/plain").build()); Object result = function.apply(MessageBuilder.withPayload("1,2,3".getBytes()).setHeader(MessageHeaders.CONTENT_TYPE, "text/plain").build());
assertThat(result instanceof Message).isTrue(); assertThat(result instanceof Message).isTrue();
byte[] payload = ((Message<byte[]>) result).getPayload();
JsonMapper mapper = this.context.getBean(JsonMapper.class);
List resultList = mapper.fromJson(payload, List.class);
assertThat(resultList.size()).isEqualTo(3);
assertThat(resultList.get(0)).isEqualTo("1");
assertThat(resultList.get(1)).isEqualTo("2");
function = catalog.lookup("parseToListOfMessages", "application/json"); function = catalog.lookup("parseToListOfMessages", "application/json");
assertThat(function).isNotNull(); assertThat(function).isNotNull();
result = function.apply(MessageBuilder.withPayload("1, 2, 3".getBytes()).setHeader(MessageHeaders.CONTENT_TYPE, "text/plain").build()); result = function.apply(MessageBuilder.withPayload("1,2,3".getBytes()).setHeader(MessageHeaders.CONTENT_TYPE, "text/plain").build());
assertThat(result instanceof Message).isFalse(); assertThat(result instanceof List).isTrue();
} }
/** /**