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

   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.plugins.executor.jython;
  22
  23 import static org.junit.Assert.assertEquals;
  24 import static org.junit.Assert.assertNotNull;
  25 import static org.junit.Assert.fail;
  26
  27 import java.util.Map;
  28 import java.util.TreeMap;
  29 import org.junit.After;
  30 import org.junit.Before;
  31 import org.junit.Test;
  32 import org.onap.policy.apex.context.ContextException;
  33 import org.onap.policy.apex.context.parameters.ContextParameterConstants;
  34 import org.onap.policy.apex.context.parameters.DistributorParameters;
  35 import org.onap.policy.apex.context.parameters.LockManagerParameters;
  36 import org.onap.policy.apex.context.parameters.PersistorParameters;
  37 import org.onap.policy.apex.core.engine.EngineParameterConstants;
  38 import org.onap.policy.apex.core.engine.EngineParameters;
  39 import org.onap.policy.apex.core.engine.context.ApexInternalContext;
  40 import org.onap.policy.apex.core.engine.event.EnEvent;
  41 import org.onap.policy.apex.core.engine.executor.StateExecutor;
  42 import org.onap.policy.apex.core.engine.executor.exception.StateMachineException;
  43 import org.onap.policy.apex.core.engine.executor.impl.ExecutorFactoryImpl;
  44 import org.onap.policy.apex.model.basicmodel.concepts.AxArtifactKey;
  45 import org.onap.policy.apex.model.basicmodel.concepts.AxReferenceKey;
  46 import org.onap.policy.apex.model.eventmodel.concepts.AxEvent;
  47 import org.onap.policy.apex.model.policymodel.concepts.AxPolicyModel;
  48 import org.onap.policy.apex.model.policymodel.concepts.AxState;
  49 import org.onap.policy.apex.model.policymodel.concepts.AxStateFinalizerLogic;
  50 import org.onap.policy.apex.model.policymodel.concepts.AxStateOutput;
  51 import org.onap.policy.common.parameters.ParameterService;
  52
  53 /**
  54  * Test the JythonStateFinalizerExecutor class.
  55  *
  56  */
  57 public class JythonStateFinalizerExecutorTest {
  58     /**
  59      * Initiate Parameters.
  60      */
  61     @Before
  62     public void initiateParameters() {
  63         ParameterService.register(new DistributorParameters());
  64         ParameterService.register(new LockManagerParameters());
  65         ParameterService.register(new PersistorParameters());
  66         ParameterService.register(new EngineParameters());
  67     }
  68
  69     /**
  70      * Clear down Parameters.
  71      */
  72     @After
  73     public void clearParameters() {
  74         ParameterService.deregister(ContextParameterConstants.DISTRIBUTOR_GROUP_NAME);
  75         ParameterService.deregister(ContextParameterConstants.LOCKING_GROUP_NAME);
  76         ParameterService.deregister(ContextParameterConstants.PERSISTENCE_GROUP_NAME);
  77         ParameterService.deregister(EngineParameterConstants.MAIN_GROUP_NAME);
  78     }
  79
  80     @Test
  81     public void testJythonStateFinalizerExecutor() {
  82         JythonStateFinalizerExecutor jsfe = new JythonStateFinalizerExecutor();
  83         assertNotNull(jsfe);
  84
  85         try {
  86             jsfe.prepare();
  87             fail("test should throw an exception here");
  88         } catch (Exception jtseException) {
  89             assertEquals(java.lang.NullPointerException.class, jtseException.getClass());
  90         }
  91
  92         ApexInternalContext internalContext = null;
  93         try {
  94             internalContext = new ApexInternalContext(new AxPolicyModel());
  95         } catch (ContextException e) {
  96             fail("test should not throw an exception here");
  97         }
  98
  99         StateExecutor parentStateExcutor = null;
 100         try {
 101             parentStateExcutor = new StateExecutor(new ExecutorFactoryImpl());
 102         } catch (StateMachineException e) {
 103             fail("test should not throw an exception here");
 104         }
 105
 106         AxState state = new AxState();
 107         Map<String, AxStateOutput> stateOutputs = new TreeMap<>();
 108         AxArtifactKey triggerKey = new AxArtifactKey("TriggerName", "0.0.1");
 109         AxStateOutput isMe = new AxStateOutput(new AxReferenceKey(), triggerKey, new AxReferenceKey());
 110         stateOutputs.put("SelectedOutputIsMe", isMe);
 111         state.setStateOutputs(stateOutputs);
 112
 113         parentStateExcutor.setContext(null, state, internalContext);
 114         AxStateFinalizerLogic stateFinalizerLogic = new AxStateFinalizerLogic();
 115         jsfe.setContext(parentStateExcutor, stateFinalizerLogic, internalContext);
 116
 117         stateFinalizerLogic.setLogic("return false");
 118
 119         try {
 120             jsfe.prepare();
 121             fail("test should throw an exception here");
 122         } catch (Exception jtseException) {
 123             assertEquals(
 124                     "failed to compile Jython code for state finalizer AxReferenceKey:"
 125                             + "(parentKeyName=NULL,parentKeyVersion=0.0.0,parentLocalName=NULL,localName=NULL)",
 126                     jtseException.getMessage());
 127         }
 128
 129         String scriptSource = "for i in range(0,10): print(i)";
 130         stateFinalizerLogic.setLogic(scriptSource);
 131         try {
 132             jsfe.prepare();
 133             jsfe.execute(-1, null);
 134             fail("test should throw an exception here");
 135         } catch (Exception jtseException) {
 136             assertEquals(
 137                     "failed to execute Jython code for state finalizer AxReferenceKey:(parentKeyName=NULL,"
 138                             + "parentKeyVersion=0.0.0,parentLocalName=NULL,localName=NULL)",
 139                     jtseException.getMessage());
 140         }
 141
 142         scriptSource = "setattr(executor, 'selectedStateOutputName', 'SelectedOutputIsMe')\n"
 143                 + "returnValue=('' if executor == -1 else True)";
 144         stateFinalizerLogic.setLogic(scriptSource);
 145         try {
 146             jsfe.prepare();
 147         } catch (Exception jteException) {
 148             fail("test should not throw an exception here");
 149         }
 150
 151         AxEvent axEvent = new AxEvent(new AxArtifactKey("Event", "0.0.1"));
 152         EnEvent event = new EnEvent(axEvent);
 153         try {
 154             jsfe.execute(0, event);
 155         } catch (Exception jtseException) {
 156             jtseException.printStackTrace();
 157             fail("test should not throw an exception here");
 158         }
 159
 160         try {
 161             jsfe.prepare();
 162             String stateOutput = jsfe.execute(0, event);
 163             assertEquals("SelectedOutputIsMe", stateOutput);
 164             jsfe.cleanUp();
 165         } catch (Exception jtseException) {
 166             jtseException.printStackTrace();
 167             fail("test should not throw an exception here");
 168         }
 169     }
 170 }