1 files changed, 46 insertions, 0 deletions
diff --git a/src/test/java/io/github/bculkin2442/esodata/ThresholdSetTest.java b/src/test/java/io/github/bculkin2442/esodata/ThresholdSetTest.java
new file mode 100644
index 0000000..c8f2bd3
--- /dev/null
+++ b/src/test/java/io/github/bculkin2442/esodata/ThresholdSetTest.java
@@ -0,0 +1,46 @@
+package io.github.bculkin2442.esodata;
+
+import org.junit.Test;
+
+import bjc.esodata.ThresholdSet;
+import io.github.bculkin2442.TestUtils;
+
+import java.util.Iterator;
+
+import static org.junit.Assert.*;
+import static io.github.bculkin2442.TestUtils.*;
+
+/**
+ * Tests for ThresholdSet
+ *
+ * @author Ben Culkin.
+ */
+public class ThresholdSetTest {
+	@Test
+	public void testAdd() {
+		ThresholdSet<String> thst = new ThresholdSet<>();
+
+		thst.addAll("a", "b");
+
+		assertIteratorEquals(false, thst.setView().iterator(), "a", "b");
+	}
+
+	@Test
+	public void testAddMulti() {
+		ThresholdSet<String> thst = new ThresholdSet<>();
+
+		thst.addAll("a", "b", "a");
+
+		assertIteratorEquals(false, thst.setView().iterator(), "b");
+	}
+
+	@Test
+	public void testRemoveMulti() {
+		ThresholdSet<String> thst = new ThresholdSet<>();
+
+		thst.addAll("a", "a", "b");
+		thst.removeAll("a");
+
+		assertIteratorEquals(false, thst.setView().iterator(), "a", "b");
+	}
+}