Updated the testcases
[dmaap/messagerouter/mirroragent.git] / src / test / java / com / att / nsa / dmaapMMAgent / TestMirrorMakerAgent.java
index 22ceaf9..ddefa5c 100644 (file)
@@ -24,7 +24,6 @@ package com.att.nsa.dmaapMMAgent;
 
 import static org.junit.Assert.*;
 import static org.mockito.Mockito.spy;
-import static org.mockito.Mockito.when;
 
 import java.util.ArrayList;
 
@@ -43,16 +42,17 @@ 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();
-       @Mock
-       ListMirrorMaker mirrorMakers;
        @InjectMocks
-       MirrorMakerAgent agent;
+       private MirrorMakerAgent agent;
+       @Mock
+       private TopicUtil topicUtil;
 
        @Before
        public void setUp() {
@@ -73,18 +73,14 @@ public class TestMirrorMakerAgent {
 
        @Test
        public void testReadAgentTopics() {
-               MirrorMakerAgent agent = spy(new MirrorMakerAgent());
-               when(agent.subscribeTopic("60000")).thenReturn(null);
+               
                agent.exitLoop = true;
                agent.readAgentTopic();
        }
 
        @Test
        public void testReadCreateMirrorMaker() {
-               agent = spy(new MirrorMakerAgent());
-               when(agent.publishTopic(
-                               "{\"messageID\":\"test\",\"listMirrorMaker\":[{\"name\":\"test\",\"consumer\":\"test\",\"producer\":\"test\",\"status\":\"STOPPED\"}]}"))
-                                               .thenReturn(null);
+               
                String topicMessage = "{ messageID:\"test\", createMirrorMaker: {   name:\"test\",   consumer:\"test\",  producer:\"test\"}}";
                LinkedTreeMap<?, ?> object = g.fromJson(topicMessage, LinkedTreeMap.class);
                agent.readAgent(object, topicMessage);
@@ -93,10 +89,7 @@ public class TestMirrorMakerAgent {
 
        @Test
        public void testReadUpdateMirrorMaker() {
-               agent = spy(new MirrorMakerAgent());
-               when(agent.publishTopic(
-                               "{\"messageID\":\"test\",\"listMirrorMaker\":[{\"name\":\"test\",\"consumer\":\"test\",\"producer\":\"test\",\"status\":\"STOPPED\"}]}"))
-                                               .thenReturn(null);
+               
                String topicMessage = "{ messageID:\"test\", updateMirrorMaker: {   name:\"test\",   consumer:\"test\",  producer:\"test\"}}";
                LinkedTreeMap<?, ?> object = g.fromJson(topicMessage, LinkedTreeMap.class);
                testReadCreateMirrorMaker();
@@ -106,10 +99,7 @@ public class TestMirrorMakerAgent {
 
        @Test
        public void testReadDeleteMirrorMaker() {
-               agent = spy(new MirrorMakerAgent());
-               when(agent.publishTopic(
-                               "{\"messageID\":\"test\",\"listMirrorMaker\":[{\"name\":\"test\",\"consumer\":\"test\",\"producer\":\"test\",\"status\":\"STOPPED\"}]}"))
-                                               .thenReturn(null);
+               
                String topicMessage = "{ messageID:\"test\", deleteMirrorMaker: {   name:\"test\",   consumer:\"test\",  producer:\"test\",  whitelist:\"test\",status:\"test\" }}";
                LinkedTreeMap<?, ?> object = g.fromJson(topicMessage, LinkedTreeMap.class);
                testReadCreateMirrorMaker();
@@ -119,10 +109,7 @@ public class TestMirrorMakerAgent {
 
        @Test
        public void testReadListMirrorMaker() {
-               agent = spy(new MirrorMakerAgent());
-               when(agent.publishTopic(
-                               "{\"messageID\":\"test\",\"listMirrorMaker\":[{\"name\":\"test\",\"consumer\":\"test\",\"producer\":\"test\",\"status\":\"STOPPED\"}]}"))
-                                               .thenReturn(null);
+               
                String topicMessage = "{ messageID:\"test\", listAllMirrorMaker: {   name:\"test\",   consumer:\"test\",  producer:\"test\",  whitelist:\"test\",status:\"test\" }}";
                LinkedTreeMap<?, ?> object = g.fromJson(topicMessage, LinkedTreeMap.class);
                testReadCreateMirrorMaker();
@@ -132,10 +119,7 @@ public class TestMirrorMakerAgent {
 
        @Test
        public void testReadWhitelistMirrorMaker() {
-               agent = spy(new MirrorMakerAgent());
-               when(agent.publishTopic(
-                               "{\"messageID\":\"test\",\"listMirrorMaker\":[{\"name\":\"test\",\"consumer\":\"test\",\"producer\":\"test\",\"status\":\"STOPPED\"}]}"))
-                                               .thenReturn(null);
+               
                String topicMessage = "{ messageID:\"test\", updateWhiteList: {   name:\"test\",   consumer:\"test\",  producer:\"test\",  whitelist:\"test\",status:\"test\" }}";
                LinkedTreeMap<?, ?> object = g.fromJson(topicMessage, LinkedTreeMap.class);
                testReadCreateMirrorMaker();