Update iterator capabilities

1 files changed, 41 insertions, 0 deletions

diff --git a/src/test/java/bjc/data/MarkListIteratorTest.java b/src/test/java/bjc/data/MarkListIteratorTest.java
new file mode 100644
index 0000000..6eb4994
--- /dev/null
+++ b/src/test/java/bjc/data/MarkListIteratorTest.java
@@ -0,0 +1,41 @@
+package bjc.data;
+
+import static org.junit.Assert.*;
+
+import java.util.Arrays;
+import java.util.List;
+
+import org.junit.Test;
+
+import bjc.test.TestUtils;
+
+/**
+ * Tests for {@link MarkListIterator}
+ * @author bjcul
+ *
+ */
+@SuppressWarnings("javadoc")
+public class MarkListIteratorTest {
+
+	@Test
+	public void test() {
+		List<String> list1 = Arrays.asList("a", "b", "c");
+		
+		MarkListIterator<String> itr1 = new MarkListIterator<>(list1);
+		itr1.mark();
+		
+		assertFalse(itr1.hasPrevious());
+		assertTrue(itr1.hasNext());
+		TestUtils.assertIteratorEquals(itr1, "a", "b", "c");
+		
+		assertTrue(itr1.hasPrevious());
+		assertFalse(itr1.hasNext());
+		
+		itr1.rollback(false);
+		
+		assertFalse(itr1.hasPrevious());
+		assertTrue(itr1.hasNext());
+		TestUtils.assertIteratorEquals(itr1, "a", "b", "c");
+	}
+
+}