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

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2019-2020 Nordix Foundation.
   4  *  Modifications Copyright (C) 2021 AT&T Intellectual Property. All rights reserved.
   5  *  Modifications Copyright (C) 2021 Bell Canada. 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  *
  19  * SPDX-License-Identifier: Apache-2.0
  20  * ============LICENSE_END=========================================================
  21  */
  22
  23 package org.onap.policy.apex.plugins.executor.javascript;
  24
  25 import static org.assertj.core.api.Assertions.assertThatThrownBy;
  26 import static org.junit.Assert.assertEquals;
  27 import static org.junit.Assert.assertNotNull;
  28
  29 import java.util.HashMap;
  30 import java.util.LinkedHashMap;
  31 import java.util.Map;
  32 import java.util.Properties;
  33 import java.util.TreeMap;
  34 import org.junit.AfterClass;
  35 import org.junit.BeforeClass;
  36 import org.junit.Test;
  37 import org.onap.policy.apex.context.ContextAlbum;
  38 import org.onap.policy.apex.context.ContextException;
  39 import org.onap.policy.apex.context.Distributor;
  40 import org.onap.policy.apex.context.impl.ContextAlbumImpl;
  41 import org.onap.policy.apex.context.impl.distribution.jvmlocal.JvmLocalDistributor;
  42 import org.onap.policy.apex.context.impl.schema.java.JavaSchemaHelperParameters;
  43 import org.onap.policy.apex.context.parameters.ContextParameterConstants;
  44 import org.onap.policy.apex.context.parameters.ContextParameters;
  45 import org.onap.policy.apex.context.parameters.SchemaParameters;
  46 import org.onap.policy.apex.core.engine.context.ApexInternalContext;
  47 import org.onap.policy.apex.model.basicmodel.concepts.AxArtifactKey;
  48 import org.onap.policy.apex.model.basicmodel.service.ModelService;
  49 import org.onap.policy.apex.model.contextmodel.concepts.AxContextAlbum;
  50 import org.onap.policy.apex.model.contextmodel.concepts.AxContextSchema;
  51 import org.onap.policy.apex.model.contextmodel.concepts.AxContextSchemas;
  52 import org.onap.policy.apex.model.eventmodel.concepts.AxEvent;
  53 import org.onap.policy.apex.model.eventmodel.concepts.AxField;
  54 import org.onap.policy.apex.model.policymodel.concepts.AxPolicyModel;
  55 import org.onap.policy.apex.model.policymodel.concepts.AxTask;
  56 import org.onap.policy.common.parameters.ParameterService;
  57 import org.onap.policy.common.utils.resources.TextFileUtils;
  58
  59 /**
  60  * Test the JavaTaskExecutor class.
  61  *
  62  */
  63 public class JavascriptTaskExecutorTest {
  64     /**
  65      * Set ups everything for the test.
  66      */
  67     @BeforeClass
  68     public static void prepareForTest() {
  69         final ContextParameters contextParameters = new ContextParameters();
  70         contextParameters.getLockManagerParameters()
  71                 .setPluginClass("org.onap.policy.apex.context.impl.locking.jvmlocal.JvmLocalLockManager");
  72
  73         contextParameters.setName(ContextParameterConstants.MAIN_GROUP_NAME);
  74         contextParameters.getDistributorParameters().setName(ContextParameterConstants.DISTRIBUTOR_GROUP_NAME);
  75         contextParameters.getLockManagerParameters().setName(ContextParameterConstants.LOCKING_GROUP_NAME);
  76         contextParameters.getPersistorParameters().setName(ContextParameterConstants.PERSISTENCE_GROUP_NAME);
  77
  78         ParameterService.register(contextParameters);
  79         ParameterService.register(contextParameters.getDistributorParameters());
  80         ParameterService.register(contextParameters.getLockManagerParameters());
  81         ParameterService.register(contextParameters.getPersistorParameters());
  82
  83         final SchemaParameters schemaParameters = new SchemaParameters();
  84         schemaParameters.setName(ContextParameterConstants.SCHEMA_GROUP_NAME);
  85         schemaParameters.getSchemaHelperParameterMap().put("JAVA", new JavaSchemaHelperParameters());
  86
  87         ParameterService.register(schemaParameters);
  88     }
  89
  90     /**
  91      * Clear down the test data.
  92      */
  93     @AfterClass
  94     public static void cleanUpAfterTest() {
  95         ParameterService.deregister(ContextParameterConstants.DISTRIBUTOR_GROUP_NAME);
  96         ParameterService.deregister(ContextParameterConstants.LOCKING_GROUP_NAME);
  97         ParameterService.deregister(ContextParameterConstants.PERSISTENCE_GROUP_NAME);
  98         ParameterService.deregister(ContextParameterConstants.SCHEMA_GROUP_NAME);
  99         ParameterService.deregister(ContextParameterConstants.MAIN_GROUP_NAME);
 100         ParameterService.clear();
 101     }
 102
 103     @Test
 104     public void testJavascriptTaskExecutor() throws Exception {
 105         JavascriptTaskExecutor jte = new JavascriptTaskExecutor();
 106
 107         assertThatThrownBy(() -> {
 108             jte.prepare();
 109         }).isInstanceOf(NullPointerException.class);
 110
 111         AxTask task = new AxTask(new AxArtifactKey("TestTask:0.0.1"));
 112         final ApexInternalContext internalContext = new ApexInternalContext(new AxPolicyModel());
 113
 114         jte.setContext(null, task, internalContext);
 115
 116         task.getTaskLogic().setLogic("return boolean;");
 117
 118         assertThatThrownBy(() -> {
 119             jte.prepare();
 120         }).hasMessage("logic failed to compile for TestTask:0.0.1 with message: invalid return (TestTask:0.0.1#1)");
 121
 122         task.getTaskLogic().setLogic("var x = 5;");
 123
 124         jte.prepare();
 125         Properties props = new Properties();
 126         assertThatThrownBy(() -> {
 127             jte.execute(-1, props, null);
 128         }).isInstanceOf(NullPointerException.class);
 129         jte.cleanUp();
 130         task.setInputEvent(new AxEvent());
 131         task.setOutputEvents(new TreeMap<>());
 132         task.getTaskLogic().setLogic("var returnValue = false;\nreturnValue;");
 133
 134         Map<String, Object> incomingParameters = new HashMap<>();
 135
 136         assertThatThrownBy(() -> {
 137             jte.prepare();
 138             jte.execute(-1, props, incomingParameters);
 139         }).hasMessage("execute-post: task logic execution failure on task \"TestTask\" in model NULL:0.0.0");
 140
 141         jte.cleanUp();
 142
 143         task.getTaskLogic().setLogic("var returnValue = true;\nreturnValue;");
 144
 145         jte.prepare();
 146         Map<String, Map<String, Object>> returnMap = jte.execute(0, new Properties(), incomingParameters);
 147         assertEquals(0, returnMap.size());
 148         jte.cleanUp();
 149     }
 150
 151     @Test
 152     public void testJavascriptTaskExecutorLogic() throws Exception {
 153
 154         JavascriptTaskExecutor jte = new JavascriptTaskExecutor();
 155         assertNotNull(jte);
 156
 157         assertThatThrownBy(() -> {
 158             jte.prepare();
 159         }).isInstanceOf(NullPointerException.class);
 160
 161         AxTask task = new AxTask(new AxArtifactKey("TestTask:0.0.1"));
 162
 163         ContextAlbum contextAlbum = createTestContextAlbum();
 164
 165         final ApexInternalContext internalContext = new ApexInternalContext(new AxPolicyModel());
 166         internalContext.getContextAlbums().put(contextAlbum.getKey(), contextAlbum);
 167
 168         task.getContextAlbumReferences().add(contextAlbum.getKey());
 169
 170         jte.setContext(null, task, internalContext);
 171         String parKey0 = "par0";
 172         Map<String, Object> incomingParameters = new HashMap<>();
 173         incomingParameters.put(parKey0, "value0");
 174
 175         AxEvent inEvent = new AxEvent();
 176         inEvent.setParameterMap(Map.of(parKey0, new AxField()));
 177         task.setInputEvent(inEvent);
 178         String parKey1 = "par1";
 179         AxEvent outEvent = new AxEvent();
 180         outEvent.setParameterMap(Map.of(parKey0, new AxField(), parKey1, new AxField()));
 181         final String eventName = "event1";
 182         task.setOutputEvents(Map.of(eventName, outEvent));
 183
 184         task.getTaskLogic().setLogic(TextFileUtils.getTextFileAsString("src/test/resources/javascript/TestLogic00.js"));
 185         jte.prepare();
 186         jte.execute(-1, new Properties(), incomingParameters);
 187         jte.cleanUp();
 188
 189         task.getTaskLogic().setLogic(TextFileUtils.getTextFileAsString("src/test/resources/javascript/TestLogic01.js"));
 190         jte.prepare();
 191
 192         Map<String, Map<String, Object>> outcomingParameters = jte.execute(-1, new Properties(), incomingParameters);
 193         jte.cleanUp();
 194
 195         assertEquals("returnVal0", outcomingParameters.get(eventName).get(parKey0));
 196         assertEquals("returnVal1", outcomingParameters.get(eventName).get(parKey1));
 197     }
 198
 199     private ContextAlbum createTestContextAlbum() throws ContextException {
 200         AxContextSchemas schemas = new AxContextSchemas();
 201         AxContextSchema simpleStringSchema =
 202                 new AxContextSchema(new AxArtifactKey("SimpleStringSchema", "0.0.1"), "JAVA", "java.lang.String");
 203         schemas.getSchemasMap().put(simpleStringSchema.getKey(), simpleStringSchema);
 204         ModelService.registerModel(AxContextSchemas.class, schemas);
 205
 206         AxContextAlbum axContextAlbum = new AxContextAlbum(new AxArtifactKey("TestContextAlbum", "0.0.1"), "Policy",
 207                 true, AxArtifactKey.getNullKey());
 208
 209         axContextAlbum.setItemSchema(simpleStringSchema.getKey());
 210         Distributor distributor = new JvmLocalDistributor();
 211         distributor.init(axContextAlbum.getKey());
 212         return new ContextAlbumImpl(axContextAlbum, distributor, new LinkedHashMap<String, Object>());
 213     }
 214 }