INT-1060 Added @Filter annotation.
This commit is contained in:
Mark Fisher
@@ -0,0 +1,135 @@
|
||||
/*
|
||||
* Copyright 2002-2010 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.integration.config.annotation;
|
||||
|
||||
import static org.junit.Assert.assertNotNull;
|
||||
import static org.junit.Assert.assertNull;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
|
||||
import org.springframework.integration.annotation.Filter;
|
||||
import org.springframework.integration.annotation.MessageEndpoint;
|
||||
import org.springframework.integration.channel.DirectChannel;
|
||||
import org.springframework.integration.channel.QueueChannel;
|
||||
import org.springframework.integration.core.Message;
|
||||
import org.springframework.integration.message.StringMessage;
|
||||
import org.springframework.integration.test.util.TestUtils;
|
||||
import org.springframework.integration.test.util.TestUtils.TestApplicationContext;
|
||||
|
||||
/**
|
||||
* @author Mark Fisher
|
||||
* @since 2.0
|
||||
*/
|
||||
public class FilterAnnotationPostProcessorTests {
|
||||
|
||||
private final TestApplicationContext context = TestUtils.createTestApplicationContext();
|
||||
|
||||
private final MessagingAnnotationPostProcessor postProcessor = new MessagingAnnotationPostProcessor();
|
||||
|
||||
private final DirectChannel inputChannel = new DirectChannel();
|
||||
|
||||
private final QueueChannel outputChannel = new QueueChannel();
|
||||
|
||||
|
||||
@Before
|
||||
public void init() {
|
||||
context.registerChannel("input", inputChannel);
|
||||
context.registerChannel("output", outputChannel);
|
||||
postProcessor.setBeanFactory(context.getBeanFactory());
|
||||
postProcessor.afterPropertiesSet();
|
||||
}
|
||||
|
||||
|
||||
@Test
|
||||
public void filterAnnotationWithBooleanPrimitive() {
|
||||
testValidFilter(new TestFilterWithBooleanPrimitive());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void filterAnnotationWithBooleanWrapperClass() {
|
||||
testValidFilter(new TestFilterWithBooleanWrapperClass());
|
||||
}
|
||||
|
||||
@Test(expected = IllegalArgumentException.class)
|
||||
public void invalidMethodWithStringReturnType() {
|
||||
Object filter = new TestFilterWithStringReturnType();
|
||||
postProcessor.postProcessAfterInitialization(filter, "testFilter");
|
||||
}
|
||||
|
||||
@Test(expected = IllegalArgumentException.class)
|
||||
public void invalidMethodWithVoidReturnType() {
|
||||
Object filter = new TestFilterWithVoidReturnType();
|
||||
postProcessor.postProcessAfterInitialization(filter, "testFilter");
|
||||
}
|
||||
|
||||
|
||||
private void testValidFilter(Object filter) {
|
||||
postProcessor.postProcessAfterInitialization(filter, "testFilter");
|
||||
context.refresh();
|
||||
inputChannel.send(new StringMessage("good"));
|
||||
Message<?> passed = outputChannel.receive(0);
|
||||
assertNotNull(passed);
|
||||
inputChannel.send(new StringMessage("bad"));
|
||||
assertNull(outputChannel.receive(0));
|
||||
context.stop();
|
||||
}
|
||||
|
||||
|
||||
@MessageEndpoint
|
||||
@SuppressWarnings("unused")
|
||||
private static class TestFilterWithBooleanPrimitive {
|
||||
|
||||
@Filter(inputChannel="input", outputChannel="output")
|
||||
public boolean filter(String s) {
|
||||
return !s.contains("bad");
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@MessageEndpoint
|
||||
@SuppressWarnings("unused")
|
||||
private static class TestFilterWithBooleanWrapperClass {
|
||||
|
||||
@Filter(inputChannel="input", outputChannel="output")
|
||||
public Boolean filter(String s) {
|
||||
return !s.contains("bad");
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@MessageEndpoint
|
||||
@SuppressWarnings("unused")
|
||||
private static class TestFilterWithStringReturnType {
|
||||
|
||||
@Filter(inputChannel="input", outputChannel="output")
|
||||
public String filter(String s) {
|
||||
return s;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@MessageEndpoint
|
||||
@SuppressWarnings("unused")
|
||||
private static class TestFilterWithVoidReturnType {
|
||||
|
||||
@Filter(inputChannel="input", outputChannel="output")
|
||||
public void filter(String s) {
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,83 @@
|
||||
/*
|
||||
* Copyright 2002-2010 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.integration.filter;
|
||||
|
||||
import static org.junit.Assert.assertFalse;
|
||||
import static org.junit.Assert.assertNotNull;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
|
||||
import org.junit.Test;
|
||||
|
||||
import org.springframework.beans.factory.support.DefaultListableBeanFactory;
|
||||
import org.springframework.integration.annotation.Filter;
|
||||
import org.springframework.integration.channel.QueueChannel;
|
||||
import org.springframework.integration.config.FilterFactoryBean;
|
||||
import org.springframework.integration.core.Message;
|
||||
import org.springframework.integration.message.MessageBuilder;
|
||||
import org.springframework.integration.message.MessageHandler;
|
||||
|
||||
/**
|
||||
* @author Mark Fisher
|
||||
* @since 2.0
|
||||
*/
|
||||
public class FilterAnnotationMethodResolutionTests {
|
||||
|
||||
@Test
|
||||
public void resolveAnnotatedMethod() throws Exception {
|
||||
FilterFactoryBean factoryBean = new FilterFactoryBean();
|
||||
factoryBean.setBeanFactory(new DefaultListableBeanFactory());
|
||||
AnnotatedTestFilter filter = new AnnotatedTestFilter();
|
||||
factoryBean.setTargetObject(filter);
|
||||
MessageHandler handler = factoryBean.getObject();
|
||||
QueueChannel replyChannel = new QueueChannel();
|
||||
handler.handleMessage(MessageBuilder.withPayload("test").setReplyChannel(replyChannel).build());
|
||||
Message<?> result = replyChannel.receive(0);
|
||||
assertNotNull(result);
|
||||
assertTrue(filter.invokedCorrectMethod);
|
||||
assertFalse(filter.invokedIncorrectMethod);
|
||||
}
|
||||
|
||||
|
||||
public static class AnnotatedTestFilter {
|
||||
|
||||
private volatile boolean invokedCorrectMethod;
|
||||
|
||||
private volatile boolean invokedIncorrectMethod;
|
||||
|
||||
|
||||
public String notThisOne(String s) {
|
||||
this.invokedIncorrectMethod = true;
|
||||
return s;
|
||||
}
|
||||
|
||||
public void norThisOne(String s) {
|
||||
this.invokedIncorrectMethod = true;
|
||||
}
|
||||
|
||||
public boolean andNotEvenThisOne(String s) {
|
||||
this.invokedIncorrectMethod = true;
|
||||
return true;
|
||||
}
|
||||
|
||||
@Filter
|
||||
public boolean thisOne(String s) {
|
||||
this.invokedCorrectMethod = true;
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user