models-provider/src/test/java/org/onap/policy/models/provider/impl/PolicyTypePersistenceTest.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.fail;
  26
  27 import com.google.gson.GsonBuilder;
  28
  29 import java.util.Base64;
  30 import java.util.List;
  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.StandardCoder;
  38 import org.onap.policy.common.utils.resources.ResourceUtils;
  39 import org.onap.policy.models.base.PfModelException;
  40 import org.onap.policy.models.provider.PolicyModelsProvider;
  41 import org.onap.policy.models.provider.PolicyModelsProviderFactory;
  42 import org.onap.policy.models.provider.PolicyModelsProviderParameters;
  43 import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicyType;
  44 import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicyTypeFilter;
  45 import org.onap.policy.models.tosca.authorative.concepts.ToscaServiceTemplate;
  46 import org.slf4j.Logger;
  47 import org.slf4j.LoggerFactory;
  48 import org.yaml.snakeyaml.Yaml;
  49
  50 /**
  51  * Test persistence of monitoring policies to and from the database.
  52  *
  53  * @author Liam Fallon (liam.fallon@est.tech)
  54  */
  55 public class PolicyTypePersistenceTest {
  56     // Logger for this class
  57     private static final Logger LOGGER = LoggerFactory.getLogger(PolicyTypePersistenceTest.class);
  58
  59     private StandardCoder standardCoder;
  60
  61     private PolicyModelsProvider databaseProvider;
  62
  63     // @formatter:off
  64     private String[] policyTypeResourceNames = {
  65         "policytypes/onap.policies.monitoring.dcaegen2.collectors.datafile.datafile-app-server.yaml",
  66         "policytypes/onap.policies.optimization.AffinityPolicy.yaml",
  67         "policytypes/onap.policies.optimization.DistancePolicy.yaml",
  68         "policytypes/onap.policies.optimization.HpaPolicy.yaml",
  69         "policytypes/onap.policies.optimization.OptimizationPolicy.yaml",
  70         "policytypes/onap.policies.optimization.PciPolicy.yaml",
  71         "policytypes/onap.policies.optimization.QueryPolicy.yaml",
  72         "policytypes/onap.policies.optimization.SubscriberPolicy.yaml",
  73         "policytypes/onap.policies.optimization.Vim_fit.yaml",
  74         "policytypes/onap.policies.optimization.VnfPolicy.yaml",
  75         "policytypes/onap.policy.monitoring.cdap.tca.hi.lo.app.yaml"
  76     };
  77     // @formatter:on
  78
  79     /**
  80      * Initialize provider.
  81      *
  82      * @throws PfModelException on exceptions in the tests
  83      */
  84     @Before
  85     public void setupParameters() throws PfModelException {
  86         PolicyModelsProviderParameters parameters = new PolicyModelsProviderParameters();
  87         parameters.setDatabaseDriver("org.h2.Driver");
  88         parameters.setDatabaseUrl("jdbc:h2:mem:testdb");
  89         parameters.setDatabaseUser("policy");
  90         parameters.setDatabasePassword(Base64.getEncoder().encodeToString("P01icY".getBytes()));
  91         parameters.setPersistenceUnit("ToscaConceptTest");
  92
  93         databaseProvider = new PolicyModelsProviderFactory().createPolicyModelsProvider(parameters);
  94     }
  95
  96     /**
  97      * Set up GSON.
  98      */
  99     @Before
 100     public void setupGson() {
 101         standardCoder = new StandardCoder();
 102     }
 103
 104     @After
 105     public void teardown() throws Exception {
 106         databaseProvider.close();
 107     }
 108
 109     @Test
 110     public void testPolicyTypePersistence() {
 111         try {
 112             for (String policyTypeResourceName : policyTypeResourceNames) {
 113                 String policyTypeString = ResourceUtils.getResourceAsString(policyTypeResourceName);
 114
 115                 if (policyTypeResourceName.endsWith("yaml")) {
 116                     testYamlStringPolicyTypePersistence(policyTypeString);
 117                 } else {
 118                     testJsonStringPolicyTypePersistence(policyTypeString);
 119                 }
 120             }
 121         } catch (Exception exc) {
 122             LOGGER.warn("error processing policy types", exc);
 123             fail("test should not throw an exception");
 124         }
 125     }
 126
 127     private void testYamlStringPolicyTypePersistence(final String policyTypeString) throws Exception {
 128         Object yamlObject = new Yaml().load(policyTypeString);
 129         String yamlAsJsonString = new GsonBuilder().setPrettyPrinting().create().toJson(yamlObject);
 130
 131         testJsonStringPolicyTypePersistence(yamlAsJsonString);
 132     }
 133
 134     /**
 135      * Check persistence of a policy.
 136      *
 137      * @param policyTypeString the policy as a string
 138      * @throws Exception any exception thrown
 139      */
 140     public void testJsonStringPolicyTypePersistence(@NonNull final String policyTypeString) throws Exception {
 141         ToscaServiceTemplate serviceTemplate = standardCoder.decode(policyTypeString, ToscaServiceTemplate.class);
 142
 143         assertNotNull(serviceTemplate);
 144         ToscaPolicyType inPolicyType = serviceTemplate.getPolicyTypes().get(0).values().iterator().next();
 145
 146         databaseProvider.createPolicyTypes(serviceTemplate);
 147         databaseProvider.updatePolicyTypes(serviceTemplate);
 148
 149         List<ToscaPolicyType> policyTypeList =
 150                 databaseProvider.getPolicyTypeList(inPolicyType.getName(), inPolicyType.getVersion());
 151
 152         policyTypeList = databaseProvider.getFilteredPolicyTypeList(ToscaPolicyTypeFilter.builder()
 153                 .name(inPolicyType.getName()).version(inPolicyType.getVersion()).build());
 154
 155         assertEquals(1, policyTypeList.size());
 156         assertEquals(inPolicyType.getName(), policyTypeList.get(0).getName());
 157
 158         policyTypeList = databaseProvider
 159                 .getFilteredPolicyTypeList(ToscaPolicyTypeFilter.builder().name(inPolicyType.getName()).build());
 160
 161         assertEquals(1, policyTypeList.size());
 162         assertEquals(inPolicyType.getName(), policyTypeList.get(0).getName());
 163
 164         policyTypeList = databaseProvider.getFilteredPolicyTypeList(ToscaPolicyTypeFilter.builder().build());
 165         assertEquals(2, policyTypeList.size());
 166         assertEquals(inPolicyType.getName(), policyTypeList.get(0).getName());
 167
 168         for (ToscaPolicyType policyType: databaseProvider.getPolicyTypeList(null, null)) {
 169             databaseProvider.deletePolicyType(policyType.getName(), policyType.getVersion());
 170         }
 171     }
 172 }