Method indexIterator

test/simpledb/BTreeFileReadTest.java:142–208

Unit test for BTreeFile.indexIterator()

()

	 * Unit test for BTreeFile.indexIterator()
	 */
	@Test public void indexIterator() throws Exception {
		BTreeFile twoLeafPageFile = BTreeUtility.createBTreeFile(2, 520,
				null, null, 0);
		Field f =  new IntField(5);

		// greater than
		IndexPredicate ipred = new IndexPredicate(Op.GREATER_THAN, f);
		DbFileIterator it = twoLeafPageFile.indexIterator(tid, ipred);
		it.open();
		int count = 0;
		while(it.hasNext()) {
			Tuple t = it.next();
			assertTrue(t.getField(0).compare(Op.GREATER_THAN, f));
			count++;
		}
		assertEquals(515, count);
		it.close();

		// less than or equal to
		ipred = new IndexPredicate(Op.LESS_THAN_OR_EQ, f);
		it = twoLeafPageFile.indexIterator(tid, ipred);
		it.open();
		count = 0;
		while(it.hasNext()) {
			Tuple t = it.next();
			assertTrue(t.getField(0).compare(Op.LESS_THAN_OR_EQ, f));
			count++;
		}
		assertEquals(5, count);
		it.close();

		// equal to
		ipred = new IndexPredicate(Op.EQUALS, f);
		it = twoLeafPageFile.indexIterator(tid, ipred);
		it.open();
		count = 0;
		while(it.hasNext()) {
			Tuple t = it.next();
			assertTrue(t.getField(0).compare(Op.EQUALS, f));
			count++;
		}
		assertEquals(1, count);
		it.close();

		// now insert a record and ensure EQUALS returns both records
		twoLeafPageFile.insertTuple(tid, BTreeUtility.getBTreeTuple(5, 2));
		ipred = new IndexPredicate(Op.EQUALS, f);
		it = twoLeafPageFile.indexIterator(tid, ipred);
		it.open();
		count = 0;
		while(it.hasNext()) {
			Tuple t = it.next();
			assertTrue(t.getField(0).compare(Op.EQUALS, f));
			count++;
		}
		assertEquals(2, count);
		it.close();