1 files changed, 49 insertions, 0 deletions
diff --git a/src/test/java/io/github/bculkin2442/data/CircularIteratorTest.java b/src/test/java/io/github/bculkin2442/data/CircularIteratorTest.java
new file mode 100644
index 0000000..80fbb54
--- /dev/null
+++ b/src/test/java/io/github/bculkin2442/data/CircularIteratorTest.java
@@ -0,0 +1,49 @@
+package io.github.bculkin2442.data;
+
+import java.util.Arrays;
+import java.util.List;
+
+import org.junit.Test;
+
+import bjc.data.CircularIterator;
+
+import static io.github.bculkin2442.TestUtils.*;
+/**
+ * Test for circular iterators.,
+ * 
+ * @author bjculkin
+ *
+ */
+public class CircularIteratorTest {
+	/**
+	 * Test regular repetition of the entire iterator.
+	 */
+	@Test
+	public void testRegular() {
+		List<String> lst = Arrays.asList("a", "b", "c");
+
+		CircularIterator<String> itr = new CircularIterator<>(lst);
+
+		// Check we get initial values correctly, and have more remaining
+		assertIteratorEquals(true, itr, "a", "b", "c");
+
+		// Check we repeat correctly, and can still repeat
+		assertIteratorEquals(true, itr, "a", "b", "c");
+	}
+
+	/**
+	 * Test that the last element repeats correctly.
+	 */
+	@Test
+	public void testRepLast() {
+		List<String> lst = Arrays.asList("a", "b", "c");
+
+		CircularIterator<String> itr = new CircularIterator<>(lst, false);
+
+		// Check we get initial values correctly, and have more remaining
+		assertIteratorEquals(true, itr, "a", "b", "c");
+
+		// Check we repeat correctly, and can still repeat
+		assertIteratorEquals(true, itr, "c", "c", "c");
+	}
+}