3ff56ab17fb19d1c464945e7d7b2ead85576c2d6
[dcaegen2/services.git] /
1 /*******************************************************************************
2  *  ============LICENSE_START=======================================================
3  *  slice-analysis-ms
4  *  ================================================================================
5  *   Copyright (C) 2020 Wipro Limited.
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  *******************************************************************************/
21
22
23 package org.onap.slice.analysis.ms.dmaap;
24
25 import static org.junit.Assert.assertTrue;
26
27 import java.io.IOException;
28 import java.util.HashMap;
29 import java.util.Map;
30
31 import org.junit.Before;
32 import org.junit.Test;
33 import org.junit.runner.RunWith;
34 import org.mockito.InjectMocks;
35 import org.mockito.Mock;
36 import org.mockito.Mockito;
37 import org.mockito.runners.MockitoJUnitRunner;
38 import org.onap.slice.analysis.ms.models.Configuration;
39 import org.onap.slice.analysis.ms.utils.DmaapUtils;
40 import org.springframework.boot.test.context.SpringBootTest;
41
42 import com.att.nsa.cambria.client.CambriaBatchingPublisher;
43 import com.att.nsa.cambria.client.CambriaConsumer;
44
45 @RunWith(MockitoJUnitRunner.class)
46 @SpringBootTest(classes = PolicyDmaapClient.class)
47 public class PolicyDmaapClientTest {
48     
49     @Mock
50     Configuration configurationMock;
51     
52     @Mock
53     DmaapUtils dmaapUtilsMock;
54     
55     @InjectMocks
56     PolicyDmaapClient policyDmaapClient;
57     
58     @Mock
59     CambriaConsumer policyResponseCambriaConsumerMock;
60     
61     @Mock
62     CambriaBatchingPublisher cambriaBatchingPublisherMock;
63     
64     @Mock
65     NotificationProducer notificationProducerMock;
66     
67     @Before
68     public void setup() {
69         policyDmaapClient = new PolicyDmaapClient(dmaapUtilsMock, configurationMock);
70     }
71     
72     @Test
73     public void sendNotificationToPolicyTest() {
74         Map<String, Object> streamsPublishes = new HashMap<>();
75         Map<String, String> topics = new HashMap<>();
76         Map<String, Object> dmaapInfo = new HashMap<>();
77         topics.put("topic_url", "https://message-router.onap.svc.cluster.local:3905/events/DCAE_CL_OUTPUT");
78         dmaapInfo.put("dmaap_info", topics);
79         streamsPublishes.put("CL_topic", dmaapInfo);
80         Mockito.when(configurationMock.getStreamsPublishes()).thenReturn(streamsPublishes);
81         Mockito.when(dmaapUtilsMock.buildPublisher(configurationMock, "DCAE_CL_OUTPUT")).thenReturn(cambriaBatchingPublisherMock);
82         try {
83             Mockito.when(cambriaBatchingPublisherMock.send("", "hello")).thenReturn(0);
84         } catch (IOException e) {
85             e.printStackTrace();
86         }
87         assertTrue(policyDmaapClient.sendNotificationToPolicy("hello"));
88         
89     }
90 }