src/test/java/com/att/nsa/cambria/backends/kafka/KafkaConsumerCacheTest.java

   1 /*-\r
   2  * ============LICENSE_START=======================================================\r
   3  * ONAP Policy Engine\r
   4  * ================================================================================\r
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.\r
   6  * ================================================================================\r
   7  * Licensed under the Apache License, Version 2.0 (the "License");\r
   8  * you may not use this file except in compliance with the License.\r
   9  * You may obtain a copy of the License at\r
  10  * \r
  11  *      http://www.apache.org/licenses/LICENSE-2.0\r
  12  * \r
  13  * Unless required by applicable law or agreed to in writing, software\r
  14  * distributed under the License is distributed on an "AS IS" BASIS,\r
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  16  * See the License for the specific language governing permissions and\r
  17  * limitations under the License.\r
  18  * ============LICENSE_END=========================================================\r
  19  */\r
  20 \r
  21 package com.att.nsa.cambria.backends.kafka;\r
  22 \r
  23 import static org.junit.Assert.*;\r
  24 \r
  25 import java.util.concurrent.ConcurrentHashMap;\r
  26 \r
  27 import org.apache.curator.framework.CuratorFramework;\r
  28 import org.junit.After;\r
  29 import org.junit.Before;\r
  30 import org.junit.Test;\r
  31 import org.junit.runner.RunWith;\r
  32 import org.mockito.Mock;\r
  33 import org.mockito.MockitoAnnotations;\r
  34 import org.powermock.api.mockito.PowerMockito;\r
  35 import org.powermock.modules.junit4.PowerMockRunner;\r
  36 import org.powermock.core.classloader.annotations.PrepareForTest;\r
  37 \r
  38 import com.att.ajsc.filemonitor.AJSCPropertiesMap;\r
  39 \r
  40 import com.att.dmf.mr.backends.MetricsSet;\r
  41 import com.att.dmf.mr.backends.kafka.Kafka011Consumer;\r
  42 import com.att.dmf.mr.backends.kafka.KafkaConsumerCache;\r
  43 import com.att.dmf.mr.backends.kafka.KafkaConsumerCache.KafkaConsumerCacheException;\r
  44 import com.att.dmf.mr.constants.CambriaConstants;\r
  45 import com.att.dmf.mr.security.DMaaPAuthenticatorImpl;\r
  46 \r
  47 @RunWith(PowerMockRunner.class)\r
  48 @PrepareForTest({ AJSCPropertiesMap.class })\r
  49 public class KafkaConsumerCacheTest {\r
  50         private KafkaConsumerCache kafkaConsumerCache =null;\r
  51         @Mock\r
  52         private ConcurrentHashMap<String, Kafka011Consumer> fConsumers;\r
  53         @Mock\r
  54         private MetricsSet fMetrics;\r
  55 \r
  56         @Before\r
  57         public void setUp() throws Exception {\r
  58                 MockitoAnnotations.initMocks(this);\r
  59                 \r
  60         }\r
  61 \r
  62         @After\r
  63         public void tearDown() throws Exception {\r
  64         }\r
  65 \r
  66 \r
  67         @Test\r
  68         public void testSweep() {\r
  69                 kafkaConsumerCache = new KafkaConsumerCache();\r
  70                 PowerMockito.mockStatic(AJSCPropertiesMap.class);\r
  71                 PowerMockito.when(AJSCPropertiesMap.getProperty(CambriaConstants.msgRtr_prop, "kSetting_TouchEveryMs")).thenReturn("100");\r
  72                 kafkaConsumerCache.sweep();\r
  73 \r
  74         }\r
  75         \r
  76 \r
  77         // DOES NOT WORK\r
  78         @Test\r
  79         public void testStartCache() {\r
  80 \r
  81                 /*\r
  82                  * KafkaConsumerCache kafka = null;\r
  83                  * \r
  84                  * try { kafka = new KafkaConsumerCache("123", null);\r
  85                  * \r
  86                  * } catch (NoClassDefFoundError e) { try { kafka.startCache("DMAAP",\r
  87                  * null); } catch (NullPointerException e1) { // TODO Auto-generated\r
  88                  * catch block assertTrue(true); } catch (KafkaConsumerCacheException\r
  89                  * e1) { // TODO Auto-generated catch block e1.printStackTrace(); } }\r
  90                  */\r
  91 \r
  92                 \r
  93                 new CuratorFrameworkImpl();\r
  94                 new MetricsSetImpl();\r
  95                 KafkaConsumerCache kafka=null;\r
  96                 try {\r
  97                         kafka = new KafkaConsumerCache();\r
  98                         kafka.setfApiId("1");\r
  99                         kafka.startCache("DMAAP", null);\r
 100                 } catch (NoClassDefFoundError e) {\r
 101 \r
 102                 } catch (KafkaConsumerCacheException e) {\r
 103                         // TODO Auto-generated catch block\r
 104                         e.printStackTrace();\r
 105                 }\r
 106 \r
 107         }\r
 108 \r
 109         @Test\r
 110         public void testGetCuratorFramework() {\r
 111 \r
 112                 CuratorFramework curator = new CuratorFrameworkImpl();\r
 113                 new MetricsSetImpl();\r
 114                 try {\r
 115 \r
 116                 } catch (NoClassDefFoundError e) {\r
 117 \r
 118                         KafkaConsumerCache.getCuratorFramework(curator);\r
 119                 }\r
 120 \r
 121         }\r
 122 \r
 123         /*\r
 124          * @Test public void testStopCache() {\r
 125          * \r
 126          * KafkaConsumerCache kafka = null; new CuratorFrameworkImpl(); new\r
 127          * MetricsSetImpl(); try { kafka = new KafkaConsumerCache("123", null);\r
 128          * kafka.stopCache(); } catch (NoClassDefFoundError e) {\r
 129          * \r
 130          * }\r
 131          * \r
 132          * }\r
 133          */\r
 134 \r
 135         @Test\r
 136         public void testGetConsumerFor() {\r
 137 \r
 138                 KafkaConsumerCache kafka = null;\r
 139 \r
 140                 try {\r
 141                         kafka = new KafkaConsumerCache();\r
 142                         kafka.getConsumerFor("testTopic", "CG1", "23");\r
 143                 } catch (NoClassDefFoundError e) {\r
 144 \r
 145                 } catch (KafkaConsumerCacheException e) {\r
 146                         // TODO Auto-generated catch block\r
 147                         e.printStackTrace();\r
 148                 }\r
 149 \r
 150         }\r
 151 \r
 152         @Test\r
 153         public void testPutConsumerFor() {\r
 154 \r
 155                 Kafka011Consumer consumer = null;\r
 156                 KafkaConsumerCache kafka = null;\r
 157 \r
 158                 try {\r
 159                         kafka = new KafkaConsumerCache();\r
 160 \r
 161                 } catch (NoClassDefFoundError e) {\r
 162                         try {\r
 163                                 kafka.putConsumerFor("testTopic", "CG1", "23", consumer);\r
 164                         } catch (NullPointerException e1) {\r
 165                                 // TODO Auto-generated catch block\r
 166                                 assertTrue(true);\r
 167                         } catch (KafkaConsumerCacheException e1) {\r
 168                                 // TODO Auto-generated catch block\r
 169                                 e1.printStackTrace();\r
 170                         }\r
 171                 }\r
 172 \r
 173         }\r
 174 \r
 175         @Test\r
 176         public void testGetConsumers() {\r
 177 \r
 178                 KafkaConsumerCache kafka = null;\r
 179 \r
 180                 try {\r
 181                         kafka = new KafkaConsumerCache();\r
 182 \r
 183                 } catch (NoClassDefFoundError e) {\r
 184                         try {\r
 185                                 kafka.getConsumers();\r
 186                         } catch (NullPointerException e1) {\r
 187                                 // TODO Auto-generated catch block\r
 188                                 assertTrue(true);\r
 189                         }\r
 190                 }\r
 191 \r
 192         }\r
 193 \r
 194         @Test\r
 195         public void testDropAllConsumers() {\r
 196 \r
 197                 KafkaConsumerCache kafka = null;\r
 198                 try {\r
 199                         kafka = new KafkaConsumerCache();\r
 200 \r
 201                 } catch (NoClassDefFoundError e) {\r
 202                         try {\r
 203                                 kafka.dropAllConsumers();\r
 204                         } catch (NullPointerException e1) {\r
 205                                 // TODO Auto-generated catch block\r
 206                                 assertTrue(true);\r
 207                         }\r
 208                 }\r
 209 \r
 210         }\r
 211 \r
 212         @Test\r
 213         public void testSignalOwnership() {\r
 214 \r
 215                 KafkaConsumerCache kafka = null;\r
 216 \r
 217                 try {\r
 218                         kafka = new KafkaConsumerCache();\r
 219                         // kafka.signalOwnership("testTopic", "CG1", "23");\r
 220                 } catch (NoClassDefFoundError e) {\r
 221                         try {\r
 222                                 kafka.signalOwnership("testTopic", "CG1", "23");\r
 223                         } catch (KafkaConsumerCacheException e1) {\r
 224                                 // TODO Auto-generated catch block\r
 225                                 e.printStackTrace();\r
 226                         } catch (NullPointerException e1) {\r
 227                                 // TODO Auto-generated catch block\r
 228                                 // assertTrue(true);\r
 229                                 e1.printStackTrace();\r
 230                         }\r
 231 \r
 232                 }\r
 233 \r
 234                 // assertTrue(true);\r
 235         }\r
 236 \r
 237         @Test\r
 238         public void testDropConsumer() {\r
 239 \r
 240                 KafkaConsumerCache kafka = null;\r
 241 \r
 242                 try {\r
 243                         kafka = new KafkaConsumerCache();\r
 244                         // kafka.dropConsumer("testTopic", "CG1", "23");\r
 245                 } catch (NoClassDefFoundError e) {\r
 246                         try {\r
 247                                 kafka.dropConsumer("testTopic", "CG1", "23");\r
 248                         } catch (NullPointerException e1) {\r
 249                                 // TODO Auto-generated catch block\r
 250                                 assertTrue(true);\r
 251                         }\r
 252                 }\r
 253 \r
 254         }\r
 255 \r
 256 }\r