@Test public void testAddAllExceptionIfLimitApplied() { ResultsBag bag = getBagObject(Object.class); bag.applyLimit(6); try { bag.addAll(new HashSet()); fail("addAll invocation to bag with limit applied should have failed"); } catch (UnsupportedOperationException soe) { // Expected exception } }
bag.add(wrap(null, bag.getCollectionType().getElementType())); bag.applyLimit(8); Iterator itr = bag.iterator(); if (!(bag instanceof StructBag)) {
bag.add(wrap(new Integer(3), bag.getCollectionType().getElementType())); bag.add(wrap(new Integer(4), bag.getCollectionType().getElementType())); bag.applyLimit(3); assertEquals(3, bag.size());
bag.add(wrap(null, bag.getCollectionType().getElementType())); bag.add(wrap(null, bag.getCollectionType().getElementType())); bag.applyLimit(4); List asList = bag.asList();
bag.add(wrap(null, bag.getCollectionType().getElementType())); bag.add(wrap(null, bag.getCollectionType().getElementType())); bag.applyLimit(4); List asList = bag.asList(); asList.add(wrap(new Integer(13), bag.getCollectionType().getElementType()));
bag.add(wrap(null, bag.getCollectionType().getElementType())); bag.add(wrap(null, bag.getCollectionType().getElementType())); bag.applyLimit(4); List asList = bag.asList();
@Test public void testAsListAndAsSetMethod() { ResultsBag bag = getBagObject(String.class); bag.add(wrap("one", bag.getCollectionType().getElementType())); bag.add(wrap("two", bag.getCollectionType().getElementType())); bag.add(wrap("three", bag.getCollectionType().getElementType())); bag.add(wrap("four", bag.getCollectionType().getElementType())); assertEquals(4, bag.size()); bag.applyLimit(2); List list = bag.asList(); assertEquals(2, list.size()); Set set = bag.asSet(); assertEquals(2, set.size()); }
bag.add(wrap(new Integer(5), bag.getCollectionType().getElementType())); bag.add(wrap(new Integer(5), bag.getCollectionType().getElementType())); bag.applyLimit(8); Iterator itr = bag.iterator(); if (!(bag instanceof StructBag)) {
bag.add(wrap(new Integer(15), bag.getCollectionType().getElementType())); bag.add(wrap(new Integer(16), bag.getCollectionType().getElementType())); bag.applyLimit(6); List temp = bag.asList(); assertEquals(6, bag.size());
@Test public void testAddExceptionIfLimitApplied() { ResultsBag bag = getBagObject(String.class); bag.add(wrap("one", bag.getCollectionType().getElementType())); bag.add(wrap("two", bag.getCollectionType().getElementType())); bag.add(wrap("two", bag.getCollectionType().getElementType())); bag.applyLimit(6); try { bag.add(wrap("four", bag.getCollectionType().getElementType())); fail("Addition to bag with limit applied should have failed"); } catch (UnsupportedOperationException soe) { // Expected exception } }
@Test public void testSizeWithLimitApplied() { ResultsBag bag = getBagObject(String.class); bag.add(wrap("two", bag.getCollectionType().getElementType())); bag.add(wrap("three", bag.getCollectionType().getElementType())); bag.add(wrap("three", bag.getCollectionType().getElementType())); bag.add(wrap("three", bag.getCollectionType().getElementType())); bag.add(wrap("three", bag.getCollectionType().getElementType())); assertEquals(5, bag.size()); // Limit less than actual size bag.applyLimit(3); assertEquals(3, bag.size()); bag = getBagObject(String.class); bag.add(wrap("two", bag.getCollectionType().getElementType())); bag.add(wrap("three", bag.getCollectionType().getElementType())); bag.applyLimit(7); assertEquals(2, bag.size()); }
bag.add(wrap("four", bag.getCollectionType().getElementType())); bag.add(wrap("four", bag.getCollectionType().getElementType())); bag.applyLimit(15); assertEquals(12, bag.size()); bag.remove(wrap(null, bag.getCollectionType().getElementType())); bag.add(wrap(new Integer(4), bag.getCollectionType().getElementType())); bag.add(wrap(new Integer(4), bag.getCollectionType().getElementType())); bag.applyLimit(8); List temp = bag.asList(); int currSize = 8; bag.add(wrap(null, bag.getCollectionType().getElementType())); bag.applyLimit(4);
@Test public void testIteratorType() { ResultsBag bag = getBagObject(String.class); bag.add(wrap("one", bag.getCollectionType().getElementType())); bag.add(wrap("two", bag.getCollectionType().getElementType())); bag.add(wrap("two", bag.getCollectionType().getElementType())); bag.add(wrap("three", bag.getCollectionType().getElementType())); bag.add(wrap("three", bag.getCollectionType().getElementType())); assertTrue(bag.iterator() instanceof Bag.BagIterator); bag.applyLimit(6); assertTrue(bag.iterator() instanceof Bag.BagIterator); bag = getBagObject(String.class); bag.add(wrap("one", bag.getCollectionType().getElementType())); bag.add(wrap("two", bag.getCollectionType().getElementType())); bag.add(wrap("two", bag.getCollectionType().getElementType())); bag.add(wrap("three", bag.getCollectionType().getElementType())); bag.add(wrap("three", bag.getCollectionType().getElementType())); bag.add(wrap("three", bag.getCollectionType().getElementType())); bag.add(wrap("four", bag.getCollectionType().getElementType())); bag.add(wrap("four", bag.getCollectionType().getElementType())); bag.add(wrap("four", bag.getCollectionType().getElementType())); bag.add(wrap("four", bag.getCollectionType().getElementType())); bag.applyLimit(6); if (!(bag instanceof StructBag)) { assertTrue(bag.iterator() instanceof Bag.LimitBagIterator); } }
toBag.add(wrap("four", toBag.getCollectionType().getElementType())); toBag.add(wrap("four", toBag.getCollectionType().getElementType())); toBag.applyLimit(9); assertEquals(9, toBag.size()); ByteArrayOutputStream baos = new ByteArrayOutputStream(10240);
@Test public void testOccurrence() { ResultsBag bag = getBagObject(String.class); bag.add(wrap(null, bag.getCollectionType().getElementType())); bag.add(wrap(null, bag.getCollectionType().getElementType())); bag.add(wrap("one", bag.getCollectionType().getElementType())); bag.add(wrap("two", bag.getCollectionType().getElementType())); bag.add(wrap("two", bag.getCollectionType().getElementType())); bag.add(wrap("three", bag.getCollectionType().getElementType())); bag.add(wrap("three", bag.getCollectionType().getElementType())); bag.add(wrap("three", bag.getCollectionType().getElementType())); bag.add(wrap("four", bag.getCollectionType().getElementType())); bag.add(wrap("four", bag.getCollectionType().getElementType())); bag.add(wrap("four", bag.getCollectionType().getElementType())); bag.add(wrap("four", bag.getCollectionType().getElementType())); assertEquals(12, bag.size()); bag.applyLimit(6); int total = bag.occurrences(wrap(null, bag.getCollectionType().getElementType())); total += bag.occurrences(wrap("one", bag.getCollectionType().getElementType())); total += bag.occurrences(wrap("two", bag.getCollectionType().getElementType())); total += bag.occurrences(wrap("three", bag.getCollectionType().getElementType())); total += bag.occurrences(wrap("four", bag.getCollectionType().getElementType())); assertEquals(6, total); }