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

GH-676 Add split condition test for header enrichment

Browse Source
This commit is contained in:
Oleg Zhurakousky
2021-05-20 07:31:48 +02:00
parent 0db0a2f555
commit 79f4540130
1 changed files with 26 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
spring-cloud-function-context/src/test/java/org/springframework/cloud/function/context/FunctionPropertiesTests.java
View File

@@ -106,22 +106,22 @@ public class FunctionPropertiesTests {
}
}
// @Test
// public void testInputHeaderMappingPropertyWithCompositionWithoutIndex() throws Exception {
// try (ConfigurableApplicationContext context = new SpringApplicationBuilder(
// SampleFunctionConfiguration.class).web(WebApplicationType.NONE).run(
// "--logging.level.org.springframework.cloud.function=DEBUG",
// "--spring.main.lazy-initialization=true",
// "--spring.cloud.function.definition=echo|foo",
// "--spring.cloud.function.configuration.echofoo.input-header-mapping-expression.key1=hello1",
// "--spring.cloud.function.configuration.echofoo.input-header-mapping-expression.key2=hello2")) {
// FunctionProperties functionProperties = context
// .getBean(FunctionProperties.class);
// FunctionConfigurationProperties configuration = functionProperties
// .getConfiguration().get("echofoo");
// assertThat(configuration.getInputHeaderMappingExpression()).containsKey("0");
// }
// }
@Test
public void testInputHeaderMappingPropertyWithSplitExpression() throws Exception {
try (ConfigurableApplicationContext context = new SpringApplicationBuilder(
SampleFunctionConfiguration.class).web(WebApplicationType.NONE).run(
"--logging.level.org.springframework.cloud.function=DEBUG",
"--spring.main.lazy-initialization=true",
"--spring.cloud.function.configuration.split.input-header-mapping-expression.key1=headers.path.split('/')[0]",
"--spring.cloud.function.configuration.split.input-header-mapping-expression.key2=headers.path.split('/')[1]",
"--spring.cloud.function.configuration.split.input-header-mapping-expression.key3=headers.path")) {
FunctionCatalog functionCatalog = context.getBean(FunctionCatalog.class);
FunctionInvocationWrapper function = functionCatalog.lookup("split");
function.apply(MessageBuilder.withPayload("helo")
.setHeader(MessageHeaders.CONTENT_TYPE, "application/json")
.setHeader("path", "foo/bar/baz").build());
}
}
@EnableAutoConfiguration
@Configuration
@@ -137,6 +137,16 @@ public class FunctionPropertiesTests {
};
}
@Bean
public Function<Message<?>, Message<?>> split() {
return m -> {
assertThat(m.getHeaders().get("key1")).isEqualTo("foo");
assertThat(m.getHeaders().get("key2")).isEqualTo("bar");
assertThat(m.getHeaders().get("key3")).isEqualTo("foo/bar/baz");
return m;
};
}
@Bean
public Function<Message<?>, Message<?>> foo() {
return x -> x;