1 files changed, 87 insertions, 0 deletions
diff --git a/EssentialsUpdate/test/net/ess3/update/VersionTest.java b/EssentialsUpdate/test/net/ess3/update/VersionTest.java
new file mode 100644
index 000000000..1187eecd7
--- /dev/null
+++ b/EssentialsUpdate/test/net/ess3/update/VersionTest.java
@@ -0,0 +1,87 @@
+package net.ess3.update;
+
+import net.ess3.update.Version.Type;
+import java.util.TreeSet;
+import junit.framework.TestCase;
+import net.ess3.update.Version;
+import org.junit.Test;
+
+
+public class VersionTest extends TestCase
+{
+	@Test
+	public void testStable()
+	{
+		final Version instance = new Version("1.2.3");
+		assertEquals("Testing Major", 1, instance.getMajor());
+		assertEquals("Testing Minor", 2, instance.getMinor());
+		assertEquals("Testing Build", 3, instance.getBuild());
+		assertEquals("Testing Type", Type.STABLE, instance.getType());
+	}
+
+	@Test
+	public void testDev()
+	{
+		final Version instance = new Version("Dev2.3.4");
+		assertEquals("Testing Major", 2, instance.getMajor());
+		assertEquals("Testing Minor", 3, instance.getMinor());
+		assertEquals("Testing Build", 4, instance.getBuild());
+		assertEquals("Testing Type", Type.DEVELOPER, instance.getType());
+	}
+
+	@Test
+	public void testTeamCity()
+	{
+		final Version instance = new Version("Unknown");
+		assertEquals("Testing Type", Type.DEVELOPER, instance.getType());
+	}
+
+	@Test
+	public void testPre()
+	{
+		final Version instance = new Version("Pre5.7.400.2");
+		assertEquals("Testing Major", 5, instance.getMajor());
+		assertEquals("Testing Minor", 7, instance.getMinor());
+		assertEquals("Testing Build", 400, instance.getBuild());
+		assertEquals("Testing Type", Type.PREVIEW, instance.getType());
+	}
+
+	@Test
+	public void testCompareTo()
+	{
+		final Version verA = new Version("1.1.1");
+		final Version verB = new Version("Dev1.1.2");
+		final Version verC = new Version("1.1.2");
+		final Version verD = new Version("1.2.0");
+		final Version verE = new Version("2.0.0");
+		final Version verF = new Version("Pre1.1.1.1");
+		final Version verG = new Version("Dev1.2.2");
+		assertTrue("Testing dev", verA.compareTo(verB) < 0);
+		assertTrue("Testing dev", verB.compareTo(verA) > 0);
+		assertTrue("Testing build", verA.compareTo(verC) < 0);
+		assertTrue("Testing build", verC.compareTo(verA) > 0);
+		assertTrue("Testing minor", verA.compareTo(verD) < 0);
+		assertTrue("Testing minor", verD.compareTo(verA) > 0);
+		assertTrue("Testing major", verA.compareTo(verE) < 0);
+		assertTrue("Testing major", verE.compareTo(verA) > 0);
+		assertTrue("Testing pre", verF.compareTo(verA) < 0);
+		assertTrue("Testing pre", verA.compareTo(verF) > 0);
+		assertTrue("Testing dev vs dev", verB.compareTo(verG) < 0);
+		assertTrue("Testing dev vs dev", verG.compareTo(verB) > 0);
+		final TreeSet<Version> set = new TreeSet<Version>();
+		set.add(verA);
+		set.add(verB);
+		set.add(verC);
+		set.add(verD);
+		set.add(verE);
+		set.add(verF);
+		set.add(verG);
+		assertEquals("Testing sorting", verF, set.pollFirst());
+		assertEquals("Testing sorting", verA, set.pollFirst());
+		assertEquals("Testing sorting", verC, set.pollFirst());
+		assertEquals("Testing sorting", verD, set.pollFirst());
+		assertEquals("Testing sorting", verE, set.pollFirst());
+		assertEquals("Testing sorting", verB, set.pollFirst());
+		assertEquals("Testing sorting", verG, set.pollFirst());
+	}
+}