Spring/spring-batch
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

BATCH-1743: extract target from proxy before making assumptions about type

Browse Source
This commit is contained in:
Dave Syer
2011-05-04 09:52:07 +01:00
parent 5b81bf0361
commit 6dfc3a4a61
3 changed files with 60 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
spring-batch-core/src/test/java/org/springframework/batch/core/configuration/xml/StepParserStepFactoryBeanTests.java
View File

@@ -22,10 +22,13 @@ import static org.junit.Assert.assertTrue;
import java.util.HashMap;
import org.junit.Test;
import org.springframework.aop.framework.Advised;
import org.springframework.aop.framework.ProxyFactory;
import org.springframework.batch.core.StepListener;
import org.springframework.batch.core.job.flow.FlowStep;
import org.springframework.batch.core.job.flow.support.SimpleFlow;
import org.springframework.batch.core.listener.StepExecutionListenerSupport;
import org.springframework.batch.core.partition.PartitionHandler;
import org.springframework.batch.core.partition.support.PartitionStep;
import org.springframework.batch.core.partition.support.SimplePartitioner;
import org.springframework.batch.core.partition.support.TaskExecutorPartitionHandler;
@@ -254,6 +257,28 @@ public class StepParserStepFactoryBeanTests {
assertTrue(handler instanceof TaskExecutorPartitionHandler);
}
@Test
public void testPartitionStepWithProxyHandler() throws Exception {
StepParserStepFactoryBean<Object, Object> fb = new StepParserStepFactoryBean<Object, Object>();
fb.setBeanName("step1");
fb.setAllowStartIfComplete(true);
fb.setJobRepository(new JobRepositorySupport());
fb.setStartLimit(5);
fb.setListeners(new StepListener[] { new StepExecutionListenerSupport() });
fb.setTaskExecutor(new SyncTaskExecutor());
SimplePartitioner partitioner = new SimplePartitioner();
fb.setPartitioner(partitioner);
fb.setStep(new StepSupport("foo"));
ProxyFactory factory = new ProxyFactory(new TaskExecutorPartitionHandler());
fb.setPartitionHandler((PartitionHandler) factory.getProxy());
Object step = fb.getObject();
assertTrue(step instanceof PartitionStep);
Object handler = ReflectionTestUtils.getField(step, "partitionHandler");
assertTrue(handler instanceof Advised);
}
@Test
public void testFlowStep() throws Exception {
StepParserStepFactoryBean<Object, Object> fb = new StepParserStepFactoryBean<Object, Object>();