gerrit.onap Code Review - policy/models.git/blob

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP
   4  * ================================================================================
   5  * Copyright (C) 2020 AT&T Intellectual Property. All rights reserved.
   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 package org.onap.policy.controlloop.actorserviceprovider.topic;
  22
  23 import static org.junit.Assert.assertNotNull;
  24 import static org.junit.Assert.assertNotSame;
  25 import static org.junit.Assert.assertSame;
  26 import static org.mockito.ArgumentMatchers.any;
  27 import static org.mockito.ArgumentMatchers.eq;
  28 import static org.mockito.Mockito.never;
  29 import static org.mockito.Mockito.verify;
  30
  31 import java.util.Arrays;
  32 import java.util.Map;
  33 import java.util.function.BiConsumer;
  34 import org.junit.Before;
  35 import org.junit.Test;
  36 import org.mockito.Mock;
  37 import org.mockito.MockitoAnnotations;
  38 import org.onap.policy.common.endpoints.event.comm.Topic.CommInfrastructure;
  39 import org.onap.policy.common.utils.coder.CoderException;
  40 import org.onap.policy.common.utils.coder.StandardCoder;
  41 import org.onap.policy.common.utils.coder.StandardCoderObject;
  42
  43 public class TopicListenerImplTest {
  44     private static final StandardCoder coder = new StandardCoder();
  45     private static final CommInfrastructure INFRA = CommInfrastructure.NOOP;
  46     private static final String MY_TOPIC = "my-topic";
  47     private static final String KEY1 = "requestId";
  48     private static final String KEY2 = "container";
  49     private static final String SUBKEY = "subRequestId";
  50
  51     private static final String VALUEA_REQID = "hello";
  52     private static final String VALUEA_SUBREQID = "world";
  53
  54     private static final String VALUEB_REQID = "bye";
  55
  56     private Forwarder forwarder1;
  57     private Forwarder forwarder2;
  58     private TopicListenerImpl topic;
  59
  60     @Mock
  61     private BiConsumer<String, StandardCoderObject> listener1;
  62
  63     @Mock
  64     private BiConsumer<String, StandardCoderObject> listener1b;
  65
  66     @Mock
  67     private BiConsumer<String, StandardCoderObject> listener2;
  68
  69
  70     /**
  71      * Sets up.
  72      */
  73     @Before
  74     public void setUp() {
  75         MockitoAnnotations.initMocks(this);
  76
  77         topic = new TopicListenerImpl();
  78
  79         forwarder1 = topic.addForwarder(new SelectorKey(KEY1));
  80         forwarder2 = topic.addForwarder(new SelectorKey(KEY1), new SelectorKey(KEY2, SUBKEY));
  81
  82         assertNotNull(forwarder1);
  83         assertNotNull(forwarder2);
  84         assertNotSame(forwarder1, forwarder2);
  85
  86         forwarder1.register(Arrays.asList(VALUEA_REQID), listener1);
  87         forwarder1.register(Arrays.asList(VALUEB_REQID), listener1b);
  88         forwarder2.register(Arrays.asList(VALUEA_REQID, VALUEA_SUBREQID), listener2);
  89     }
  90
  91     @Test
  92     public void testShutdown() {
  93         // shut it down, which should clear all forwarders
  94         topic.shutdown();
  95
  96         // should get a new forwarder now
  97         Forwarder forwarder = topic.addForwarder(new SelectorKey(KEY1));
  98         assertNotSame(forwarder1, forwarder);
  99         assertNotSame(forwarder2, forwarder);
 100
 101         // new forwarder should be unchanged
 102         assertSame(forwarder, topic.addForwarder(new SelectorKey(KEY1)));
 103     }
 104
 105     @Test
 106     public void testAddForwarder() {
 107         assertSame(forwarder1, topic.addForwarder(new SelectorKey(KEY1)));
 108         assertSame(forwarder2, topic.addForwarder(new SelectorKey(KEY1), new SelectorKey(KEY2, SUBKEY)));
 109     }
 110
 111     @Test
 112     public void testOnTopicEvent() {
 113         /*
 114          * send a message that should go to listener1 on forwarder1 and listener2 on
 115          * forwarder2
 116          */
 117         String msg = makeMessage(Map.of(KEY1, VALUEA_REQID, KEY2, Map.of(SUBKEY, VALUEA_SUBREQID)));
 118         topic.onTopicEvent(INFRA, MY_TOPIC, msg);
 119
 120         verify(listener1).accept(eq(msg), any());
 121         verify(listener2).accept(eq(msg), any());
 122
 123         // not to listener1b
 124         verify(listener1b, never()).accept(any(), any());
 125
 126         /*
 127          * now send a message that should only go to listener1b on forwarder1
 128          */
 129         msg = makeMessage(Map.of(KEY1, VALUEB_REQID, KEY2, Map.of(SUBKEY, VALUEA_SUBREQID)));
 130         topic.onTopicEvent(INFRA, MY_TOPIC, msg);
 131
 132         // should route to listener1 on forwarder1 and listener2 on forwarder2
 133         verify(listener1b).accept(eq(msg), any());
 134
 135         // try one where the coder throws an exception
 136         topic.onTopicEvent(INFRA, MY_TOPIC, "{invalid-json");
 137
 138         // no extra invocations
 139         verify(listener1).accept(any(), any());
 140         verify(listener1b).accept(any(), any());
 141         verify(listener2).accept(any(), any());
 142     }
 143
 144     /**
 145      * Makes a message from a map.
 146      */
 147     private String makeMessage(Map<String, Object> map) {
 148         try {
 149             return coder.encode(map);
 150         } catch (CoderException e) {
 151             throw new IllegalArgumentException(e);
 152         }
 153     }
 154 }