diff --git a/spring-batch-infrastructure/src/test/java/org/springframework/batch/item/file/MultiResourceItemReaderFlatFileTests.java b/spring-batch-infrastructure/src/test/java/org/springframework/batch/item/file/MultiResourceItemReaderFlatFileTests.java index 1ae99fc19..227a05a8f 100644 --- a/spring-batch-infrastructure/src/test/java/org/springframework/batch/item/file/MultiResourceItemReaderFlatFileTests.java +++ b/spring-batch-infrastructure/src/test/java/org/springframework/batch/item/file/MultiResourceItemReaderFlatFileTests.java @@ -2,16 +2,15 @@ package org.springframework.batch.item.file; import java.util.Comparator; +import org.junit.internal.runners.JUnit4ClassRunner; +import org.junit.runner.RunWith; import org.springframework.batch.item.CommonItemStreamItemReaderTests; import org.springframework.batch.item.ExecutionContext; import org.springframework.batch.item.ItemReader; -import org.springframework.batch.item.file.mapping.FieldSet; -import org.springframework.batch.item.file.mapping.FieldSetMapper; +import org.springframework.batch.item.file.mapping.LineMapper; import org.springframework.batch.item.sample.Foo; import org.springframework.core.io.ByteArrayResource; import org.springframework.core.io.Resource; -import org.junit.runner.RunWith; -import org.junit.internal.runners.JUnit4ClassRunner; @RunWith(JUnit4ClassRunner.class) public class MultiResourceItemReaderFlatFileTests extends @@ -20,14 +19,16 @@ public class MultiResourceItemReaderFlatFileTests extends protected ItemReader getItemReader() throws Exception { MultiResourceItemReader multiReader = new MultiResourceItemReader(); - FlatFileItemReader fileReader = new FlatFileItemReader(); + ResourceLineReader fileReader = new ResourceLineReader(); - fileReader.setFieldSetMapper(new FieldSetMapper() { - public Foo process(FieldSet fs) { + fileReader.setLineMapper(new LineMapper() { + + public Foo mapLine(String line, int lineNumber) throws Exception { Foo foo = new Foo(); - foo.setValue(fs.readInt(0)); + foo.setValue(Integer.valueOf(line)); return foo; } + }); fileReader.setSaveState(true); diff --git a/spring-batch-infrastructure/src/test/java/org/springframework/batch/repeat/support/AbstractTradeBatchTests.java b/spring-batch-infrastructure/src/test/java/org/springframework/batch/repeat/support/AbstractTradeBatchTests.java index 9acf3128c..a313aae9e 100644 --- a/spring-batch-infrastructure/src/test/java/org/springframework/batch/repeat/support/AbstractTradeBatchTests.java +++ b/spring-batch-infrastructure/src/test/java/org/springframework/batch/repeat/support/AbstractTradeBatchTests.java @@ -22,9 +22,11 @@ import junit.framework.TestCase; import org.springframework.batch.item.ExecutionContext; import org.springframework.batch.item.ItemWriter; -import org.springframework.batch.item.file.FlatFileItemReader; +import org.springframework.batch.item.file.ResourceLineReader; +import org.springframework.batch.item.file.mapping.DefaultLineMapper; import org.springframework.batch.item.file.mapping.FieldSet; import org.springframework.batch.item.file.mapping.FieldSetMapper; +import org.springframework.batch.item.file.transform.DelimitedLineTokenizer; import org.springframework.core.io.ClassPathResource; import org.springframework.core.io.Resource; @@ -50,12 +52,15 @@ public abstract class AbstractTradeBatchTests extends TestCase { provider.open(new ExecutionContext()); } - protected static class TradeItemReader extends FlatFileItemReader { + protected static class TradeItemReader extends ResourceLineReader { protected TradeItemReader(Resource resource) throws Exception { super(); setResource(resource); - setFieldSetMapper(new TradeMapper()); + DefaultLineMapper mapper = new DefaultLineMapper(); + mapper.setLineTokenizer(new DelimitedLineTokenizer()); + mapper.setFieldSetMapper(new TradeMapper()); + setLineMapper(mapper); afterPropertiesSet(); }