Updated the testcases

[dmaap/messagerouter/mirroragent.git] / src / test / java / com / att / nsa / dmaapMMAgent / TestMirrorMakerAgent.java

diff --git a/src/test/java/com/att/nsa/dmaapMMAgent/TestMirrorMakerAgent.java b/src/test/java/com/att/nsa/dmaapMMAgent/TestMirrorMakerAgent.java
index e7d9a5c..ddefa5c 100644 (file)
--- a/src/test/java/com/att/nsa/dmaapMMAgent/TestMirrorMakerAgent.java
+++ b/src/test/java/com/att/nsa/dmaapMMAgent/TestMirrorMakerAgent.java
@@ -23,43 +23,110 @@
 package com.att.nsa.dmaapMMAgent;
 
 import static org.junit.Assert.*;
+import static org.mockito.Mockito.spy;
 
-import java.io.DataOutputStream;
-import java.io.FileInputStream;
-import java.io.FileNotFoundException;
-import java.net.HttpURLConnection;
 import java.util.ArrayList;
 
+import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
-import org.powermock.api.mockito.PowerMockito;
-import org.powermock.core.classloader.annotations.Mock;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
 import org.powermock.modules.junit4.PowerMockRunner;
 
 import com.att.nsa.dmaapMMAgent.dao.ListMirrorMaker;
 import com.att.nsa.dmaapMMAgent.dao.MirrorMaker;
+import com.google.gson.Gson;
+import com.google.gson.internal.LinkedTreeMap;
 
 @RunWith(PowerMockRunner.class)
 public class TestMirrorMakerAgent {
+
        MirrorMakerAgent mirrorMakerAgent = new MirrorMakerAgent();
        ListMirrorMaker listMirrorMaker = new ListMirrorMaker();
        MirrorMaker mirrorMaker = new MirrorMaker();
        MirrorMaker mirrorMaker2 = new MirrorMaker();
        ArrayList<MirrorMaker> listsMirrorMaker = new ArrayList<MirrorMaker>();
+       Gson g = new Gson();
+       @InjectMocks
+       private MirrorMakerAgent agent;
+       @Mock
+       private TopicUtil topicUtil;
+
+       @Before
+       public void setUp() {
+               MockitoAnnotations.initMocks(this);
+       }
 
        @Test
        public void testcheckStartup() {
                String currentDirectory = System.getProperty("user.dir");
                String MMAGENTHOME = currentDirectory + "/src/test/resources/";
-               String parameters[] = {"-encrypt", "test"};
+               String parameters[] = { "-encrypt", "test" };
                String args[] = null;
-               
+
                System.setProperty("MMAGENTHOME", MMAGENTHOME);
-               
                mirrorMakerAgent.main(args);
+
+       }
+
+       @Test
+       public void testReadAgentTopics() {
+               
+               agent.exitLoop = true;
+               agent.readAgentTopic();
+       }
+
+       @Test
+       public void testReadCreateMirrorMaker() {
+               
+               String topicMessage = "{ messageID:\"test\", createMirrorMaker: {   name:\"test\",   consumer:\"test\",  producer:\"test\"}}";
+               LinkedTreeMap<?, ?> object = g.fromJson(topicMessage, LinkedTreeMap.class);
+               agent.readAgent(object, topicMessage);
+
+       }
+
+       @Test
+       public void testReadUpdateMirrorMaker() {
+               
+               String topicMessage = "{ messageID:\"test\", updateMirrorMaker: {   name:\"test\",   consumer:\"test\",  producer:\"test\"}}";
+               LinkedTreeMap<?, ?> object = g.fromJson(topicMessage, LinkedTreeMap.class);
+               testReadCreateMirrorMaker();
+               agent.readAgent(object, topicMessage);
+
+       }
+
+       @Test
+       public void testReadDeleteMirrorMaker() {
+               
+               String topicMessage = "{ messageID:\"test\", deleteMirrorMaker: {   name:\"test\",   consumer:\"test\",  producer:\"test\",  whitelist:\"test\",status:\"test\" }}";
+               LinkedTreeMap<?, ?> object = g.fromJson(topicMessage, LinkedTreeMap.class);
+               testReadCreateMirrorMaker();
+               agent.readAgent(object, topicMessage);
+
+       }
+
+       @Test
+       public void testReadListMirrorMaker() {
                
+               String topicMessage = "{ messageID:\"test\", listAllMirrorMaker: {   name:\"test\",   consumer:\"test\",  producer:\"test\",  whitelist:\"test\",status:\"test\" }}";
+               LinkedTreeMap<?, ?> object = g.fromJson(topicMessage, LinkedTreeMap.class);
+               testReadCreateMirrorMaker();
+               agent.readAgent(object, topicMessage);
+
        }
-       
+
+       @Test
+       public void testReadWhitelistMirrorMaker() {
+               
+               String topicMessage = "{ messageID:\"test\", updateWhiteList: {   name:\"test\",   consumer:\"test\",  producer:\"test\",  whitelist:\"test\",status:\"test\" }}";
+               LinkedTreeMap<?, ?> object = g.fromJson(topicMessage, LinkedTreeMap.class);
+               testReadCreateMirrorMaker();
+               agent.readAgent(object, topicMessage);
+
+       }
+
        @Test
        public void testCreateMirrorMaker() {
                mirrorMaker.setConsumer("consumer");
@@ -67,20 +134,20 @@ public class TestMirrorMakerAgent {
                mirrorMaker.setProducer("producer");
                mirrorMaker.setStatus("200");
                mirrorMaker.setWhitelist("whitelist");
-               
+
                mirrorMaker2.setConsumer("consumer");
                mirrorMaker2.setName("MirrorMaker2");
                mirrorMaker2.setProducer("producer");
                mirrorMaker2.setStatus("200");
                mirrorMaker2.setWhitelist("whitelist");
-               
+
                listsMirrorMaker.add(mirrorMaker2);
                listMirrorMaker.setListMirrorMaker(listsMirrorMaker);
-               
+
                mirrorMakerAgent.mirrorMakers = listMirrorMaker;
-               
+
                mirrorMakerAgent.createMirrorMaker(mirrorMaker);
-               
+
                assertEquals(2, mirrorMakerAgent.mirrorMakers.getListMirrorMaker().size());
        }