f4196811f67880eaef814ee15c3b53398b7dc4fd
[policy/models.git] / models-provider / src / test / java / org / onap / policy / models / provider / impl / PolicyToscaPersistenceTest.java
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.models.provider.impl;
22
23 import static org.junit.Assert.assertEquals;
24 import static org.junit.Assert.assertNotNull;
25
26 import java.net.URISyntaxException;
27 import java.util.Base64;
28 import java.util.List;
29 import java.util.Map;
30 import java.util.Set;
31
32 import lombok.NonNull;
33
34 import org.junit.After;
35 import org.junit.Before;
36 import org.junit.Test;
37 import org.onap.policy.common.utils.coder.CoderException;
38 import org.onap.policy.common.utils.coder.StandardCoder;
39 import org.onap.policy.common.utils.resources.ResourceUtils;
40 import org.onap.policy.models.base.PfModelException;
41 import org.onap.policy.models.provider.PolicyModelsProvider;
42 import org.onap.policy.models.provider.PolicyModelsProviderFactory;
43 import org.onap.policy.models.provider.PolicyModelsProviderParameters;
44 import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicy;
45 import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicyFilter;
46 import org.onap.policy.models.tosca.authorative.concepts.ToscaServiceTemplate;
47 import org.yaml.snakeyaml.Yaml;
48
49 /**
50  * Test persistence of monitoring policies to and from the database.
51  *
52  * @author Liam Fallon (liam.fallon@est.tech)
53  */
54 public class PolicyToscaPersistenceTest {
55     private StandardCoder standardCoder;
56
57     private PolicyModelsProvider databaseProvider;
58
59     /**
60      * Initialize provider.
61      *
62      * @throws PfModelException on exceptions in the tests
63      * @throws CoderException on JSON encoding and decoding errors
64      */
65     @Before
66     public void setupParameters() throws Exception {
67         // H2, use "org.mariadb.jdbc.Driver" and "jdbc:mariadb://localhost:3306/policy" for locally installed MariaDB
68
69         PolicyModelsProviderParameters parameters = new PolicyModelsProviderParameters();
70         parameters.setDatabaseDriver("org.h2.Driver");
71         parameters.setDatabaseUrl("jdbc:h2:mem:testdb");
72         parameters.setDatabaseUser("policy");
73         parameters.setDatabasePassword(Base64.getEncoder().encodeToString("P01icY".getBytes()));
74         parameters.setPersistenceUnit("ToscaConceptTest");
75
76         databaseProvider = new PolicyModelsProviderFactory().createPolicyModelsProvider(parameters);
77
78         createPolicyTypes();
79     }
80
81     /**
82      * Set up standard coder.
83      */
84     @Before
85     public void setupStandardCoder() {
86         standardCoder = new StandardCoder();
87     }
88
89     @After
90     public void teardown() throws Exception {
91         databaseProvider.close();
92     }
93
94     @Test
95     public void testPolicyPersistence() throws Exception {
96         Set<String> policyResources = ResourceUtils.getDirectoryContents("policies");
97
98         for (String policyResource : policyResources) {
99             if (!policyResource.contains("\\.tosca\\.")) {
100                 continue;
101             }
102
103             String policyString = ResourceUtils.getResourceAsString(policyResource);
104
105             if (policyResource.endsWith("yaml")) {
106                 testYamlStringPolicyPersistence(policyString);
107             } else {
108                 testJsonStringPolicyPersistence(policyString);
109             }
110         }
111     }
112
113     private void testYamlStringPolicyPersistence(final String policyString) throws Exception {
114         Object yamlObject = new Yaml().load(policyString);
115         String yamlAsJsonString = new StandardCoder().encode(yamlObject);
116
117         testJsonStringPolicyPersistence(yamlAsJsonString);
118     }
119
120     /**
121      * Check persistence of a policy.
122      *
123      * @param policyString the policy as a string
124      * @throws Exception any exception thrown
125      */
126     public void testJsonStringPolicyPersistence(@NonNull final String policyString) throws Exception {
127         ToscaServiceTemplate serviceTemplate = standardCoder.decode(policyString, ToscaServiceTemplate.class);
128
129         assertNotNull(serviceTemplate);
130
131         databaseProvider.createPolicies(serviceTemplate);
132         databaseProvider.updatePolicies(serviceTemplate);
133
134         for (Map<String, ToscaPolicy> policyMap : serviceTemplate.getToscaTopologyTemplate().getPolicies()) {
135             for (ToscaPolicy policy : policyMap.values()) {
136                 ToscaServiceTemplate gotToscaServiceTemplate =
137                         databaseProvider.getPolicies(policy.getName(), policy.getVersion());
138
139                 assertEquals(policy.getType(), gotToscaServiceTemplate.getToscaTopologyTemplate().getPolicies().get(0)
140                         .get(policy.getName()).getType());
141
142                 gotToscaServiceTemplate = databaseProvider.getFilteredPolicies(ToscaPolicyFilter.builder().build());
143
144                 assertEquals(policy.getType(),
145                         getToscaPolicyFromMapList(gotToscaServiceTemplate.getToscaTopologyTemplate().getPolicies(),
146                                 policy.getName()).getType());
147
148                 gotToscaServiceTemplate = databaseProvider.getFilteredPolicies(
149                         ToscaPolicyFilter.builder().name(policy.getName()).version(policy.getVersion()).build());
150
151                 assertEquals(policy.getType(), gotToscaServiceTemplate.getToscaTopologyTemplate().getPolicies().get(0)
152                         .get(policy.getName()).getType());
153             }
154         }
155     }
156
157     private ToscaPolicy getToscaPolicyFromMapList(List<Map<String, ToscaPolicy>> toscaPolicyMapList,
158             String policyName) {
159         ToscaPolicy toscaPolicy = new ToscaPolicy();
160         for (Map<String, ToscaPolicy> policyMap : toscaPolicyMapList) {
161             toscaPolicy = policyMap.get(policyName);
162             if (toscaPolicy != null) {
163                 break;
164             }
165         }
166         return toscaPolicy;
167     }
168
169     private void createPolicyTypes() throws CoderException, PfModelException, URISyntaxException {
170         Set<String> policyTypeResources = ResourceUtils.getDirectoryContents("policytypes");
171
172         for (String policyTypeResource : policyTypeResources) {
173             Object yamlObject = new Yaml().load(ResourceUtils.getResourceAsString(policyTypeResource));
174             String yamlAsJsonString = new StandardCoder().encode(yamlObject);
175
176             ToscaServiceTemplate toscaServiceTemplatePolicyType =
177                     standardCoder.decode(yamlAsJsonString, ToscaServiceTemplate.class);
178
179             assertNotNull(toscaServiceTemplatePolicyType);
180             databaseProvider.createPolicyTypes(toscaServiceTemplatePolicyType);
181         }
182     }
183 }