8b58c38d8d190458c67f6fbb0f864cb0d966b184
[policy/apex-pdp.git] /
1 /*-\r
2  * ============LICENSE_START=======================================================\r
3  *  Copyright (C) 2019 Nordix Foundation.\r
4  * ================================================================================\r
5  * Licensed under the Apache License, Version 2.0 (the "License");\r
6  * you may not use this file except in compliance with the License.\r
7  * You may obtain a copy of the License at\r
8  *\r
9  *      http://www.apache.org/licenses/LICENSE-2.0\r
10  *\r
11  * Unless required by applicable law or agreed to in writing, software\r
12  * distributed under the License is distributed on an "AS IS" BASIS,\r
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
14  * See the License for the specific language governing permissions and\r
15  * limitations under the License.\r
16  *\r
17  * SPDX-License-Identifier: Apache-2.0\r
18  * ============LICENSE_END=========================================================\r
19  */\r
20 \r
21 package org.onap.policy.apex.plugins.executor.jruby;\r
22 \r
23 import static org.junit.Assert.assertEquals;\r
24 import static org.junit.Assert.assertNotNull;\r
25 import static org.junit.Assert.fail;\r
26 \r
27 import java.lang.reflect.Field;\r
28 import java.util.HashMap;\r
29 import java.util.Map;\r
30 import java.util.Properties;\r
31 \r
32 import org.junit.After;\r
33 import org.junit.Before;\r
34 import org.junit.Test;\r
35 import org.onap.policy.apex.context.ContextException;\r
36 import org.onap.policy.apex.context.parameters.ContextParameterConstants;\r
37 import org.onap.policy.apex.context.parameters.DistributorParameters;\r
38 import org.onap.policy.apex.context.parameters.LockManagerParameters;\r
39 import org.onap.policy.apex.context.parameters.PersistorParameters;\r
40 import org.onap.policy.apex.core.engine.context.ApexInternalContext;\r
41 import org.onap.policy.apex.model.policymodel.concepts.AxPolicyModel;\r
42 import org.onap.policy.apex.model.policymodel.concepts.AxTask;\r
43 import org.onap.policy.common.parameters.ParameterService;\r
44 \r
45 /**\r
46  * Test the JrubyTaskExecutor class.\r
47  *\r
48  */\r
49 public class JrubyTaskExecutorTest {\r
50     /**\r
51      * Initiate Parameters.\r
52      */\r
53     @Before\r
54     public void initiateParameters() {\r
55         ParameterService.register(new DistributorParameters());\r
56         ParameterService.register(new LockManagerParameters());\r
57         ParameterService.register(new PersistorParameters());\r
58     }\r
59 \r
60     /**\r
61      * Clear Parameters.\r
62      */\r
63     @After\r
64     public void clearParameters() {\r
65         ParameterService.deregister(ContextParameterConstants.DISTRIBUTOR_GROUP_NAME);\r
66         ParameterService.deregister(ContextParameterConstants.LOCKING_GROUP_NAME);\r
67         ParameterService.deregister(ContextParameterConstants.PERSISTENCE_GROUP_NAME);\r
68     }\r
69 \r
70     @Test\r
71     public void testJrubyTaskExecutor() {\r
72         JrubyTaskExecutor jte = new JrubyTaskExecutor();\r
73         assertNotNull(jte);\r
74         try {\r
75             Field fieldContainer = JrubyTaskExecutor.class.getDeclaredField("container");\r
76             fieldContainer.setAccessible(true);\r
77             fieldContainer.set(jte, null);\r
78             jte.prepare();\r
79             fail("test should throw an exception here");\r
80         } catch (Exception jtseException) {\r
81             assertEquals(java.lang.NullPointerException.class, jtseException.getClass());\r
82         }\r
83 \r
84         AxTask task = new AxTask();\r
85         ApexInternalContext internalContext = null;\r
86         try {\r
87             internalContext = new ApexInternalContext(new AxPolicyModel());\r
88         } catch (ContextException e) {\r
89             fail("test should not throw an exception here");\r
90         }\r
91         jte.setContext(null, task, internalContext);\r
92         try {\r
93             jte.prepare();\r
94         } catch (Exception jtseException) {\r
95             fail("test should not throw an exception here");\r
96         }\r
97 \r
98         Map<String, Object> incomingParameters = new HashMap<>();\r
99         try {\r
100             jte.execute(-1, new Properties(), incomingParameters);\r
101             fail("test should throw an exception here");\r
102         } catch (Exception jteException) {\r
103             assertEquals("execute-post: task logic execution failure on task \"NULL\" in model NULL:0.0.0",\r
104                     jteException.getMessage());\r
105         }\r
106 \r
107         final String jrubyLogic =\r
108                 "if executor.executionId == -1" + "\n return false" + "\n else " + "\n return true" + "\n end";\r
109         task.getTaskLogic().setLogic(jrubyLogic);\r
110 \r
111         try {\r
112             jte.prepare();\r
113             Map<String, Object> returnMap = jte.execute(0, new Properties(), incomingParameters);\r
114             assertEquals(0, returnMap.size());\r
115             jte.cleanUp();\r
116         } catch (Exception jteException) {\r
117             fail("test should not throw an exception here");\r
118         }\r
119     }\r
120 }\r