Test done

extension-objects/src/test/java/AppTest.java

@@ -0,0 +1,16 @@
+import org.junit.Test;
+
+import static org.junit.Assert.*;
+
+/**
+ * Created by Srdjan on 03-May-17.
+ */
+public class AppTest {
+  @Test
+  public void main() throws Exception {
+
+    String[] args = {};
+    App.main(args);
+  }
+
+}

extension-objects/src/test/java/concreteextensions/CommanderTest.java

@@ -0,0 +1,19 @@
+package concreteextensions;
+
+import org.junit.Test;
+import units.CommanderUnit;
+
+import static org.junit.Assert.*;
+
+/**
+ * Created by Srdjan on 03-May-17.
+ */
+public class CommanderTest {
+  @Test
+  public void commanderReady() throws Exception {
+    final Commander commander = new Commander(new CommanderUnit("CommanderUnitTest"));
+
+    commander.commanderReady();
+  }
+
+}

extension-objects/src/test/java/concreteextensions/SergeantTest.java

@@ -0,0 +1,19 @@
+package concreteextensions;
+
+import org.junit.Test;
+import units.SergeantUnit;
+
+import static org.junit.Assert.*;
+
+/**
+ * Created by Srdjan on 03-May-17.
+ */
+public class SergeantTest {
+  @Test
+  public void sergeantReady() throws Exception {
+    final Sergeant sergeant = new Sergeant(new SergeantUnit("SergeantUnitTest"));
+
+    sergeant.sergeantReady();
+  }
+
+}

extension-objects/src/test/java/concreteextensions/SoldierTest.java

@@ -0,0 +1,19 @@
+package concreteextensions;
+
+import org.junit.Test;
+import units.SoldierUnit;
+
+import static org.junit.Assert.*;
+
+/**
+ * Created by Srdjan on 03-May-17.
+ */
+public class SoldierTest {
+  @Test
+  public void soldierReady() throws Exception {
+    final Soldier soldier = new Soldier(new SoldierUnit("SoldierUnitTest"));
+
+    soldier.soldierReady();
+  }
+
+}

extension-objects/src/test/java/units/CommanderUnitTest.java

@@ -0,0 +1,22 @@
+package units;
+
+import abstractextensions.CommanderExtension;
+import org.junit.Test;
+
+import static org.junit.Assert.*;
+
+/**
+ * Created by Srdjan on 03-May-17.
+ */
+public class CommanderUnitTest {
+  @Test
+  public void getUnitExtension() throws Exception {
+
+    final Unit unit = new CommanderUnit("CommanderUnitName");
+
+    assertNull(unit.getUnitExtension("SoldierExtension"));
+    assertNull(unit.getUnitExtension("SergeantExtension"));
+    assertNotNull((CommanderExtension)unit.getUnitExtension("CommanderExtension"));
+  }
+
+}

extension-objects/src/test/java/units/SergeantUnitTest.java

@@ -0,0 +1,22 @@
+package units;
+
+import abstractextensions.SergeantExtension;
+import org.junit.Test;
+
+import static org.junit.Assert.*;
+
+/**
+ * Created by Srdjan on 03-May-17.
+ */
+public class SergeantUnitTest {
+  @Test
+  public void getUnitExtension() throws Exception {
+
+    final Unit unit = new SergeantUnit("SergeantUnitName");
+
+    assertNull(unit.getUnitExtension("SoldierExtension"));
+    assertNotNull((SergeantExtension)unit.getUnitExtension("SergeantExtension"));
+    assertNull(unit.getUnitExtension("CommanderExtension"));
+  }
+
+}

extension-objects/src/test/java/units/SoldierUnitTest.java

@@ -0,0 +1,24 @@
+package units;
+
+import abstractextensions.SoldierExtension;
+import org.junit.Test;
+
+import static org.junit.Assert.*;
+
+/**
+ * Created by Srdjan on 03-May-17.
+ */
+public class SoldierUnitTest {
+  @Test
+  public void getUnitExtension() throws Exception {
+
+    final Unit unit = new SoldierUnit("SoldierUnitName");
+
+    assertNotNull((SoldierExtension)unit.getUnitExtension("SoldierExtension"));
+    assertNull(unit.getUnitExtension("SergeantExtension"));
+    assertNull(unit.getUnitExtension("CommanderExtension"));
+
+
+  }
+
+}

extension-objects/src/test/java/units/UnitTest.java

@@ -0,0 +1,33 @@
+package units;
+
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+
+import static org.junit.Assert.*;
+
+/**
+ * Created by Srdjan on 03-May-17.
+ */
+public class UnitTest {
+
+  private Unit unit;
+
+  @Test
+  public void testConstGetSet() throws Exception {
+    final String name = "testName";
+    final Unit unit = new Unit(name);
+    assertEquals(name,unit.getName());
+
+    final String newName = "newName";
+    unit.setName(newName);
+    assertEquals(newName,unit.getName());
+
+
+    assertNull(unit.getUnitExtension(""));
+    assertNull(unit.getUnitExtension("SoldierExtension"));
+    assertNull(unit.getUnitExtension("SergeantExtension"));
+    assertNull(unit.getUnitExtension("CommanderExtension"));
+  }
+
+}