modified return type of TreasureChestItemIterator's constructor to be Iterator

mitchellirvin · mitchellirvin · commit e6f84f2f41fd · 2018-08-28T07:46:02.000-04:00
diff --git a/iterator/src/main/java/com/iluwatar/iterator/list/TreasureChest.java b/iterator/src/main/java/com/iluwatar/iterator/list/TreasureChest.java
@@ -18,6 +18,7 @@
  */
 package com.iluwatar.iterator.list;
 
+import com.iluwatar.iterator.Iterator;
 import java.util.ArrayList;
 import java.util.List;
 
@@ -47,7 +48,7 @@ public TreasureChest() {
     items.add(new Item(ItemType.WEAPON, "Dagger of poison"));
   }
 
-  public TreasureChestItemIterator iterator(ItemType itemType) {
+  public Iterator iterator(ItemType itemType) {
     return new TreasureChestItemIterator(this, itemType);
   }
 
diff --git a/iterator/src/test/java/com/iluwatar/iterator/list/TreasureChestTest.java b/iterator/src/test/java/com/iluwatar/iterator/list/TreasureChestTest.java
@@ -22,6 +22,7 @@
 import static org.junit.jupiter.api.Assertions.assertNotNull;
 import static org.junit.jupiter.api.Assertions.fail;
 
+import com.iluwatar.iterator.Iterator;
 import java.util.ArrayList;
 import java.util.List;
 import org.junit.jupiter.params.ParameterizedTest;
@@ -62,11 +63,11 @@ public static List<Object[]> dataProvider() {
   @MethodSource("dataProvider")
   public void testIterator(Item expectedItem) {
     final TreasureChest chest = new TreasureChest();
-    final TreasureChestItemIterator iterator = chest.iterator(expectedItem.getType());
+    final Iterator iterator = chest.iterator(expectedItem.getType());
     assertNotNull(iterator);
 
     while (iterator.hasNext()) {
-      final Item item = iterator.next();
+      final Item item = (Item) iterator.next();
       assertNotNull(item);
       assertEquals(expectedItem.getType(), item.getType());
 

Original file line number	Diff line number	Diff line change
`@@ -18,6 +18,7 @@`
`18`	`18`	`*/`
`19`	`19`	`package com.iluwatar.iterator.list;`
`20`	`20`
	`21`	`+import com.iluwatar.iterator.Iterator;`
`21`	`22`	`import java.util.ArrayList;`
`22`	`23`	`import java.util.List;`
`23`	`24`
`@@ -47,7 +48,7 @@ public TreasureChest() {`
`47`	`48`	`items.add(new Item(ItemType.WEAPON, "Dagger of poison"));`
`48`	`49`	`}`
`49`	`50`
`50`		`- public TreasureChestItemIterator iterator(ItemType itemType) {`
	`51`	`+ public Iterator iterator(ItemType itemType) {`
`51`	`52`	`return new TreasureChestItemIterator(this, itemType);`
`52`	`53`	`}`
`53`	`54`