Test done

extension-objects/pom.xml

@@ -10,6 +10,12 @@
     <modelVersion>4.0.0</modelVersion>
 
     <artifactId>extension-objects</artifactId>
+    <dependencies>
+        <dependency>
+            <groupId>junit</groupId>
+            <artifactId>junit</artifactId>
+        </dependency>
+    </dependencies>
 
 
-</project>
+</project>

extension-objects/src/main/java/App.java

@@ -38,6 +38,7 @@ public class App {
     SergeantExtension sergeantExtension = (SergeantExtension) unit.getUnitExtension("SergeantExtension");
     CommanderExtension commanderExtension = (CommanderExtension) unit.getUnitExtension("CommanderExtension");
 
+    //if unit have extension call the method
     if (soldierExtension != null) {
       soldierExtension.soldierReady();
     } else {
@@ -51,7 +52,7 @@ public class App {
     }
 
     if (commanderExtension != null) {
-      // commanderExtension.sergeantReady();
+      commanderExtension.commanderReady();
     } else {
       System.out.println(unit.getName() + " without CommanderExtension");
     }

extension-objects/src/main/java/abstractextensions/CommanderExtension.java

@@ -4,4 +4,6 @@ package abstractextensions;
  * Created by Srdjan on 27-Apr-17.
  */
 public interface CommanderExtension extends UnitExtension {
+
+  void commanderReady();
 }

extension-objects/src/main/java/concreteextensions/Commander.java

@@ -13,4 +13,9 @@ public class Commander implements CommanderExtension {
   public Commander(CommanderUnit commanderUnit) {
     this.unit = commanderUnit;
   }
+
+  @Override
+  public void commanderReady() {
+    System.out.println("[Commander] " + unit.getName() + " is ready!");
+  }
 }

extension-objects/src/main/java/concreteextensions/Sergeant.java

@@ -16,6 +16,6 @@ public class Sergeant implements SergeantExtension {
 
   @Override
   public void sergeantReady() {
-    System.out.println("[Sergeant] " + unit.getName() + " do command! ");
+    System.out.println("[Sergeant] " + unit.getName() + " is ready! ");
   }
 }

extension-objects/src/main/java/concreteextensions/Soldier.java

@@ -16,6 +16,6 @@ public class Soldier implements SoldierExtension {
 
   @Override
   public void soldierReady() {
-    System.out.println("[Solider] " + unit.getName() + "  do command");
+    System.out.println("[Solider] " + unit.getName() + "  is ready!");
   }
 }

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