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

RESOLVED - BATCH-649: Fetch Limit in Hibernate item reader

Browse Source
This commit is contained in:
robokaso
2008-06-05 09:36:32 +00:00
parent 23b5f47d82
commit 05b3e0ad36
2 changed files with 44 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
spring-batch-infrastructure/src/test/java/org/springframework/batch/item/database/HibernateCursorItemReaderStatefulIntegrationTests.java
View File

@@ -24,7 +24,7 @@ public class HibernateCursorItemReaderStatefulIntegrationTests extends Hibernate
SessionFactory sessionFactory = (SessionFactory) sessionFactoryControl.getMock();
MockControl sessionControl = MockControl.createControl(Session.class);
Session session = (Session) sessionControl.getMock();
MockControl resultsControl = MockControl.createControl(Query.class);
MockControl resultsControl = MockControl.createNiceControl(Query.class);
Query scrollableResults = (Query) resultsControl.getMock();
HibernateCursorItemReader itemReader = new HibernateCursorItemReader();
itemReader.setSessionFactory(sessionFactory);
@@ -35,10 +35,13 @@ public class HibernateCursorItemReaderStatefulIntegrationTests extends Hibernate
sessionFactoryControl.setReturnValue(session);
session.createQuery("testQuery");
sessionControl.setReturnValue(scrollableResults);
scrollableResults.setFetchSize(0);
resultsControl.setReturnValue(scrollableResults);
session.close();
sessionControl.setReturnValue(null);
sessionFactoryControl.replay();
sessionControl.replay();
resultsControl.replay();
itemReader.open(new ExecutionContext());
itemReader.close(new ExecutionContext());
sessionFactoryControl.verify();