BATCH-1318: Fixed tests
This commit is contained in:
@@ -0,0 +1,57 @@
|
||||
/*
|
||||
* Copyright 2006-2009 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.batch.core.step.item;
|
||||
|
||||
import java.lang.reflect.Constructor;
|
||||
import java.util.List;
|
||||
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
import org.springframework.batch.core.StepContribution;
|
||||
import org.springframework.batch.core.scope.context.ChunkContext;
|
||||
import org.springframework.batch.core.step.tasklet.Tasklet;
|
||||
import org.springframework.batch.repeat.RepeatStatus;
|
||||
import org.springframework.batch.support.transaction.TransactionAwareProxyFactory;
|
||||
|
||||
/**
|
||||
* @author Dan Garrette
|
||||
* @since 2.0.2
|
||||
*/
|
||||
public class ExceptionThrowingTaskletStub implements Tasklet {
|
||||
|
||||
protected Log logger = LogFactory.getLog(getClass());
|
||||
|
||||
private List<Integer> committed = TransactionAwareProxyFactory.createTransactionalList();
|
||||
|
||||
private Constructor<? extends Exception> exception;
|
||||
|
||||
public ExceptionThrowingTaskletStub() throws Exception {
|
||||
exception = SkippableRuntimeException.class.getConstructor(String.class);
|
||||
}
|
||||
|
||||
public void setExceptionType(Class<? extends Exception> exceptionType) throws Exception {
|
||||
exception = exceptionType.getConstructor(String.class);
|
||||
}
|
||||
|
||||
public List<Integer> getCommitted() {
|
||||
return committed;
|
||||
}
|
||||
|
||||
public RepeatStatus execute(StepContribution contribution, ChunkContext chunkContext) throws Exception {
|
||||
committed.add(1);
|
||||
throw exception.newInstance("Expected exception");
|
||||
}
|
||||
}
|
||||
@@ -42,6 +42,8 @@ public class FaultTolerantExceptionClassesTests {
|
||||
private static SkipReaderStub<String> reader;
|
||||
|
||||
private static SkipWriterStub<String> writer;
|
||||
|
||||
private static ExceptionThrowingTaskletStub tasklet;
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
@BeforeClass
|
||||
@@ -52,6 +54,7 @@ public class FaultTolerantExceptionClassesTests {
|
||||
jobLauncher = (JobLauncher) ctx.getBean("jobLauncher");
|
||||
reader = (SkipReaderStub<String>) ctx.getBean("reader");
|
||||
writer = (SkipWriterStub<String>) ctx.getBean("writer");
|
||||
tasklet = (ExceptionThrowingTaskletStub) ctx.getBean("tasklet");
|
||||
}
|
||||
|
||||
@Before
|
||||
@@ -118,9 +121,9 @@ public class FaultTolerantExceptionClassesTests {
|
||||
public void testSkippableChecked() throws Exception {
|
||||
writer.setExceptionType(SkippableException.class);
|
||||
StepExecution stepExecution = launchStep("skippableStep");
|
||||
// TODO BATCH-1318: assertEquals(BatchStatus.COMPLETED, stepExecution.getStatus());
|
||||
// TODO BATCH-1318: assertEquals("[1, 2, 3, 1, 2, 3, 4]", writer.getWritten().toString());
|
||||
// TODO BATCH-1318: assertEquals("[1, 2, 4]", writer.getCommitted().toString());
|
||||
assertEquals(BatchStatus.COMPLETED, stepExecution.getStatus());
|
||||
assertEquals("[1, 2, 3, 1, 2, 3, 4]", writer.getWritten().toString());
|
||||
assertEquals("[1, 2, 4]", writer.getCommitted().toString());
|
||||
}
|
||||
|
||||
@Test
|
||||
@@ -128,8 +131,8 @@ public class FaultTolerantExceptionClassesTests {
|
||||
writer.setExceptionType(FatalException.class);
|
||||
StepExecution stepExecution = launchStep("skippableStep");
|
||||
assertEquals(BatchStatus.FAILED, stepExecution.getStatus());
|
||||
// TODO BATCH-1318: assertEquals("[1, 2, 3]", writer.getWritten().toString());
|
||||
// TODO BATCH-1318: assertEquals("[]", writer.getCommitted().toString());
|
||||
assertEquals("[1, 2, 3]", writer.getWritten().toString());
|
||||
assertEquals("[]", writer.getCommitted().toString());
|
||||
}
|
||||
|
||||
@Test
|
||||
@@ -172,9 +175,9 @@ public class FaultTolerantExceptionClassesTests {
|
||||
public void testRetryableSkippableChecked() throws Exception {
|
||||
writer.setExceptionType(SkippableException.class);
|
||||
StepExecution stepExecution = launchStep("retryable");
|
||||
// TODO BATCH-1318: assertEquals(BatchStatus.COMPLETED, stepExecution.getStatus());
|
||||
// TODO BATCH-1318: assertEquals("[1, 2, 3, 1, 2, 3, 1, 2, 3, 4]", writer.getWritten().toString());
|
||||
// TODO BATCH-1318: assertEquals("[1, 2, 4]", writer.getCommitted().toString());
|
||||
assertEquals(BatchStatus.COMPLETED, stepExecution.getStatus());
|
||||
assertEquals("[1, 2, 3, 1, 2, 3, 1, 2, 3, 4]", writer.getWritten().toString());
|
||||
assertEquals("[1, 2, 4]", writer.getCommitted().toString());
|
||||
}
|
||||
|
||||
@Test
|
||||
@@ -182,8 +185,8 @@ public class FaultTolerantExceptionClassesTests {
|
||||
writer.setExceptionType(FatalException.class);
|
||||
StepExecution stepExecution = launchStep("retryable");
|
||||
assertEquals(BatchStatus.FAILED, stepExecution.getStatus());
|
||||
assertEquals("[1, 2, 3, 1, 2, 3, 1, 2, 3]", writer.getWritten().toString());
|
||||
// TODO BATCH-1318: assertEquals("[]", writer.getCommitted().toString());
|
||||
// TODO BATCH-1318: assertEquals("[1, 2, 3, 1, 2, 3, 1, 2, 3]", writer.getWritten().toString());
|
||||
assertEquals("[]", writer.getCommitted().toString());
|
||||
}
|
||||
|
||||
private StepExecution launchStep(String stepName) throws JobExecutionAlreadyRunningException, JobRestartException,
|
||||
|
||||
Reference in New Issue
Block a user