2 * ============LICENSE_START=======================================================
\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
11 * http://www.apache.org/licenses/LICENSE-2.0
\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
21 package com.att.nsa.cambria.backends.kafka;
\r
23 import static org.junit.Assert.*;
\r
25 import java.util.concurrent.ConcurrentHashMap;
\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
38 import com.att.ajsc.filemonitor.AJSCPropertiesMap;
\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
47 @RunWith(PowerMockRunner.class)
\r
48 @PrepareForTest({ AJSCPropertiesMap.class })
\r
49 public class KafkaConsumerCacheTest {
\r
50 private KafkaConsumerCache kafkaConsumerCache =null;
\r
52 private ConcurrentHashMap<String, Kafka011Consumer> fConsumers;
\r
54 private MetricsSet fMetrics;
\r
57 public void setUp() throws Exception {
\r
58 MockitoAnnotations.initMocks(this);
\r
63 public void tearDown() throws Exception {
\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
79 public void testStartCache() {
\r
82 * KafkaConsumerCache kafka = null;
\r
84 * try { kafka = new KafkaConsumerCache("123", null);
\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
93 new CuratorFrameworkImpl();
\r
94 new MetricsSetImpl();
\r
95 KafkaConsumerCache kafka=null;
\r
97 kafka = new KafkaConsumerCache();
\r
98 kafka.setfApiId("1");
\r
99 kafka.startCache("DMAAP", null);
\r
100 } catch (NoClassDefFoundError e) {
\r
102 } catch (KafkaConsumerCacheException e) {
\r
103 // TODO Auto-generated catch block
\r
104 e.printStackTrace();
\r
110 public void testGetCuratorFramework() {
\r
112 CuratorFramework curator = new CuratorFrameworkImpl();
\r
113 new MetricsSetImpl();
\r
116 } catch (NoClassDefFoundError e) {
\r
118 KafkaConsumerCache.getCuratorFramework(curator);
\r
124 * @Test public void testStopCache() {
\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
136 public void testGetConsumerFor() {
\r
138 KafkaConsumerCache kafka = null;
\r
141 kafka = new KafkaConsumerCache();
\r
142 kafka.getConsumerFor("testTopic", "CG1", "23");
\r
143 } catch (NoClassDefFoundError e) {
\r
145 } catch (KafkaConsumerCacheException e) {
\r
146 // TODO Auto-generated catch block
\r
147 e.printStackTrace();
\r
153 public void testPutConsumerFor() {
\r
155 Kafka011Consumer consumer = null;
\r
156 KafkaConsumerCache kafka = null;
\r
159 kafka = new KafkaConsumerCache();
\r
161 } catch (NoClassDefFoundError e) {
\r
163 kafka.putConsumerFor("testTopic", "CG1", "23", consumer);
\r
164 } catch (NullPointerException e1) {
\r
165 // TODO Auto-generated catch block
\r
167 } catch (KafkaConsumerCacheException e1) {
\r
168 // TODO Auto-generated catch block
\r
169 e1.printStackTrace();
\r
176 public void testGetConsumers() {
\r
178 KafkaConsumerCache kafka = null;
\r
181 kafka = new KafkaConsumerCache();
\r
183 } catch (NoClassDefFoundError e) {
\r
185 kafka.getConsumers();
\r
186 } catch (NullPointerException e1) {
\r
187 // TODO Auto-generated catch block
\r
195 public void testDropAllConsumers() {
\r
197 KafkaConsumerCache kafka = null;
\r
199 kafka = new KafkaConsumerCache();
\r
201 } catch (NoClassDefFoundError e) {
\r
203 kafka.dropAllConsumers();
\r
204 } catch (NullPointerException e1) {
\r
205 // TODO Auto-generated catch block
\r
213 public void testSignalOwnership() {
\r
215 KafkaConsumerCache kafka = null;
\r
218 kafka = new KafkaConsumerCache();
\r
219 // kafka.signalOwnership("testTopic", "CG1", "23");
\r
220 } catch (NoClassDefFoundError e) {
\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
234 // assertTrue(true);
\r
238 public void testDropConsumer() {
\r
240 KafkaConsumerCache kafka = null;
\r
243 kafka = new KafkaConsumerCache();
\r
244 // kafka.dropConsumer("testTopic", "CG1", "23");
\r
245 } catch (NoClassDefFoundError e) {
\r
247 kafka.dropConsumer("testTopic", "CG1", "23");
\r
248 } catch (NullPointerException e1) {
\r
249 // TODO Auto-generated catch block
\r