@Test public void checkIterator() { int size = 20; IElement singleDummy; ArrayList<IElement> dummies = new ArrayList<>(); for (int i = 0; i < size; i++) { singleDummy = mock(this.elementClass); /* * Nodes are stored with an index, so set it please! */ if (singleDummy instanceof Node) { when(((Node) singleDummy).getIndex()).thenReturn(i); } if (singleDummy instanceof Edge) { when(((Edge) singleDummy).getHashString()).thenReturn("" + i); } dataStructure.add(singleDummy); dummies.add(singleDummy); } Iterator<IElement> elemIterator = dataStructure.iterator(); while (elemIterator.hasNext()) { singleDummy = elemIterator.next(); assertTrue(dummies.contains(singleDummy)); dummies.remove(singleDummy); } assertTrue(dummies.isEmpty()); }
@Test public void checkNullsInIterator() { IElement singleDummy; dataStructure.reinitializeWithSize(10); Iterator<IElement> elemIterator = dataStructure.iterator(); while (elemIterator.hasNext()) { singleDummy = elemIterator.next(); assertNotNull(singleDummy); } }