update the testcases after the kafka 11 changes

[dmaap/messagerouter/msgrtr.git] / src / test / java / com / att / nsa / cambria / service / impl / MMServiceImplTest.java

diff --git a/src/test/java/com/att/nsa/cambria/service/impl/MMServiceImplTest.java b/src/test/java/com/att/nsa/cambria/service/impl/MMServiceImplTest.java
index 8a69a8a..f97952c 100644 (file)
--- a/src/test/java/com/att/nsa/cambria/service/impl/MMServiceImplTest.java
+++ b/src/test/java/com/att/nsa/cambria/service/impl/MMServiceImplTest.java
@@ -26,25 +26,27 @@ import java.io.ByteArrayInputStream;
 import java.io.IOException;\r
 import java.io.InputStream;\r
 import java.util.Arrays;\r
+import java.util.ConcurrentModificationException;\r
 import java.util.HashSet;\r
 import java.util.Map;\r
 import java.util.Set;\r
 \r
 import com.att.ajsc.beans.PropertiesMapBean;\r
 import com.att.ajsc.filemonitor.AJSCPropertiesMap;\r
-import com.att.nsa.cambria.CambriaApiException;\r
-import com.att.nsa.cambria.backends.ConsumerFactory;\r
-import com.att.nsa.cambria.backends.ConsumerFactory.UnavailableException;\r
-import com.att.nsa.cambria.beans.DMaaPContext;\r
-import com.att.nsa.cambria.beans.DMaaPKafkaMetaBroker;\r
-import com.att.nsa.cambria.constants.CambriaConstants;\r
-import com.att.nsa.cambria.exception.DMaaPErrorMessages;\r
-import com.att.nsa.cambria.metabroker.Topic;\r
-import com.att.nsa.cambria.metabroker.Broker.TopicExistsException;\r
-import com.att.nsa.cambria.security.DMaaPAuthenticatorImpl;\r
-import com.att.nsa.cambria.utils.ConfigurationReader;\r
-import com.att.nsa.cambria.utils.DMaaPResponseBuilder;\r
-import com.att.nsa.cambria.utils.Emailer;\r
+import com.att.dmf.mr.CambriaApiException;\r
+import com.att.dmf.mr.backends.ConsumerFactory;\r
+import com.att.dmf.mr.backends.ConsumerFactory.UnavailableException;\r
+import com.att.dmf.mr.beans.DMaaPContext;\r
+import com.att.dmf.mr.beans.DMaaPKafkaMetaBroker;\r
+import com.att.dmf.mr.constants.CambriaConstants;\r
+import com.att.dmf.mr.exception.DMaaPErrorMessages;\r
+import com.att.dmf.mr.metabroker.Topic;\r
+import com.att.dmf.mr.metabroker.Broker.TopicExistsException;\r
+import com.att.dmf.mr.security.DMaaPAuthenticatorImpl;\r
+import com.att.dmf.mr.service.impl.MMServiceImpl;\r
+import com.att.dmf.mr.utils.ConfigurationReader;\r
+import com.att.dmf.mr.utils.DMaaPResponseBuilder;\r
+import com.att.dmf.mr.utils.Emailer;\r
 import com.att.nsa.configs.ConfigDbException;\r
 import com.att.nsa.drumlin.till.nv.rrNvReadable.missingReqdSetting;\r
 import com.att.nsa.limits.Blacklist;\r
@@ -191,6 +193,15 @@ public class MMServiceImplTest {
                }\r
 \r
        }\r
+       \r
+       @Test(expected = CambriaApiException.class)\r
+       public void testSubscribe_NullTopic_Error() throws ConfigDbException, TopicExistsException, AccessDeniedException, UnavailableException,\r
+       CambriaApiException, IOException {\r
+\r
+                   PowerMockito.when(configReader.getfMetrics()).thenThrow(new ConcurrentModificationException("Error occurred"));\r
+                       PowerMockito.when(dmaapKafkaMetaBroker.getTopic(anyString())).thenReturn(metatopic);\r
+                       service.subscribe(dmaapContext, "testTopic", "CG1", "23");\r
+       }\r
 \r
        @Test\r
        public void testPushEvents_wttransaction() {\r
@@ -227,6 +238,20 @@ public class MMServiceImplTest {
                        e.printStackTrace();\r
                }\r
        }\r
+       \r
+       @Test(expected = CambriaApiException.class)\r
+       public void testPushEvents_wttransaction_error() throws Exception {\r
+\r
+               String source = "source of my InputStream";\r
+\r
+               InputStream iStream = new ByteArrayInputStream(source.getBytes("UTF-8"));\r
+               PowerMockito.mockStatic(AJSCPropertiesMap.class);\r
+               PowerMockito.mockStatic(PropertiesMapBean.class);\r
+               PowerMockito.when(AJSCPropertiesMap.getProperty(CambriaConstants.msgRtr_prop, "event.batch.length")).thenReturn("-5");\r
+               PowerMockito.when(configReader.getfPublisher()).thenThrow(new ConcurrentModificationException("Error occurred"));\r
+               service.pushEvents(dmaapContext, "msgrtr.apinode.metrics.dmaap1", iStream, "3", "12:00:00");\r
+\r
+       }\r
 \r
        @Test\r
        public void testPushEvents() {\r