@Test public void testReset() throws Exception { BatchIterator bi = new BatchIterator(new FakeRelationalNode(1, new List[] { Arrays.asList(1), Arrays.asList(1), Arrays.asList(1) }, 1)); BufferManager bm = BufferManagerFactory.getStandaloneBufferManager(); bi.setBuffer(bm.createTupleBuffer(Arrays.asList(new ElementSymbol("x", null, DataTypeManager.DefaultDataClasses.INTEGER)), "test", TupleSourceType.PROCESSOR), true); //$NON-NLS-1$ bi.mark(); bi.nextTuple(); bi.nextTuple(); bi.reset(); bi.nextTuple(); }
@Test public void testReset1() throws Exception { BatchIterator bi = new BatchIterator(new FakeRelationalNode(1, new List[] { Arrays.asList(1), Arrays.asList(2), Arrays.asList(3) }, 2)); BufferManager bm = BufferManagerFactory.getStandaloneBufferManager(); TupleBuffer tb = bm.createTupleBuffer(Arrays.asList(new ElementSymbol("x", null, DataTypeManager.DefaultDataClasses.INTEGER)), "test", TupleSourceType.PROCESSOR); bi.setBuffer(tb, true); //$NON-NLS-1$ bi.nextTuple(); bi.mark(); bi.nextTuple(); bi.reset(); assertEquals(2, bi.getCurrentIndex()); assertEquals(2, bi.nextTuple().get(0)); }
@Test public void testReset2() throws Exception { BatchIterator bi = new BatchIterator(new FakeRelationalNode(1, new List[] { Arrays.asList(1), Arrays.asList(2), }, 2)); BufferManager bm = BufferManagerFactory.getStandaloneBufferManager(); TupleBuffer tb = bm.createTupleBuffer(Arrays.asList(new ElementSymbol("x", null, DataTypeManager.DefaultDataClasses.INTEGER)), "test", TupleSourceType.PROCESSOR); bi.setBuffer(tb, true); //$NON-NLS-1$ bi.hasNext(); bi.mark(); bi.nextTuple(); bi.nextTuple(); assertNull(bi.nextTuple()); bi.reset(); bi.hasNext(); assertEquals(1, bi.getCurrentIndex()); assertEquals(1, bi.nextTuple().get(0)); }
@Test public void testBatchReadDuringMark() throws Exception { BatchIterator bi = new BatchIterator(new FakeRelationalNode(1, new List[] { Arrays.asList(1), Arrays.asList(1), Arrays.asList(1), Arrays.asList(1), }, 2)); BufferManager bm = BufferManagerFactory.getStandaloneBufferManager(); bi.setBuffer(bm.createTupleBuffer(Arrays.asList(new ElementSymbol("x", null, DataTypeManager.DefaultDataClasses.INTEGER)), "test", TupleSourceType.PROCESSOR), true); //$NON-NLS-1$ bi.mark(); assertNotNull(bi.nextTuple()); assertNotNull(bi.nextTuple()); assertNotNull(bi.nextTuple()); bi.reset(); assertNotNull(bi.nextTuple()); assertNotNull(bi.nextTuple()); assertNotNull(bi.nextTuple()); assertNotNull(bi.nextTuple()); assertNull(bi.nextTuple()); }