1 files changed, 148 insertions, 0 deletions
diff --git a/models-sim/policy-models-sim-pdp/src/test/java/org/onap/policy/models/sim/pdp/comm/TestPdpUpdateListener.java b/models-sim/policy-models-sim-pdp/src/test/java/org/onap/policy/models/sim/pdp/comm/TestPdpUpdateListener.java
new file mode 100644
index 000000000..f3885f42d
--- /dev/null
+++ b/models-sim/policy-models-sim-pdp/src/test/java/org/onap/policy/models/sim/pdp/comm/TestPdpUpdateListener.java
@@ -0,0 +1,148 @@
+/*-
+ * ============LICENSE_START=======================================================
+ *  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.
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.models.sim.pdp.comm;
+
+import static org.junit.Assert.assertEquals;
+
+import java.io.FileInputStream;
+import java.io.FileNotFoundException;
+import java.io.IOException;
+import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
+import java.nio.file.Paths;
+import java.util.ArrayList;
+import java.util.LinkedHashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Properties;
+
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.onap.policy.common.endpoints.event.comm.Topic.CommInfrastructure;
+import org.onap.policy.common.utils.services.Registry;
+import org.onap.policy.models.pdp.concepts.PdpStatus;
+import org.onap.policy.models.pdp.concepts.PdpUpdate;
+import org.onap.policy.models.sim.pdp.PdpSimulatorActivator;
+import org.onap.policy.models.sim.pdp.PdpSimulatorCommandLineArguments;
+import org.onap.policy.models.sim.pdp.PdpSimulatorConstants;
+import org.onap.policy.models.sim.pdp.exception.PdpSimulatorException;
+import org.onap.policy.models.sim.pdp.handler.PdpMessageHandler;
+import org.onap.policy.models.sim.pdp.parameters.PdpSimulatorParameterGroup;
+import org.onap.policy.models.sim.pdp.parameters.PdpSimulatorParameterHandler;
+import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicy;
+
+/**
+ * Class to perform unit test of {@link PdpUpdateListener}.
+ *
+ * @author Ajith Sreekumar (ajith.sreekumar@est.tech)
+ */
+public class TestPdpUpdateListener {
+    private PdpUpdateListener pdpUpdateMessageListener;
+    private static final CommInfrastructure INFRA = CommInfrastructure.NOOP;
+    private static final String TOPIC = "my-topic";
+    private PdpSimulatorActivator activator;
+
+    /**
+     * Method for setup before each test.
+     *
+     * @throws PdpSimulatorException if some error occurs while starting up the pdp simulator
+     * @throws FileNotFoundException if the file is missing
+     * @throws IOException if IO exception occurs
+     */
+    @Before
+    public void setUp() throws PdpSimulatorException, FileNotFoundException, IOException {
+        Registry.newRegistry();
+        final String[] pdpSimulatorConfigParameters = { "-c", "src/test/resources/PdpSimulatorConfigParameters.json",
+            "-p", "src/test/resources/topic.properties" };
+        final PdpSimulatorCommandLineArguments arguments = new PdpSimulatorCommandLineArguments();
+        PdpSimulatorParameterGroup pdpSimulatorParameterGroup;
+        // The arguments return a string if there is a message to print and we should
+        // exit
+        final String argumentMessage = arguments.parse(pdpSimulatorConfigParameters);
+        if (argumentMessage != null) {
+            return;
+        }
+        // Validate that the arguments are sane
+        arguments.validate();
+
+        // Read the parameters
+        pdpSimulatorParameterGroup = new PdpSimulatorParameterHandler().getParameters(arguments);
+
+        // Read the properties
+        final Properties topicProperties = new Properties();
+        final String propFile = arguments.getFullPropertyFilePath();
+        try (FileInputStream stream = new FileInputStream(propFile)) {
+            topicProperties.load(stream);
+        }
+        activator = new PdpSimulatorActivator(pdpSimulatorParameterGroup, topicProperties);
+        Registry.register(PdpSimulatorConstants.REG_PDP_SIMULATOR_ACTIVATOR, activator);
+        activator.initialize();
+        pdpUpdateMessageListener = new PdpUpdateListener();
+    }
+
+    /**
+     * Method for cleanup after each test.
+     *
+     * @throws Exception if an error occurs
+     */
+    @After
+    public void teardown() throws Exception {
+
+        // clear the pdp simulator activator
+        if (activator != null && activator.isAlive()) {
+            activator.terminate();
+        }
+    }
+
+    @Test
+    public void testPdpUpdateMssageListener() {
+        final PdpStatus pdpStatus = Registry.get(PdpSimulatorConstants.REG_PDP_STATUS_OBJECT);
+        final PdpUpdate pdpUpdateMsg = new PdpUpdate();
+        pdpUpdateMsg.setDescription("dummy pdp status for test");
+        pdpUpdateMsg.setPdpGroup("pdpGroup");
+        pdpUpdateMsg.setPdpSubgroup("pdpSubgroup");
+        pdpUpdateMsg.setName(pdpStatus.getName());
+        pdpUpdateMsg.setPdpHeartbeatIntervalMs(Long.valueOf(3000));
+        final ToscaPolicy toscaPolicy = new ToscaPolicy();
+        toscaPolicy.setType("apexpolicytype");
+        toscaPolicy.setVersion("1.0");
+        toscaPolicy.setName("apex policy name");
+        final Map<String, Object> propertiesMap = new LinkedHashMap<>();
+        String properties;
+        try {
+            properties = new String(Files.readAllBytes(Paths.get("src\\test\\resources\\dummyProperties.json")),
+                    StandardCharsets.UTF_8);
+            propertiesMap.put("content", properties);
+        } catch (final IOException e) {
+            propertiesMap.put("content", "");
+        }
+        toscaPolicy.setProperties(propertiesMap);
+        final List<ToscaPolicy> toscaPolicies = new ArrayList<ToscaPolicy>();
+        toscaPolicies.add(toscaPolicy);
+        pdpUpdateMsg.setPolicies(toscaPolicies);
+        pdpUpdateMessageListener.onTopicEvent(INFRA, TOPIC, null, pdpUpdateMsg);
+        assertEquals(pdpStatus.getPdpGroup(), pdpUpdateMsg.getPdpGroup());
+        assertEquals(pdpStatus.getPdpSubgroup(), pdpUpdateMsg.getPdpSubgroup());
+        assertEquals(pdpStatus.getPolicies(),
+                new PdpMessageHandler().getToscaPolicyIdentifiers(pdpUpdateMsg.getPolicies()));
+    }
+}