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

RESOLVED - issue BATCH-1253: Ability to throttle the number of rows returned by a HibernateCursorItemReader

Browse Source
This commit is contained in:
dsyer
2009-07-07 08:45:10 +00:00
parent 456c16962b
commit e516cbd4a0
2 changed files with 140 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

78
spring-batch-infrastructure/src/test/java/org/springframework/batch/item/file/FlatFileItemReaderTests.java
View File

@@ -103,6 +103,84 @@ public class FlatFileItemReaderTests {
assertEquals(4, executionContext.getInt(ClassUtils.getShortName(FlatFileItemReader.class) + ".read.count"));
}
@Test
public void testCurrentItemCount() throws Exception {
reader.setCurrentItemCount(2);
reader.open(executionContext);
// read some records
reader.read();
reader.read();
// get restart data
reader.update(executionContext);
assertEquals(4, executionContext.getInt(ClassUtils.getShortName(FlatFileItemReader.class) + ".read.count"));
// close input
reader.close();
}
@Test
public void testMaxItemCount() throws Exception {
reader.setMaxItemCount(2);
reader.open(executionContext);
// read some records
reader.read();
reader.read();
// get restart data
reader.update(executionContext);
assertNull(reader.read());
assertEquals(2, executionContext.getInt(ClassUtils.getShortName(FlatFileItemReader.class) + ".read.count"));
// close input
reader.close();
}
@Test
public void testMaxItemCountFromContext() throws Exception {
reader.setMaxItemCount(2);
executionContext.putInt(reader.getClass().getSimpleName()+".read.count.max", Integer.MAX_VALUE);
reader.open(executionContext);
// read some records
reader.read();
reader.read();
assertNotNull(reader.read());
// close input
reader.close();
}
@Test
public void testCurrentItemCountFromContext() throws Exception {
reader.setCurrentItemCount(2);
executionContext.putInt(reader.getClass().getSimpleName()+".read.count", 3);
reader.open(executionContext);
// read some records
assertEquals("testLine4", reader.read());
// close input
reader.close();
}
@Test
public void testMaxAndCurrentItemCount() throws Exception {
reader.setMaxItemCount(2);
reader.setCurrentItemCount(2);
reader.open(executionContext);
// read some records
assertNull(reader.read());
// close input
reader.close();
}
@Test
public void testNonExistentResource() throws Exception {