gerrit.onap Code Review - policy/apex-pdp.git/blob

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2016-2018 Ericsson. 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
   8  *
   9  *      http://www.apache.org/licenses/LICENSE-2.0
  10  *
  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.
  16  *
  17  * SPDX-License-Identifier: Apache-2.0
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.policy.apex.testsuites.integration.uservice.adapt.jms;
  22
  23 import static org.junit.Assert.assertEquals;
  24
  25 import java.io.IOException;
  26 import java.nio.file.Paths;
  27 import java.util.ArrayList;
  28 import java.util.Arrays;
  29 import java.util.List;
  30
  31 import javax.jms.JMSException;
  32
  33 import org.apache.activemq.ActiveMQConnectionFactory;
  34 import org.apache.activemq.broker.BrokerPlugin;
  35 import org.apache.activemq.broker.BrokerService;
  36 import org.apache.activemq.security.AuthenticationUser;
  37 import org.apache.activemq.security.SimpleAuthenticationPlugin;
  38 import org.junit.AfterClass;
  39 import org.junit.BeforeClass;
  40 import org.junit.Test;
  41 import org.onap.policy.apex.core.infrastructure.threading.ThreadUtilities;
  42 import org.onap.policy.apex.model.basicmodel.concepts.ApexException;
  43 import org.onap.policy.apex.service.engine.main.ApexMain;
  44 import org.slf4j.ext.XLogger;
  45 import org.slf4j.ext.XLoggerFactory;
  46
  47 /**
  48  * The Class TestJms2Jms.
  49  */
  50 public class TestJms2Jms {
  51     public static final String PORT = "5445";
  52     public static final String HOST = "localhost";
  53     public static final String JMS_TOPIC_APEX_IN = "jms/topic/apexIn";
  54     public static final String JMS_TOPIC_APEX_OUT = "jms/topic/apexOut";
  55
  56     private static final int SLEEP_TIME = 1500;
  57     private static final String GROUP_ROLE = "guests";
  58     private static final String PACKAGE_NAME = "org.onap.policy.apex.testsuites.integration.uservice.adapt.jms";
  59     private static final String USERNAME = "guest";
  60     private static final String PASSWORD = "IAmAGuest";
  61     private static final String URL = "tcp://" + HOST + ":" + PORT;
  62
  63     private static final String DATA_PARENT_DIR = Paths.get("target", "activemq-data").toString();
  64
  65     private static final XLogger LOGGER = XLoggerFactory.getXLogger(TestJms2Jms.class);
  66
  67     private static final long MAX_TEST_LENGTH = 10000;
  68     private static final int EVENT_COUNT = 100;
  69     private static final int EVENT_INTERVAL = 20;
  70
  71     private static BrokerService broker;
  72
  73     public static ActiveMQConnectionFactory connectionFactory;
  74
  75
  76     /**
  77      * Setup embedded jms server.
  78      *
  79      * @throws Exception the exception
  80      */
  81     @BeforeClass
  82     public static void setupEmbeddedJmsServer() throws Exception {
  83         final ArrayList<BrokerPlugin> plugins = new ArrayList<BrokerPlugin>();
  84         final BrokerPlugin authenticationPlugin = getAuthenticationBrokerPlugin();
  85         plugins.add(authenticationPlugin);
  86
  87         broker = new BrokerService();
  88         broker.setUseJmx(false);
  89         broker.setPersistent(false);
  90         broker.addConnector(URL);
  91         broker.setDeleteAllMessagesOnStartup(true);
  92         broker.setPlugins(plugins.toArray(new BrokerPlugin[0]));
  93         broker.setDataDirectory(DATA_PARENT_DIR);
  94         broker.start();
  95         broker.waitUntilStarted();
  96         connectionFactory = new ActiveMQConnectionFactory(URL);
  97         connectionFactory.setTrustedPackages(Arrays.asList(PACKAGE_NAME));
  98     }
  99
 100     /**
 101      * Gets the authentication broker plugin.
 102      *
 103      * @return the authentication broker plugin
 104      */
 105     private static BrokerPlugin getAuthenticationBrokerPlugin() {
 106         final List<AuthenticationUser> users = new ArrayList<AuthenticationUser>();
 107         users.add(new AuthenticationUser(USERNAME, PASSWORD, GROUP_ROLE));
 108         final SimpleAuthenticationPlugin authenticationPlugin = new SimpleAuthenticationPlugin(users);
 109         return authenticationPlugin;
 110     }
 111
 112     /**
 113      * Shutdown embedded jms server.
 114      *
 115      * @throws IOException Signals that an I/O exception has occurred.
 116      */
 117     @AfterClass
 118     public static void shutdownEmbeddedJmsServer() throws IOException {
 119         try {
 120             if (broker != null) {
 121                 broker.stop();
 122             }
 123         } catch (final Exception e) {
 124             LOGGER.warn("Failed to stop JMS server", e);
 125         }
 126
 127     }
 128
 129     /**
 130      * Test jms object events.
 131      *
 132      * @throws ApexException the apex exception
 133      * @throws JMSException the JMS exception
 134      */
 135     @Test
 136     public void testJmsObjectEvents() throws ApexException, JMSException {
 137         final String[] args = { "src/test/resources/prodcons/JMS2JMSObjectEvent.json" };
 138         testJmsEvents(args, true);
 139     }
 140
 141     /**
 142      * Test jms json events.
 143      *
 144      * @throws ApexException the apex exception
 145      * @throws JMSException the JMS exception
 146      */
 147     @Test
 148     public void testJmsJsonEvents() throws ApexException, JMSException {
 149         final String[] args = { "src/test/resources/prodcons/JMS2JMSJsonEvent.json" };
 150         testJmsEvents(args, false);
 151     }
 152
 153     /**
 154      * Test jms events.
 155      *
 156      * @param args the args
 157      * @param sendObjects the send objects
 158      * @throws ApexException the apex exception
 159      * @throws JMSException the JMS exception
 160      */
 161     private void testJmsEvents(final String[] args, final Boolean sendObjects) throws ApexException, JMSException {
 162         final JmsEventSubscriber subscriber =
 163                 new JmsEventSubscriber(JMS_TOPIC_APEX_OUT, connectionFactory, USERNAME, PASSWORD);
 164         final JmsEventProducer producer = new JmsEventProducer(JMS_TOPIC_APEX_IN, connectionFactory, USERNAME, PASSWORD,
 165                 EVENT_COUNT, sendObjects, EVENT_INTERVAL);
 166
 167         final ApexMain apexMain = new ApexMain(args);
 168         ThreadUtilities.sleep(3000);
 169
 170         producer.sendEvents();
 171
 172         final long testStartTime = System.currentTimeMillis();
 173
 174         while (isTimedOut(testStartTime) && subscriber.getEventsReceivedCount() < EVENT_COUNT) {
 175             ThreadUtilities.sleep(EVENT_INTERVAL);
 176         }
 177
 178         ThreadUtilities.sleep(SLEEP_TIME);
 179         apexMain.shutdown();
 180         subscriber.shutdown();
 181         producer.shutdown();
 182         ThreadUtilities.sleep(SLEEP_TIME);
 183
 184         assertEquals(EVENT_COUNT, producer.getEventsSentCount());
 185         assertEquals(producer.getEventsSentCount(), subscriber.getEventsReceivedCount());
 186
 187     }
 188
 189     /**
 190      * Checks if is timed out.
 191      *
 192      * @param testStartTime the test start time
 193      * @return true, if is timed out
 194      */
 195     private boolean isTimedOut(final long testStartTime) {
 196         return System.currentTimeMillis() < testStartTime + MAX_TEST_LENGTH;
 197     }
 198 }