appc-adapters/appc-dmaap-adapter/appc-dmaap-adapter-bundle/src/test/java/org/onap/appc/adapter/messaging/dmaap/impl/TestDmaapProducerImpl.java

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * ONAP : APPC
   4  * ================================================================================
   5  * Copyright (C) 2018 AT&T Intellectual Property. All rights reserved.
   6  * =============================================================================
   7  * Modifications Copyright (C) 2018 IBM.
   8  * =================================================================================
   9  * Modifications Copyright (C) 2019 Ericsson
  10  * =============================================================================
  11  * Licensed under the Apache License, Version 2.0 (the "License");
  12  * you may not use this file except in compliance with the License.
  13  * You may obtain a copy of the License at
  14  *
  15  *      http://www.apache.org/licenses/LICENSE-2.0
  16  *
  17  * Unless required by applicable law or agreed to in writing, software
  18  * distributed under the License is distributed on an "AS IS" BASIS,
  19  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  20  * See the License for the specific language governing permissions and
  21  * limitations under the License.
  22  * ============LICENSE_END=========================================================
  23  */
  24
  25 package org.onap.appc.adapter.messaging.dmaap.impl;
  26
  27 import static org.junit.Assert.assertEquals;
  28 import static org.junit.Assert.assertNotNull;
  29 import static org.junit.Assert.assertNull;
  30 import java.io.FileNotFoundException;
  31 import java.io.IOException;
  32 import java.util.Arrays;
  33 import java.util.Collection;
  34 import java.util.HashSet;
  35 import java.util.Properties;
  36 import java.util.Set;
  37 import org.junit.Before;
  38 import org.junit.Test;
  39 import org.junit.runner.RunWith;
  40 import org.mockito.Mockito;
  41 import org.onap.appc.configuration.Configuration;
  42 import org.onap.appc.configuration.ConfigurationFactory;
  43 import org.onap.appc.metricservice.MetricRegistry;
  44 import org.onap.appc.metricservice.MetricService;
  45 import org.onap.appc.metricservice.metric.DmaapRequestCounterBuilder;
  46 import org.onap.appc.metricservice.metric.DmaapRequestCounterMetric;
  47 import org.onap.appc.metricservice.metric.MetricBuilderFactory;
  48 import org.onap.appc.metricservice.metric.impl.MetricBuilderFactoryImpl;
  49 import org.onap.appc.metricservice.policy.PolicyBuilderFactory;
  50 import org.onap.appc.metricservice.policy.PublishingPolicy;
  51 import org.onap.appc.metricservice.policy.ScheduledPolicyBuilder;
  52 import org.osgi.framework.FrameworkUtil;
  53 import org.powermock.api.mockito.PowerMockito;
  54 import org.powermock.core.classloader.annotations.PrepareForTest;
  55 import org.powermock.modules.junit4.PowerMockRunner;
  56 import org.powermock.reflect.Whitebox;
  57 import com.att.nsa.mr.client.MRBatchingPublisher;
  58 import com.att.nsa.mr.client.MRClientFactory;
  59
  60 @RunWith(PowerMockRunner.class)
  61 @PrepareForTest({ConfigurationFactory.class, FrameworkUtil.class, MRClientFactory.class})
  62 public class TestDmaapProducerImpl {
  63     String[] hostList = { "192.168.1.1" };
  64     Collection<String> hosts = new HashSet<String>(Arrays.asList(hostList));
  65
  66     String topic = "JunitTopicOne";
  67     String group = "junit-client";
  68     String id = "junit-consumer-one";
  69     String key = "key";
  70     String secret = "secret";
  71     String filter = null;
  72
  73     private DmaapProducerImpl producer;
  74
  75     @Before
  76     public void setUp() {
  77         producer = new DmaapProducerImpl(hosts, topic, null, null);
  78     }
  79
  80     @Test
  81     public void testDmaapProducerImplSingleTopic() {
  82         producer = new DmaapProducerImpl(hosts, topic, key, secret);
  83
  84         assertNotNull(producer);
  85
  86         Properties props = producer.getProperties();
  87
  88         assertNotNull(props);
  89
  90         assertEquals("key", props.getProperty("username"));
  91         assertEquals("secret", props.getProperty("password"));
  92     }
  93
  94     @Test
  95     public void testDmaapProducerImplMultipleTopic() {
  96         String[] topicList = { "topic1", "topic2" };
  97         Set<String> topicNames = new HashSet<String>(Arrays.asList(topicList));
  98
  99         producer = new DmaapProducerImpl(hosts, topicNames, key, secret);
 100
 101         assertNotNull(producer);
 102
 103         Properties props = producer.getProperties();
 104
 105         assertNotNull(props);
 106
 107         assertEquals("key", props.getProperty("username"));
 108         assertEquals("secret", props.getProperty("password"));
 109
 110     }
 111
 112     @Test
 113     public void testDmaapProducerImplNoUserPass() {
 114         assertNotNull(producer);
 115
 116         Properties props = producer.getProperties();
 117
 118         assertNotNull(props);
 119
 120         assertNull(props.getProperty("username"));
 121         assertNull(props.getProperty("password"));
 122     }
 123
 124     @Test
 125     public void testUpdateCredentials() {
 126         assertNotNull(producer);
 127
 128         Properties props = producer.getProperties();
 129
 130         assertNotNull(props);
 131
 132         assertNull(props.getProperty("username"));
 133         assertNull(props.getProperty("password"));
 134
 135         producer.updateCredentials(key, secret);
 136
 137         props = producer.getProperties();
 138
 139         assertNotNull(props);
 140
 141         assertEquals("key", props.getProperty("username"));
 142         assertEquals("secret", props.getProperty("password"));
 143
 144     }
 145
 146     @Test
 147     public void testPost() {
 148         producer = new DmaapProducerImpl(hosts, topic, key, secret);
 149         boolean successful = producer.post("partition", "data");
 150         assertEquals(true, successful);
 151     }
 152
 153     @Test
 154     public void testCloseNoClient() {
 155         producer = new DmaapProducerImpl(hosts, topic, key, secret);
 156
 157         assertNotNull(producer);
 158
 159         producer.close();
 160     }
 161
 162     @Test
 163     public void testCloseWithClient() {
 164         producer = new DmaapProducerImpl(hosts, topic, key, secret);
 165         producer.post("partition", "data");
 166         assertNotNull(producer);
 167         producer.close();
 168     }
 169
 170     @Test
 171     public void testUseHttps() {
 172         producer = new DmaapProducerImpl(hosts, topic, key, secret);
 173
 174         assertNotNull(producer);
 175
 176         assertEquals(false, producer.isHttps());
 177
 178         producer.useHttps(true);
 179
 180         assertEquals(true, producer.isHttps());
 181
 182     }
 183
 184     @Test
 185     public void testInitMetric() throws FileNotFoundException, IOException
 186     {
 187         Configuration configuration = Mockito.mock(Configuration.class);
 188         Properties properties = new Properties();
 189         properties.put("metric.enabled", "true");
 190         Mockito.when(configuration.getProperties()).thenReturn(properties);
 191         PowerMockito.mockStatic(MRClientFactory.class);
 192         PowerMockito.when(MRClientFactory.createConsumer(Mockito.anyString())).thenThrow(new IOException());
 193         DmaapProducerImpl producer = Mockito.spy(new DmaapProducerImpl(hosts, topic, key, secret));
 194         Whitebox.setInternalState(DmaapProducerImpl.class, "configuration", configuration);
 195         MetricService metricService = Mockito.mock(MetricService.class);
 196         MetricRegistry metricRegistry = Mockito.mock(MetricRegistry.class);
 197         MetricBuilderFactory metricBuilderFactory = Mockito.spy(new MetricBuilderFactoryImpl());
 198         DmaapRequestCounterBuilder builder = Mockito.mock(DmaapRequestCounterBuilder.class);
 199         DmaapRequestCounterMetric metric = Mockito.mock(DmaapRequestCounterMetric.class);
 200         Mockito.doNothing().when(metric).incrementPublishedMessage();
 201         Mockito.when(builder.withName(Mockito.anyString())).thenReturn(builder);
 202         Mockito.when(builder.withType(Mockito.any())).thenReturn(builder);
 203         Mockito.when(builder.withPublishedMessage(Mockito.anyLong())).thenReturn(builder);
 204         Mockito.when(builder.withRecievedMessage(Mockito.anyLong())).thenReturn(builder);
 205         Mockito.when(builder.build()).thenReturn(metric);
 206         Mockito.when(metricBuilderFactory.dmaapRequestCounterBuilder()).thenReturn(builder);
 207         Mockito.when(metricRegistry.metric("DMAAP_KPI")).thenReturn(metric);
 208         Mockito.when(metricRegistry.register(Mockito.any())).thenReturn(true);
 209         PublishingPolicy policy = Mockito.mock(PublishingPolicy.class);
 210         PolicyBuilderFactory policyFactory = Mockito.mock(PolicyBuilderFactory.class);
 211         Mockito.when(metricRegistry.policyBuilderFactory()).thenReturn(policyFactory);
 212         ScheduledPolicyBuilder policyBuilder = Mockito.mock(ScheduledPolicyBuilder.class);
 213         Mockito.when(policyBuilder.withPublishers(Mockito.any())).thenReturn(policyBuilder);
 214         Mockito.when(policyBuilder.withMetrics(Mockito.any())).thenReturn(policyBuilder);
 215         Mockito.when(policyBuilder.build()).thenReturn(policy);
 216         Mockito.when(policyFactory.scheduledPolicyBuilder()).thenReturn(policyBuilder);
 217         Mockito.when(metricRegistry.metricBuilderFactory()).thenReturn(metricBuilderFactory);
 218         Mockito.when(metricService.createRegistry("APPC")).thenReturn(metricRegistry);
 219         Mockito.doReturn(metricService).when(producer).getMetricservice();
 220         Set<MRBatchingPublisher> clients = new HashSet<>();
 221         MRBatchingPublisher client = Mockito.mock(MRBatchingPublisher.class);
 222         clients.add(client);
 223         Mockito.doReturn(0).when(client).send(Mockito.anyString(), Mockito.anyString());
 224         Whitebox.setInternalState(producer, "clients", clients);
 225         producer.post(null, null);
 226         Mockito.verify(policy).init();
 227     }
 228
 229 }