Test done

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