c6ed44556c8eee4acd56a380548ff6dff7ca152c
[policy/apex-pdp.git] /
1 /*-
2  * ============LICENSE_START=======================================================
3  *  Copyright (C) 2019 Nordix Foundation.
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.services.onappf.comm;
22
23 import static org.junit.Assert.assertEquals;
24
25 import java.io.FileInputStream;
26 import java.io.FileNotFoundException;
27 import java.io.IOException;
28 import java.nio.charset.StandardCharsets;
29 import java.nio.file.Files;
30 import java.nio.file.Paths;
31 import java.util.ArrayList;
32 import java.util.LinkedHashMap;
33 import java.util.List;
34 import java.util.Map;
35 import java.util.Properties;
36
37 import org.junit.After;
38 import org.junit.Before;
39 import org.junit.Test;
40 import org.onap.policy.apex.services.onappf.ApexStarterActivator;
41 import org.onap.policy.apex.services.onappf.ApexStarterCommandLineArguments;
42 import org.onap.policy.apex.services.onappf.ApexStarterConstants;
43 import org.onap.policy.apex.services.onappf.exception.ApexStarterException;
44 import org.onap.policy.apex.services.onappf.parameters.ApexStarterParameterGroup;
45 import org.onap.policy.apex.services.onappf.parameters.ApexStarterParameterHandler;
46 import org.onap.policy.common.endpoints.event.comm.Topic.CommInfrastructure;
47 import org.onap.policy.common.utils.services.Registry;
48 import org.onap.policy.models.pdp.concepts.PdpStateChange;
49 import org.onap.policy.models.pdp.concepts.PdpStatus;
50 import org.onap.policy.models.pdp.concepts.PdpUpdate;
51 import org.onap.policy.models.pdp.enums.PdpState;
52 import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicy;
53
54 /**
55  * Class to perform unit test of {@link PdpStateChangeListener}.
56  *
57  * @author Ajith Sreekumar (ajith.sreekumar@est.tech)
58  */
59 public class TestPdpStateChangeListener {
60     private PdpUpdateListener pdpUpdateMessageListener;
61     private PdpStateChangeListener pdpStateChangeListener;
62     private static final CommInfrastructure INFRA = CommInfrastructure.NOOP;
63     private static final String TOPIC = "my-topic";
64     private ApexStarterActivator activator;
65
66     /**
67      * Method for setup before each test.
68      *
69      * @throws ApexStarterException if some error occurs while starting up the apex starter
70      * @throws FileNotFoundException if the file is missing
71      * @throws IOException if IO exception occurs
72      */
73     @Before
74     public void setUp() throws ApexStarterException, FileNotFoundException, IOException {
75         pdpUpdateMessageListener = new PdpUpdateListener();
76         pdpStateChangeListener = new PdpStateChangeListener();
77         Registry.newRegistry();
78         final String[] apexStarterConfigParameters = { "-c", "src/test/resources/ApexStarterConfigParameters.json",
79             "-p", "src/test/resources/topic.properties" };
80         final ApexStarterCommandLineArguments arguments = new ApexStarterCommandLineArguments();
81         ApexStarterParameterGroup apexStarterParameterGroup;
82         // The arguments return a string if there is a message to print and we should
83         // exit
84         final String argumentMessage = arguments.parse(apexStarterConfigParameters);
85         if (argumentMessage != null) {
86             return;
87         }
88         // Validate that the arguments are sane
89         arguments.validate();
90
91         // Read the parameters
92         apexStarterParameterGroup = new ApexStarterParameterHandler().getParameters(arguments);
93
94         // Read the properties
95         final Properties topicProperties = new Properties();
96         final String propFile = arguments.getFullPropertyFilePath();
97         try (FileInputStream stream = new FileInputStream(propFile)) {
98             topicProperties.load(stream);
99         }
100         activator = new ApexStarterActivator(apexStarterParameterGroup, topicProperties);
101         Registry.register(ApexStarterConstants.REG_APEX_STARTER_ACTIVATOR, activator);
102         activator.initialize();
103     }
104
105     /**
106      * Method for cleanup after each test.
107      *
108      * @throws Exception if an error occurs
109      */
110     @After
111     public void teardown() throws Exception {
112
113         // clear the apex starter activator
114         if (activator != null && activator.isAlive()) {
115             activator.terminate();
116         }
117     }
118
119     /**
120      * Method to initiate a PdpUpdate.
121      *
122      * @param instance the instance id
123      * @return PdpUpdate the pdp update message
124      */
125     private PdpUpdate performPdpUpdate(final String instance) {
126         final PdpUpdate pdpUpdateMsg = new PdpUpdate();
127         pdpUpdateMsg.setDescription("dummy pdp status for test");
128         pdpUpdateMsg.setPdpGroup("pdpGroup");
129         pdpUpdateMsg.setPdpSubgroup("pdpSubgroup");
130         pdpUpdateMsg.setName(instance);
131         final ToscaPolicy toscaPolicy = new ToscaPolicy();
132         toscaPolicy.setType("apexpolicytype");
133         toscaPolicy.setVersion("1.0");
134         final Map<String, Object> propertiesMap = new LinkedHashMap<>();
135
136         String properties;
137         try {
138             properties = new String(Files.readAllBytes(Paths.get("src\\test\\resources\\dummyProperties.json")),
139                     StandardCharsets.UTF_8);
140             propertiesMap.put("content", properties);
141         } catch (final IOException e) {
142             propertiesMap.put("content", "");
143         }
144         toscaPolicy.setProperties(propertiesMap);
145         final List<ToscaPolicy> toscaPolicies = new ArrayList<ToscaPolicy>();
146         toscaPolicies.add(toscaPolicy);
147         pdpUpdateMsg.setPolicies(toscaPolicies);
148         pdpUpdateMessageListener.onTopicEvent(INFRA, TOPIC, null, pdpUpdateMsg);
149         return pdpUpdateMsg;
150     }
151
152     @Test
153     public void testPdpStateChangeMessageListener_passivetopassive() {
154         final PdpStatus pdpStatus = Registry.get(ApexStarterConstants.REG_PDP_STATUS_OBJECT);
155         performPdpUpdate(pdpStatus.getName());
156         final PdpStateChange pdpStateChangeMsg = new PdpStateChange();
157         pdpStateChangeMsg.setState(PdpState.PASSIVE);
158         pdpStateChangeMsg.setPdpGroup("pdpGroup");
159         pdpStateChangeMsg.setPdpSubgroup("pdpSubgroup");
160         pdpStateChangeMsg.setName(pdpStatus.getName());
161         pdpStateChangeListener.onTopicEvent(INFRA, TOPIC, null, pdpStateChangeMsg);
162
163         assertEquals(pdpStatus.getState(), pdpStateChangeMsg.getState());
164     }
165
166     @Test
167     public void testPdpStateChangeMessageListener_activetoactive() {
168         final PdpStatus pdpStatus = Registry.get(ApexStarterConstants.REG_PDP_STATUS_OBJECT);
169         performPdpUpdate(pdpStatus.getName());
170         pdpStatus.setState(PdpState.ACTIVE);
171         final PdpStateChange pdpStateChangeMsg = new PdpStateChange();
172         pdpStateChangeMsg.setState(PdpState.ACTIVE);
173         pdpStateChangeMsg.setPdpGroup("pdpGroup");
174         pdpStateChangeMsg.setPdpSubgroup("pdpSubgroup");
175         pdpStateChangeMsg.setName(pdpStatus.getName());
176         pdpStateChangeListener.onTopicEvent(INFRA, TOPIC, null, pdpStateChangeMsg);
177
178         assertEquals(pdpStatus.getState(), pdpStateChangeMsg.getState());
179     }
180 }