BATCH-894: wrong import somehow crept into last commit

This commit is contained in:
dsyer
2008-11-07 18:54:24 +00:00
parent 2276fafd61
commit bf77927385
5 changed files with 20 additions and 22 deletions

View File

@@ -7,49 +7,48 @@ import static org.easymock.EasyMock.createMock;
import static org.junit.Assert.assertTrue;
import org.junit.Test;
import org.springframework.batch.core.ExitStatus;
import org.springframework.batch.core.JobExecution;
import org.springframework.batch.core.StepExecution;
import org.springframework.batch.core.repository.JobRepository;
import org.springframework.batch.core.step.AbstractStep;
import org.springframework.batch.repeat.ExitStatus;
/**
* @author Lucas Ward
*
*
*/
public class StepComponentBeanPostProcessorTests {
@BatchComponent
private class TestComponent {
boolean afterStepCalled = false;
@AfterStep
public void testMethod(){
public void testMethod() {
afterStepCalled = true;
}
}
@Test
public void testNormalCase() throws Exception{
public void testNormalCase() throws Exception {
AbstractStep step = new StubStep();
step.setJobRepository(createMock(JobRepository.class));
StepComponentBeanPostProcessor postProcessor = new StepComponentBeanPostProcessor(step, "org.springframework.batch.core.annotation");
StepComponentBeanPostProcessor postProcessor = new StepComponentBeanPostProcessor(
step, "org.springframework.batch.core.annotation");
TestComponent testComponent = new TestComponent();
postProcessor.postProcessAfterInitialization(testComponent, "testComponent");
postProcessor.postProcessAfterInitialization(testComponent,
"testComponent");
step.execute(new StepExecution("teststep", new JobExecution(11L)));
assertTrue(testComponent.afterStepCalled);
}
private class StubStep extends AbstractStep{
private class StubStep extends AbstractStep {
@Override
protected ExitStatus doExecute(StepExecution stepExecution)
throws Exception {
// TODO Auto-generated method stub
return ExitStatus.FINISHED;
protected void doExecute(StepExecution stepExecution) throws Exception {
stepExecution.setExitStatus(ExitStatus.FINISHED);
}
}
}

View File

@@ -1,8 +1,8 @@
package org.springframework.batch.core.configuration.xml;
import org.springframework.batch.core.ExitStatus;
import org.springframework.batch.core.StepExecution;
import org.springframework.batch.core.StepExecutionListener;
import org.springframework.batch.repeat.ExitStatus;
public class TestListener extends AbstractTestComponent implements StepExecutionListener {