1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
/*-
* ============LICENSE_START=======================================================
* org.onap.dcaegen2.collectors.ves
* ================================================================================
* Copyright (C) 2018,2020 Nokia. All rights reserved.
* Copyright (C) 2020 AT&T. All rights reserved.
* ================================================================================
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
* ============LICENSE_END=========================================================
*/
package org.onap.dcae.common.publishing;
import com.att.nsa.cambria.client.CambriaBatchingPublisher;
import org.json.JSONObject;
import org.junit.Before;
import org.junit.Test;
import org.onap.dcae.common.model.VesEvent;
import java.io.IOException;
import static io.vavr.API.Option;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.BDDMockito.given;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
public class DMaaPEventPublisherTest {
private static final String STREAM_ID = "sampleStreamId";
private static final JSONObject EXPECTED_EVENT =
new JSONObject(
"{\"VESversion\":\"v7\",\"event\":{"
+ "\"commonEventHeader\":{\"startEpochMicrosec\":1537562659253019,"
+ "\"sourceId\":\"79e90d76-513a-4f79-886d-470a0037c5cf\","
+ "\"eventId\":\"Heartbeat_vDNS_100.100.10.10\",\"nfcNamingCode\":\"DNS\","
+ "\"reportingEntityId\":\"79e90d76-513a-4f79-886d-470a0037c5cf\","
+ "\"eventType\":\"applicationVnf\",\"priority\":\"Normal\",\"version\":3,"
+ "\"reportingEntityName\":\"dns01cmd004\",\"sequence\":36312,\"domain\":\"heartbeat\","
+ "\"lastEpochMicrosec\":1537562659253019,\"eventName\":\"Heartbeat_vDNS\","
+ "\"sourceName\":\"dns01cmd004\",\"nfNamingCode\":\"MDNS\"}}}");
private static final String PARTITION = "dns01cmd004";
private DMaaPEventPublisher eventPublisher;
private CambriaBatchingPublisher cambriaPublisher;
private DMaaPPublishersCache DMaaPPublishersCache;
@Before
public void setUp() {
cambriaPublisher = mock(CambriaBatchingPublisher.class);
DMaaPPublishersCache = mock(DMaaPPublishersCache.class);
when(DMaaPPublishersCache.getPublisher(anyString())).thenReturn(Option(cambriaPublisher));
eventPublisher = new DMaaPEventPublisher(DMaaPPublishersCache);
}
@Test
public void shouldSendEventToTopic() throws Exception {
// when
eventPublisher.sendEvent(givenVesEventWithoutVESuniqueIdField(), STREAM_ID);
// then
verify(cambriaPublisher).send(PARTITION, EXPECTED_EVENT.toString());
}
@Test
public void shouldRemoveInternalVESUIDBeforeSending() throws Exception {
// when
eventPublisher.sendEvent(givenVesEventWithVESUniqueIdField(), STREAM_ID);
// then
verify(cambriaPublisher).send(PARTITION, EXPECTED_EVENT.toString());
}
@Test
public void shouldCloseConnectionWhenExceptionOccurred() throws Exception {
// given
given(cambriaPublisher.send(anyString(), anyString()))
.willThrow(new IOException("Expected exception - test case scenario!"));
// when
eventPublisher.sendEvent(givenVesEventWithVESUniqueIdField(), STREAM_ID);
// then
verify(DMaaPPublishersCache).closePublisherFor(STREAM_ID);
}
private VesEvent givenVesEventWithVESUniqueIdField() {
return new VesEvent(
new JSONObject(
"{\"VESversion\":\"v7\",\"VESuniqueId\":\"fd69d432-5cd5-4c15-9d34-407c81c61c6a-0\"," +
"\"event\":{" +
"\"commonEventHeader\":{\"startEpochMicrosec\":1537562659253019," +
"\"sourceId\":\"79e90d76-513a-4f79-886d-470a0037c5cf\"," +
"\"eventId\":\"Heartbeat_vDNS_100.100.10.10\",\"nfcNamingCode\":\"DNS\"," +
"\"reportingEntityId\":\"79e90d76-513a-4f79-886d-470a0037c5cf\",\"eventType\":\"applicationVnf\"," +
"\"priority\":\"Normal\",\"version\":3,\"reportingEntityName\":\"dns01cmd004\",\"sequence\":36312," +
"\"domain\":\"heartbeat\",\"lastEpochMicrosec\":1537562659253019,\"eventName\":\"Heartbeat_vDNS\"," +
"\"sourceName\":\"dns01cmd004\",\"nfNamingCode\":\"MDNS\"}}}"));
}
private VesEvent givenVesEventWithoutVESuniqueIdField() {
return new VesEvent(
new JSONObject(
"{\"VESversion\":\"v7\"," +
"\"event\":{" +
"\"commonEventHeader\":{\"startEpochMicrosec\":1537562659253019," +
"\"sourceId\":\"79e90d76-513a-4f79-886d-470a0037c5cf\"," +
"\"eventId\":\"Heartbeat_vDNS_100.100.10.10\",\"nfcNamingCode\":\"DNS\"," +
"\"reportingEntityId\":\"79e90d76-513a-4f79-886d-470a0037c5cf\",\"eventType\":\"applicationVnf\"," +
"\"priority\":\"Normal\",\"version\":3,\"reportingEntityName\":\"dns01cmd004\",\"sequence\":36312," +
"\"domain\":\"heartbeat\",\"lastEpochMicrosec\":1537562659253019,\"eventName\":\"Heartbeat_vDNS\"," +
"\"sourceName\":\"dns01cmd004\",\"nfNamingCode\":\"MDNS\"}}}"));
}
}
|