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 import static org.junit.Assert.assertTrue;
  26 import static org.junit.Assert.fail;
  27
  28 import com.google.gson.Gson;
  29 import com.google.gson.GsonBuilder;
  30
  31 import java.util.Base64;
  32
  33 import lombok.NonNull;
  34
  35 import org.junit.After;
  36 import org.junit.Before;
  37 import org.junit.Test;
  38 import org.onap.policy.common.utils.resources.ResourceUtils;
  39 import org.onap.policy.models.base.PfConceptKey;
  40 import org.onap.policy.models.base.PfModelException;
  41 import org.onap.policy.models.base.PfValidationResult;
  42 import org.onap.policy.models.provider.PolicyModelsProvider;
  43 import org.onap.policy.models.provider.PolicyModelsProviderFactory;
  44 import org.onap.policy.models.provider.PolicyModelsProviderParameters;
  45 import org.onap.policy.models.tosca.simple.concepts.ToscaPolicy;
  46 import org.onap.policy.models.tosca.simple.concepts.ToscaServiceTemplate;
  47 import org.onap.policy.models.tosca.simple.serialization.ToscaServiceTemplateMessageBodyHandler;
  48 import org.slf4j.Logger;
  49 import org.slf4j.LoggerFactory;
  50 import org.yaml.snakeyaml.Yaml;
  51
  52 /**
  53  * Test persistence of monitoring policies to and from the database.
  54  *
  55  * @author Liam Fallon (liam.fallon@est.tech)
  56  */
  57 public class PolicyToscaPersistenceTest {
  58     // Logger for this class
  59     private static final Logger LOGGER = LoggerFactory.getLogger(PolicyToscaPersistenceTest.class);
  60
  61     private Gson gson;
  62
  63     private PolicyModelsProvider databaseProvider;
  64
  65     // @formatter:off
  66     private String[] policyResourceNames = {
  67         "policies/vCPE.policy.monitoring.input.tosca.json",
  68         "policies/vCPE.policy.monitoring.input.tosca.yaml",
  69         "policies/vCPE.policy.operational.input.tosca.yaml",
  70         "policies/vDNS.policy.guard.frequency.input.tosca.json",
  71         "policies/vDNS.policy.guard.frequency.input.tosca.yaml",
  72         "policies/vDNS.policy.monitoring.input.tosca.json",
  73         "policies/vDNS.policy.monitoring.input.tosca.yaml",
  74         "policies/vDNS.policy.operational.input.tosca.yaml",
  75         "policies/vFirewall.policy.monitoring.input.tosca.json",
  76         "policies/vFirewall.policy.monitoring.input.tosca.yaml",
  77         "policies/vFirewall.policy.operational.input.tosca.json",
  78         "policies/vFirewall.policy.operational.input.tosca.yaml"
  79     };
  80     // @formatter:on
  81
  82     /**
  83      * Initialize provider.
  84      *
  85      * @throws PfModelException on exceptions in the tests
  86      */
  87     @Before
  88     public void setupParameters() throws PfModelException {
  89         PolicyModelsProviderParameters parameters = new PolicyModelsProviderParameters();
  90         parameters.setDatabaseUrl("jdbc:h2:mem:testdb");
  91         parameters.setDatabaseUser("policy");
  92         parameters.setDatabasePassword(Base64.getEncoder().encodeToString("P01icY".getBytes()));
  93         parameters.setPersistenceUnit("ToscaConceptTest");
  94
  95         databaseProvider = new PolicyModelsProviderFactory().createPolicyModelsProvider(parameters);
  96         databaseProvider.init();
  97     }
  98
  99     /**
 100      * Set up GSON.
 101      */
 102     @Before
 103     public void setupGson() {
 104         gson = new ToscaServiceTemplateMessageBodyHandler().getGson();
 105     }
 106
 107     @After
 108     public void teardown() throws Exception {
 109         databaseProvider.close();
 110     }
 111
 112     @Test
 113     public void testPolicyPersistence() {
 114         try {
 115             for (String policyResourceName : policyResourceNames) {
 116                 String policyString = ResourceUtils.getResourceAsString(policyResourceName);
 117
 118                 if (policyResourceName.endsWith("yaml")) {
 119                     testYamlStringPolicyPersistence(policyString);
 120                 } else {
 121                     testJsonStringPolicyPersistence(policyString);
 122                 }
 123             }
 124         } catch (Exception exc) {
 125             LOGGER.warn("error processing policies", exc);
 126             fail("test should not throw an exception");
 127         }
 128     }
 129
 130     private void testYamlStringPolicyPersistence(final String policyString) throws Exception {
 131         Object yamlObject = new Yaml().load(policyString);
 132         String yamlAsJsonString = new GsonBuilder().setPrettyPrinting().create().toJson(yamlObject);
 133
 134         testJsonStringPolicyPersistence(yamlAsJsonString);
 135     }
 136
 137     /**
 138      * Check persistence of a policy.
 139      *
 140      * @param policyString the policy as a string
 141      * @throws Exception any exception thrown
 142      */
 143     public void testJsonStringPolicyPersistence(@NonNull final String policyString) throws Exception {
 144         ToscaServiceTemplate serviceTemplate = gson.fromJson(policyString, ToscaServiceTemplate.class);
 145
 146         assertNotNull(serviceTemplate);
 147         LOGGER.info(serviceTemplate.validate(new PfValidationResult()).toString());
 148         assertTrue(serviceTemplate.validate(new PfValidationResult()).isValid());
 149
 150         databaseProvider.createPolicies(serviceTemplate);
 151
 152         for (PfConceptKey policyKey : serviceTemplate.getTopologyTemplate().getPolicies().getConceptMap().keySet()) {
 153             ToscaPolicy incomingPolicy = serviceTemplate.getTopologyTemplate().getPolicies().get(policyKey);
 154             ToscaPolicy databasePolicy =
 155                     databaseProvider.getPolicies(policyKey).getTopologyTemplate().getPolicies().get(policyKey);
 156             assertEquals(incomingPolicy, databasePolicy);
 157         }
 158     }
 159 }