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

   1 /*-\r
   2  * ============LICENSE_START=======================================================\r
   3  *  Copyright (C) 2019 Nordix Foundation.\r
   4  *  Modifications Copyright (C) 2019 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.junit.Assert.assertEquals;\r
  25 import static org.junit.Assert.assertNotNull;\r
  26 import static org.junit.Assert.fail;\r
  27 \r
  28 import java.lang.reflect.Field;\r
  29 import java.util.HashMap;\r
  30 import java.util.Map;\r
  31 import java.util.Properties;\r
  32 \r
  33 import org.junit.After;\r
  34 import org.junit.Before;\r
  35 import org.junit.Test;\r
  36 import org.onap.policy.apex.context.ContextException;\r
  37 import org.onap.policy.apex.context.parameters.ContextParameterConstants;\r
  38 import org.onap.policy.apex.context.parameters.DistributorParameters;\r
  39 import org.onap.policy.apex.context.parameters.LockManagerParameters;\r
  40 import org.onap.policy.apex.context.parameters.PersistorParameters;\r
  41 import org.onap.policy.apex.core.engine.context.ApexInternalContext;\r
  42 import org.onap.policy.apex.model.policymodel.concepts.AxPolicyModel;\r
  43 import org.onap.policy.apex.model.policymodel.concepts.AxTask;\r
  44 import org.onap.policy.common.parameters.ParameterService;\r
  45 \r
  46 /**\r
  47  * Test the JrubyTaskExecutor class.\r
  48  *\r
  49  */\r
  50 public class JrubyTaskExecutorTest {\r
  51     /**\r
  52      * Initiate Parameters.\r
  53      */\r
  54     @Before\r
  55     public void initiateParameters() {\r
  56         ParameterService.register(new DistributorParameters());\r
  57         ParameterService.register(new LockManagerParameters());\r
  58         ParameterService.register(new PersistorParameters());\r
  59     }\r
  60 \r
  61     /**\r
  62      * Clear Parameters.\r
  63      */\r
  64     @After\r
  65     public void clearParameters() {\r
  66         ParameterService.deregister(ContextParameterConstants.DISTRIBUTOR_GROUP_NAME);\r
  67         ParameterService.deregister(ContextParameterConstants.LOCKING_GROUP_NAME);\r
  68         ParameterService.deregister(ContextParameterConstants.PERSISTENCE_GROUP_NAME);\r
  69     }\r
  70 \r
  71     @Test\r
  72     public void testJrubyTaskExecutor() {\r
  73         // Run test twice to check for incorrect shutdown activity\r
  74         jrubyExecutorTest();\r
  75         jrubyExecutorTest();\r
  76     }\r
  77 \r
  78     /**\r
  79      * Test the JRuby executor.\r
  80      */\r
  81     private void jrubyExecutorTest() {\r
  82         JrubyTaskExecutor jte = new JrubyTaskExecutor();\r
  83         assertNotNull(jte);\r
  84 \r
  85         try {\r
  86             Field fieldContainer = JrubyTaskExecutor.class.getDeclaredField("container");\r
  87             fieldContainer.setAccessible(true);\r
  88             fieldContainer.set(jte, null);\r
  89             jte.prepare();\r
  90             fail("test should throw an exception here");\r
  91         } catch (Exception jtseException) {\r
  92             assertEquals(java.lang.NullPointerException.class, jtseException.getClass());\r
  93         }\r
  94 \r
  95         AxTask task = new AxTask();\r
  96         ApexInternalContext internalContext = null;\r
  97         try {\r
  98             internalContext = new ApexInternalContext(new AxPolicyModel());\r
  99         } catch (ContextException e) {\r
 100             fail("test should not throw an exception here");\r
 101         }\r
 102         jte.setContext(null, task, internalContext);\r
 103         try {\r
 104             jte.prepare();\r
 105         } catch (Exception jtseException) {\r
 106             fail("test should not throw an exception here");\r
 107         }\r
 108 \r
 109         Map<String, Object> incomingParameters = new HashMap<>();\r
 110         try {\r
 111             jte.execute(-1, new Properties(), incomingParameters);\r
 112             fail("test should throw an exception here");\r
 113         } catch (Exception jteException) {\r
 114             assertEquals("execute-post: task logic execution failure on task \"NULL\" in model NULL:0.0.0",\r
 115                             jteException.getMessage());\r
 116         }\r
 117 \r
 118         final String jrubyLogic = "if executor.executionId == -1" + "\n return false" + "\n else " + "\n return true"\r
 119                         + "\n end";\r
 120         task.getTaskLogic().setLogic(jrubyLogic);\r
 121 \r
 122         try {\r
 123             jte.prepare();\r
 124             Map<String, Object> returnMap = jte.execute(0, new Properties(), incomingParameters);\r
 125             assertEquals(0, returnMap.size());\r
 126             jte.cleanUp();\r
 127         } catch (Exception jteException) {\r
 128             fail("test should not throw an exception here");\r
 129         }\r
 130 \r
 131         try {\r
 132             jte.prepare();\r
 133             Map<String, Object> returnMap = jte.execute(0, new Properties(), incomingParameters);\r
 134             assertEquals(0, returnMap.size());\r
 135             jte.cleanUp();\r
 136         } catch (Exception jteException) {\r
 137             fail("test should not throw an exception here");\r
 138         }\r
 139     }\r
 140 }\r