Test done

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"));
+
+
+  }
+
+}