plugins/plugins-executor/plugins-executor-jruby/src/test/java/org/onap/policy/apex/plugins/executor/jruby/JrubyStateFinalizerExecutorTest.java

   1 /*-\r
   2  * ============LICENSE_START=======================================================\r
   3  *  Copyright (C) 2019 Nordix Foundation.\r
   4  *  Modifications Copyright (C) 2020 Nordix Foundation\r
   5  * ================================================================================\r
   6  * Licensed under the Apache License, Version 2.0 (the "License");\r
   7  * you may not use this file except in compliance with the License.\r
   8  * You may obtain a copy of the License at\r
   9  *\r
  10  *      http://www.apache.org/licenses/LICENSE-2.0\r
  11  *\r
  12  * Unless required by applicable law or agreed to in writing, software\r
  13  * distributed under the License is distributed on an "AS IS" BASIS,\r
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  15  * See the License for the specific language governing permissions and\r
  16  * limitations under the License.\r
  17  *\r
  18  * SPDX-License-Identifier: Apache-2.0\r
  19  * ============LICENSE_END=========================================================\r
  20  */\r
  21 \r
  22 package org.onap.policy.apex.plugins.executor.jruby;\r
  23 \r
  24 import static org.assertj.core.api.Assertions.assertThatThrownBy;\r
  25 import static org.junit.Assert.assertEquals;\r
  26 import static org.junit.Assert.assertNotNull;\r
  27 \r
  28 import java.lang.reflect.Field;\r
  29 import java.util.Properties;\r
  30 import org.junit.After;\r
  31 import org.junit.Before;\r
  32 import org.junit.Test;\r
  33 import org.onap.policy.apex.context.ContextException;\r
  34 import org.onap.policy.apex.context.parameters.ContextParameterConstants;\r
  35 import org.onap.policy.apex.context.parameters.DistributorParameters;\r
  36 import org.onap.policy.apex.context.parameters.LockManagerParameters;\r
  37 import org.onap.policy.apex.context.parameters.PersistorParameters;\r
  38 import org.onap.policy.apex.core.engine.EngineParameterConstants;\r
  39 import org.onap.policy.apex.core.engine.EngineParameters;\r
  40 import org.onap.policy.apex.core.engine.context.ApexInternalContext;\r
  41 import org.onap.policy.apex.core.engine.event.EnEvent;\r
  42 import org.onap.policy.apex.core.engine.executor.StateExecutor;\r
  43 import org.onap.policy.apex.core.engine.executor.exception.StateMachineException;\r
  44 import org.onap.policy.apex.core.engine.executor.impl.ExecutorFactoryImpl;\r
  45 import org.onap.policy.apex.model.basicmodel.concepts.AxArtifactKey;\r
  46 import org.onap.policy.apex.model.eventmodel.concepts.AxEvent;\r
  47 import org.onap.policy.apex.model.policymodel.concepts.AxPolicyModel;\r
  48 import org.onap.policy.apex.model.policymodel.concepts.AxState;\r
  49 import org.onap.policy.apex.model.policymodel.concepts.AxStateFinalizerLogic;\r
  50 import org.onap.policy.common.parameters.ParameterService;\r
  51 \r
  52 /**\r
  53  * Test the JrubyStateFinalizerExecutor class.\r
  54  *\r
  55  */\r
  56 public class JrubyStateFinalizerExecutorTest {\r
  57     /**\r
  58      * Initiate Parameters.\r
  59      */\r
  60     @Before\r
  61     public void initiateParameters() {\r
  62         ParameterService.register(new DistributorParameters());\r
  63         ParameterService.register(new LockManagerParameters());\r
  64         ParameterService.register(new PersistorParameters());\r
  65         ParameterService.register(new EngineParameters());\r
  66     }\r
  67 \r
  68     /**\r
  69      * Clear down Parameters.\r
  70      */\r
  71     @After\r
  72     public void clearParameters() {\r
  73         ParameterService.deregister(ContextParameterConstants.DISTRIBUTOR_GROUP_NAME);\r
  74         ParameterService.deregister(ContextParameterConstants.LOCKING_GROUP_NAME);\r
  75         ParameterService.deregister(ContextParameterConstants.PERSISTENCE_GROUP_NAME);\r
  76         ParameterService.deregister(EngineParameterConstants.MAIN_GROUP_NAME);\r
  77     }\r
  78 \r
  79     @Test\r
  80     public void testJrubyStateFinalizerExecutor() throws StateMachineException, ContextException,\r
  81         NoSuchFieldException, SecurityException, IllegalArgumentException, IllegalAccessException {\r
  82         JrubyStateFinalizerExecutor jsfe = new JrubyStateFinalizerExecutor();\r
  83         assertNotNull(jsfe);\r
  84 \r
  85         Field fieldContainer = JrubyStateFinalizerExecutor.class.getDeclaredField("container");\r
  86         fieldContainer.setAccessible(true);\r
  87         fieldContainer.set(jsfe, null);\r
  88         assertThatThrownBy(jsfe::prepare).isInstanceOf(java.lang.NullPointerException.class);\r
  89         ApexInternalContext internalContext = null;\r
  90 \r
  91         internalContext = new ApexInternalContext(new AxPolicyModel());\r
  92 \r
  93         StateExecutor parentStateExcutor = null;\r
  94         parentStateExcutor = new StateExecutor(new ExecutorFactoryImpl());\r
  95 \r
  96         AxState state = new AxState();\r
  97         parentStateExcutor.setContext(null, state, internalContext);\r
  98         AxStateFinalizerLogic stateFinalizerLogic = new AxStateFinalizerLogic();\r
  99         jsfe.setContext(parentStateExcutor, stateFinalizerLogic, internalContext);\r
 100 \r
 101         jsfe.prepare();\r
 102 \r
 103         assertThatThrownBy(() -> jsfe.execute(-1, new Properties(), null))\r
 104             .hasMessage("execute-post: state finalizer logic execution failure on state \""\r
 105                 + "NULL:0.0.0:NULL:NULL\" on finalizer logic NULL:0.0.0:NULL:NULL");\r
 106         AxEvent axEvent = new AxEvent(new AxArtifactKey("Event", "0.0.1"));\r
 107 \r
 108         final String jrubyLogic = "if executor.executionId == -1" + "\n return false" + "\n else "\r
 109                 + "\n executor.setSelectedStateOutputName(\"SelectedOutputIsMe\")" + "\n return true" + "\n end";\r
 110         stateFinalizerLogic.setLogic(jrubyLogic);\r
 111 \r
 112         EnEvent event = new EnEvent(axEvent);\r
 113         state.getStateOutputs().put("SelectedOutputIsMe", null);\r
 114         jsfe.prepare();\r
 115         String stateOutput = jsfe.execute(0, new Properties(), event);\r
 116         assertEquals("SelectedOutputIsMe", stateOutput);\r
 117         jsfe.cleanUp();\r
 118 \r
 119     }\r
 120 }\r