[teiid-commits] teiid SVN: r4275 - trunk/engine/src/test/java/org/teiid/query/processor.

[teiid-commits at lists.jboss.org]

Author: shawkins
Date: 2012-07-30 09:07:44 -0400 (Mon, 30 Jul 2012)
New Revision: 4275

Modified:
   trunk/engine/src/test/java/org/teiid/query/processor/TestBatchIterator.java
Log:
TEIID-2120 updating unit test for batchiterator early termination

Modified: trunk/engine/src/test/java/org/teiid/query/processor/TestBatchIterator.java
===================================================================
--- trunk/engine/src/test/java/org/teiid/query/processor/TestBatchIterator.java	2012-07-30 12:59:25 UTC (rev 4274)
+++ trunk/engine/src/test/java/org/teiid/query/processor/TestBatchIterator.java	2012-07-30 13:07:44 UTC (rev 4275)
@@ -94,16 +94,20 @@
 		BatchIterator bi = new BatchIterator(new FakeRelationalNode(1, new List[] {
 			Arrays.asList(1),
 			Arrays.asList(1),
-		}, 1));
+			Arrays.asList(1),
+			Arrays.asList(1),
+		}, 2));
 		BufferManager bm = BufferManagerFactory.getStandaloneBufferManager();
 		bi.setBuffer(bm.createTupleBuffer(Arrays.asList(new ElementSymbol("x")), "test", TupleSourceType.PROCESSOR), true);  //$NON-NLS-1$
 		bi.mark();
 		assertNotNull(bi.nextTuple());
 		assertNotNull(bi.nextTuple());
-		assertNull(bi.nextTuple());
+		assertNotNull(bi.nextTuple());
 		bi.reset();
 		assertNotNull(bi.nextTuple());
 		assertNotNull(bi.nextTuple());
+		assertNotNull(bi.nextTuple());
+		assertNotNull(bi.nextTuple());
 		assertNull(bi.nextTuple());
 	}