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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
/*-
* ============LICENSE_START=======================================================
* ONAP PAP
* ================================================================================
* Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
* Modifications Copyright (C) 2019 Nordix Foundation.
* ================================================================================
* 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.policy.common.endpoints.event.comm.client;
import static org.assertj.core.api.Assertions.assertThatThrownBy;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.mockito.Matchers.anyString;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;
import java.util.Properties;
import java.util.concurrent.atomic.AtomicReference;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.Test;
import org.onap.policy.common.endpoints.event.comm.TopicEndpointManager;
import org.onap.policy.common.endpoints.event.comm.TopicSink;
public class TopicSinkClientTest {
private static final String TOPIC = "my-topic";
private TopicSinkClient client;
private TopicSink sink;
private List<TopicSink> sinks;
/**
* Creates mocks and an initial client object.
*
* @throws Exception if an error occurs
*/
@Before
public void setUp() throws Exception {
sink = mock(TopicSink.class);
when(sink.send(anyString())).thenReturn(true);
sinks = Arrays.asList(sink, null);
client = new TopicSinkClient2(TOPIC);
Properties props = new Properties();
props.setProperty("noop.sink.topics", TOPIC);
// clear all topics and then configure one topic
TopicEndpointManager.getManager().shutdown();
TopicEndpointManager.getManager().addTopicSinks(props);
}
@AfterClass
public static void tearDown() {
// clear all topics after the tests
TopicEndpointManager.getManager().shutdown();
}
/**
* Uses a real NO-OP topic sink.
*/
@Test
public void testGetTopicSinks() throws Exception {
sink = TopicEndpointManager.getManager().getNoopTopicSink(TOPIC);
assertNotNull(sink);
final AtomicReference<String> evref = new AtomicReference<>(null);
sink.register((infra, topic, event) -> evref.set(event));
sink.start();
client = new TopicSinkClient(TOPIC);
client.send(100);
assertEquals("100", evref.get());
}
@Test
public void testTopicSinkClient() {
// unknown topic -> should throw exception
sinks = new LinkedList<>();
assertThatThrownBy(() -> new TopicSinkClient2(TOPIC)).isInstanceOf(TopicSinkClientException.class)
.hasMessage("no sinks for topic: my-topic");
}
@Test
public void testTopicSinkClient_GetTopic() throws TopicSinkClientException {
assertEquals(TOPIC, new TopicSinkClient(TopicEndpointManager.getManager().getNoopTopicSink(TOPIC)).getTopic());
assertEquals(TOPIC, new TopicSinkClient(TOPIC).getTopic());
assertThatThrownBy(() -> new TopicSinkClient((TopicSink) null)).isInstanceOf(IllegalArgumentException.class);
assertThatThrownBy(() -> new TopicSinkClient("blah")).isInstanceOf(TopicSinkClientException.class)
.hasMessage("no sinks for topic: blah");
}
@Test
public void testSend() {
client.send(Arrays.asList("abc", "def"));
verify(sink).send("['abc','def']".replace('\'', '"'));
// sink send fails
when(sink.send(anyString())).thenReturn(false);
assertFalse(client.send("ghi"));
// sink send throws an exception
final RuntimeException ex = new RuntimeException("expected exception");
when(sink.send(anyString())).thenThrow(ex);
assertFalse(client.send("jkl"));
}
/**
* TopicSinkClient with some overrides.
*/
private class TopicSinkClient2 extends TopicSinkClient {
public TopicSinkClient2(final String topic) throws TopicSinkClientException {
super(topic);
}
@Override
protected List<TopicSink> getTopicSinks(final String topic) {
return sinks;
}
}
}
|
