2 * ============LICENSE_START=======================================================
3 * Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
4 * ================================================================================
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
9 * http://www.apache.org/licenses/LICENSE-2.0
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
17 * SPDX-License-Identifier: Apache-2.0
18 * ============LICENSE_END=========================================================
21 package org.onap.policy.pdpx.main.comm.listeners;
23 import static org.junit.Assert.assertNotNull;
24 import static org.mockito.Matchers.any;
25 import static org.mockito.Matchers.anyLong;
26 import static org.mockito.Mockito.doThrow;
27 import static org.mockito.Mockito.never;
28 import static org.mockito.Mockito.verify;
29 import static org.mockito.Mockito.when;
31 import org.junit.Before;
32 import org.junit.Test;
33 import org.mockito.Mock;
34 import org.mockito.MockitoAnnotations;
35 import org.onap.policy.common.endpoints.event.comm.Topic.CommInfrastructure;
36 import org.onap.policy.common.endpoints.event.comm.client.TopicSinkClient;
37 import org.onap.policy.models.pdp.concepts.PdpUpdate;
38 import org.onap.policy.pdpx.main.XacmlState;
39 import org.onap.policy.pdpx.main.comm.XacmlPdpHearbeatPublisher;
40 import org.onap.policy.pdpx.main.comm.XacmlPdpUpdatePublisher;
41 import org.onap.policy.pdpx.main.comm.listeners.XacmlPdpUpdateListener;
42 import org.onap.policy.pdpx.main.rest.XacmlPdpApplicationManager;
43 import org.powermock.reflect.Whitebox;
45 public class XacmlPdpUpdateListenerTest {
46 private static final String EXPECTED_EXCEPTION = "expected exception";
47 private static final String TOPIC = "my-topic";
48 private static final long HB_INTERVAL = 100L;
51 private TopicSinkClient client;
54 private XacmlState state;
57 private XacmlPdpHearbeatPublisher heartbeat;
60 private XacmlPdpApplicationManager appmgr;
63 private XacmlPdpUpdatePublisher publisher;
65 private PdpUpdate update;
67 private XacmlPdpUpdateListener listener;
70 * Initializes objects, including the listener.
74 MockitoAnnotations.initMocks(this);
76 listener = new MyListener(client, state, heartbeat, appmgr);
77 update = new PdpUpdate();
79 when(state.shouldHandle(update)).thenReturn(true);
81 update.setPdpHeartbeatIntervalMs(HB_INTERVAL);
85 public void testOnTopicEvent_Unhandled() {
86 when(state.shouldHandle(update)).thenReturn(false);
87 listener.onTopicEvent(CommInfrastructure.NOOP, TOPIC, null, update);
89 verify(publisher, never()).handlePdpUpdate(any());
90 verify(heartbeat, never()).restart(anyLong());
94 public void testOnTopicEvent_SendOk() {
95 listener.onTopicEvent(CommInfrastructure.NOOP, TOPIC, null, update);
97 verify(publisher).handlePdpUpdate(update);
98 verify(heartbeat).restart(HB_INTERVAL);
102 public void testOnTopicEvent_SendEx() {
103 doThrow(new RuntimeException(EXPECTED_EXCEPTION)).when(publisher).handlePdpUpdate(update);
105 listener.onTopicEvent(CommInfrastructure.NOOP, TOPIC, null, update);
107 verify(publisher).handlePdpUpdate(update);
108 verify(heartbeat, never()).restart(anyLong());
112 public void testMakePublisher() {
113 // create a plain listener to test the "real" makePublisher() method
114 listener = new XacmlPdpUpdateListener(client, state, heartbeat, appmgr);
115 assertNotNull(Whitebox.getInternalState(listener, "publisher"));
118 private class MyListener extends XacmlPdpUpdateListener {
120 public MyListener(TopicSinkClient client, XacmlState state, XacmlPdpHearbeatPublisher heartbeat,
121 XacmlPdpApplicationManager appManager) {
122 super(client, state, heartbeat, appManager);
126 protected XacmlPdpUpdatePublisher makePublisher(TopicSinkClient client, XacmlState state,
127 XacmlPdpApplicationManager appManager) {