src/test/java/com/att/nsa/cambria/beans/DMaaPKafkaMetaBrokerTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP Policy Engine
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20 package com.att.nsa.cambria.beans;
  21
  22 import static org.junit.Assert.assertTrue;
  23
  24 import org.I0Itec.zkclient.ZkClient;
  25 import org.I0Itec.zkclient.exception.ZkNoNodeException;
  26 import org.junit.Before;
  27 import org.junit.Test;
  28 import org.junit.runner.RunWith;
  29 import org.mockito.InjectMocks;
  30 import org.mockito.Mock;
  31 import org.mockito.MockitoAnnotations;
  32 import org.powermock.api.mockito.PowerMockito;
  33 import org.powermock.core.classloader.annotations.PrepareForTest;
  34 import org.powermock.modules.junit4.PowerMockRunner;
  35
  36 import com.att.nsa.cambria.CambriaApiException;
  37 import com.att.nsa.cambria.metabroker.Broker.TopicExistsException;
  38 import com.att.nsa.configs.ConfigDb;
  39 import com.att.nsa.configs.ConfigDbException;
  40 import com.att.nsa.configs.ConfigPath;
  41
  42 import kafka.admin.AdminUtils;
  43
  44 @RunWith(PowerMockRunner.class)
  45 @PrepareForTest({ AdminUtils.class, ZkClientFactory.class })
  46 public class DMaaPKafkaMetaBrokerTest {
  47
  48         @InjectMocks
  49         private DMaaPKafkaMetaBroker dMaaPKafkaMetaBroker;
  50
  51         @Mock
  52         private ZkClient zk;
  53         @Mock
  54         private ConfigDb configDb;
  55         @Mock
  56         ConfigPath fBaseTopicData;
  57         @Mock
  58         private ZkClient zkClient;
  59
  60         @Before
  61         public void setUp() {
  62                 MockitoAnnotations.initMocks(this);
  63                 PowerMockito.mockStatic(AdminUtils.class);
  64                 PowerMockito.mockStatic(ZkClientFactory.class);
  65                 PowerMockito.when(configDb.parse("/topics")).thenReturn(fBaseTopicData);
  66
  67         }
  68
  69         @Test
  70         public void testGetAlltopics() {
  71                 try {
  72                         dMaaPKafkaMetaBroker.getAllTopics();
  73                 } catch (ConfigDbException e) {
  74                         // TODO Auto-generated catch block
  75                         e.printStackTrace();
  76                 }
  77
  78         }
  79
  80         @Test
  81         public void testcreateTopic() {
  82                 try {
  83                         PowerMockito.when(ZkClientFactory.createZkClient()).thenReturn(zkClient);
  84                         dMaaPKafkaMetaBroker.createTopic("testtopic", "testtopic", "admin", 1, 1, true);
  85                 } catch (CambriaApiException e) {
  86                         // TODO Auto-generated catch block
  87                         e.printStackTrace();
  88                 } catch (TopicExistsException e) {
  89                         // TODO Auto-generated catch block
  90                         e.printStackTrace();
  91                 } catch (Exception e) {
  92                         // TODO Auto-generatee.printStackTrace();
  93                 }
  94
  95         }
  96
  97         @Test
  98         public void testcreateTopic_wrongPartition() {
  99                 try {
 100
 101                         PowerMockito.when(ZkClientFactory.createZkClient()).thenReturn(zkClient);
 102                         dMaaPKafkaMetaBroker.createTopic("testtopic", "testtopic", "admin", 0, 1, true);
 103                 } catch (CambriaApiException e) {
 104                         assertTrue(true);
 105                 } catch (TopicExistsException e) {
 106                         // TODO Auto-generated catch block
 107                         e.printStackTrace();
 108                 } catch (Exception e) {
 109                         // TODO Auto-generatee.printStackTrace();
 110                 }
 111
 112         }
 113
 114         @Test
 115         public void testcreateTopic_wrongReplica() {
 116                 try {
 117
 118                         PowerMockito.when(ZkClientFactory.createZkClient()).thenReturn(zkClient);
 119                         dMaaPKafkaMetaBroker.createTopic("testtopic", "testtopic", "admin", 1, 0, true);
 120                 } catch (CambriaApiException e) {
 121                         assertTrue(true);
 122                 } catch (TopicExistsException e) {
 123                         // TODO Auto-generated catch block
 124                         e.printStackTrace();
 125                 } catch (Exception e) {
 126                         // TODO Auto-generatee.printStackTrace();
 127                 }
 128
 129         }
 130
 131         @Test
 132         public void testcreateTopic_error1() {
 133                 try {
 134                         PowerMockito.when(ZkClientFactory.createZkClient()).thenThrow(new ZkNoNodeException());
 135                         dMaaPKafkaMetaBroker.createTopic("testtopic", "testtopic", "admin", 1, 1, true);
 136                 } catch (CambriaApiException e) {
 137                         assertTrue(true);
 138                 } catch (TopicExistsException e) {
 139                         // TODO Auto-generated catch block
 140                         e.printStackTrace();
 141                 } catch (Exception e) {
 142                         e.printStackTrace();
 143                 }
 144
 145         }
 146
 147         @Test
 148         public void testcreateTopic_error2() {
 149                 try {
 150                         PowerMockito.when(ZkClientFactory.createZkClient())
 151                                         .thenThrow(new kafka.admin.AdminOperationException("error"));
 152                         dMaaPKafkaMetaBroker.createTopic("testtopic", "testtopic", "admin", 1, 1, true);
 153                 } catch (CambriaApiException e) {
 154                         assertTrue(true);
 155                 } catch (TopicExistsException e) {
 156                         // TODO Auto-generated catch block
 157                         e.printStackTrace();
 158                 } catch (Exception e) {
 159                         e.printStackTrace();
 160                 }
 161         }
 162
 163         @Test
 164         public void testcreateTopic_error3() {
 165                 try {
 166                         PowerMockito.when(ZkClientFactory.createZkClient()).thenThrow(new kafka.common.TopicExistsException());
 167                         dMaaPKafkaMetaBroker.createTopic("testtopic", "testtopic", "admin", 1, 1, true);
 168                 } catch (CambriaApiException e) {
 169                         // TODO Auto-generated catch block
 170                         e.printStackTrace();
 171                 } catch (TopicExistsException e) {
 172                         assertTrue(true);
 173
 174                 } catch (Exception e) {
 175                         e.printStackTrace();
 176                 }
 177
 178         }
 179
 180         @Test
 181         public void testDeleteTopic() {
 182                 try {
 183                         PowerMockito.when(ZkClientFactory.createZkClient()).thenReturn(zkClient);
 184                         dMaaPKafkaMetaBroker.deleteTopic("testtopic");
 185                 } catch (CambriaApiException e) {
 186                         // TODO Auto-generated catch block
 187                         e.printStackTrace();
 188                 } catch (TopicExistsException e) {
 189                         // TODO Auto-generated catch block
 190                         e.printStackTrace();
 191                 } catch (Exception e) {
 192                         e.printStackTrace();
 193                 }
 194                 assertTrue(true);
 195
 196         }
 197
 198         @Test
 199         public void testDeleteTopic_error1() {
 200                 try {
 201                         PowerMockito.when(ZkClientFactory.createZkClient()).thenThrow(new ZkNoNodeException());
 202                         dMaaPKafkaMetaBroker.deleteTopic("testtopic");
 203                 } catch (CambriaApiException e) {
 204                         assertTrue(true);
 205                 } catch (TopicExistsException e) {
 206                         // TODO Auto-generated catch block
 207                         e.printStackTrace();
 208                 } catch (Exception e) {
 209                         e.printStackTrace();
 210                 }
 211
 212         }
 213
 214         @Test
 215         public void testDeleteTopic_error2() {
 216                 try {
 217                         PowerMockito.when(ZkClientFactory.createZkClient())
 218                                         .thenThrow(new kafka.admin.AdminOperationException("error"));
 219                         dMaaPKafkaMetaBroker.deleteTopic("testtopic");
 220                 } catch (CambriaApiException e) {
 221                         assertTrue(true);
 222                 } catch (TopicExistsException e) {
 223                         // TODO Auto-generated catch block
 224                         e.printStackTrace();
 225                 } catch (Exception e) {
 226                         e.printStackTrace();
 227                 }
 228
 229         }
 230
 231         @Test
 232         public void testDeleteTopic_error3() {
 233                 try {
 234                         PowerMockito.when(ZkClientFactory.createZkClient()).thenThrow(new kafka.common.TopicExistsException());
 235                         dMaaPKafkaMetaBroker.deleteTopic("testtopic");
 236                 } catch (CambriaApiException e) {
 237                         // TODO Auto-generated catch block
 238                         e.printStackTrace();
 239                 } catch (TopicExistsException e) {
 240                         assertTrue(true);
 241                 } catch (Exception e) {
 242                         e.printStackTrace();
 243                 }
 244
 245         }
 246
 247 }