Fixed all the code changes after review

queue-load-leveling/src/test/java/org/queue/load/leveling/AppTest.java

@@ -27,7 +27,7 @@ import java.io.IOException;
 import org.junit.Test;
 
 /**
- * Tests that Caching example runs without errors.
+ * Application Test
  */
 public class AppTest {
   @Test

queue-load-leveling/src/test/java/org/queue/load/leveling/MessageQueueTest.java

@@ -22,9 +22,10 @@
  */
 package org.queue.load.leveling;
 
-import org.junit.Test;
 import static org.junit.Assert.assertEquals;
 
+import org.junit.Test;
+
 /**
  * 
  * Test case for submitting and retrieving messages from Blocking Queue.
@@ -36,13 +37,12 @@ public class MessageQueueTest {
   public void messageQueueTest() {
     
     MessageQueue msgQueue = new MessageQueue();
-    Message msg = new Message("MessageQueue Test");
     
     // submit message
-    msgQueue.submitMsg(msg);
+    msgQueue.submitMsg(new Message("MessageQueue Test"));
     
     // retrieve message
-    assertEquals(msg.getMsg(), msgQueue.retrieveMsg().getMsg());
+    assertEquals(msgQueue.retrieveMsg().getMsg(), "MessageQueue Test");
   }
 
 }

queue-load-leveling/src/test/java/org/queue/load/leveling/MessageTest.java

@@ -39,10 +39,5 @@ public class MessageTest {
     String testMsg = "Message Test";
     Message msg = new Message(testMsg);
     assertEquals(msg.getMsg(), testMsg);
-
-    // Default constructor and setter method test.
-    Message simpleMsg = new Message();
-    simpleMsg.setMsg(testMsg);
-    assertEquals(simpleMsg.getMsg(), testMsg);
   }
 }