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()); }