models-tosca/src/test/java/org/onap/policy/models/tosca/authorative/concepts/ToscaPolicyTypeFilterTest.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.tosca.authorative.concepts;
  22
  23 import static org.assertj.core.api.Assertions.assertThatThrownBy;
  24 import static org.junit.Assert.assertEquals;
  25 import static org.junit.Assert.assertNotNull;
  26 import static org.junit.Assert.assertTrue;
  27
  28 import com.google.gson.GsonBuilder;
  29
  30 import java.util.ArrayList;
  31 import java.util.List;
  32 import java.util.Map;
  33 import java.util.Map.Entry;
  34
  35 import org.junit.BeforeClass;
  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.PfKey;
  41 import org.slf4j.Logger;
  42 import org.slf4j.LoggerFactory;
  43 import org.yaml.snakeyaml.Yaml;
  44
  45 /**
  46  * Test of the {@link ToscaPolicyTypeFilter} class.
  47  *
  48  * @author Liam Fallon (liam.fallon@est.tech)
  49  */
  50 public class ToscaPolicyTypeFilterTest {
  51     // Logger for this class
  52     private static final Logger LOGGER = LoggerFactory.getLogger(ToscaPolicyTypeFilterTest.class);
  53
  54     // @formatter:off
  55     private static final String[] policyTypeResourceNames = {
  56         "policytypes/onap.policies.monitoring.dcaegen2.collectors.datafile.datafile-app-server.yaml",
  57         "policytypes/onap.policies.optimization.AffinityPolicy.yaml",
  58         "policytypes/onap.policies.optimization.DistancePolicy.yaml",
  59         "policytypes/onap.policies.optimization.HpaPolicy.yaml",
  60         "policytypes/onap.policies.optimization.OptimizationPolicy.yaml",
  61         "policytypes/onap.policies.optimization.PciPolicy.yaml",
  62         "policytypes/onap.policies.optimization.QueryPolicy.yaml",
  63         "policytypes/onap.policies.optimization.SubscriberPolicy.yaml",
  64         "policytypes/onap.policies.optimization.Vim_fit.yaml",
  65         "policytypes/onap.policies.optimization.VnfPolicy.yaml",
  66         "policytypes/onap.policy.monitoring.cdap.tca.hi.lo.app.yaml"
  67     };
  68     // @formatter:on
  69
  70     private static List<ToscaPolicyType> typeList = new ArrayList<>();
  71
  72     /**
  73      * Set up a Tosca Policy type list for filtering.
  74      *
  75      * @throws CoderException on JSON decoding errors
  76      */
  77     @BeforeClass
  78     public static void setupTypeList() throws CoderException {
  79         for (String policyTypeResourceName : policyTypeResourceNames) {
  80             String policyTypeString = ResourceUtils.getResourceAsString(policyTypeResourceName);
  81             Object yamlObject = new Yaml().load(policyTypeString);
  82             String yamlAsJsonString = new GsonBuilder().setPrettyPrinting().create().toJson(yamlObject);
  83
  84             ToscaServiceTemplate serviceTemplate =
  85                     new StandardCoder().decode(yamlAsJsonString, ToscaServiceTemplate.class);
  86             assertNotNull(serviceTemplate);
  87
  88             for (Map<String, ToscaPolicyType> foundPolicyTypeMap : serviceTemplate.getPolicyTypes()) {
  89                 for (Entry<String, ToscaPolicyType> policyTypeEntry : foundPolicyTypeMap.entrySet()) {
  90                     ToscaPolicyType policyType = policyTypeEntry.getValue();
  91                     if (policyType.getName() == null) {
  92                         policyType.setName(policyTypeEntry.getKey());
  93                     }
  94                     if (policyType.getVersion() == null) {
  95                         policyType.setVersion(PfKey.NULL_KEY_VERSION);
  96                     }
  97                     if (!typeList.contains(policyType)) {
  98                         typeList.add(policyType);
  99                     }
 100                 }
 101             }
 102         }
 103
 104         for (ToscaPolicyType type : typeList) {
 105             LOGGER.info("using policy type-" + type.getName() + ":" + type.getVersion());
 106         }
 107     }
 108
 109     @Test
 110     public void testNullList() {
 111         ToscaPolicyTypeFilter filter = ToscaPolicyTypeFilter.builder().build();
 112
 113         assertThatThrownBy(() -> {
 114             filter.filter(null);
 115         }).hasMessage("originalList is marked @NonNull but is null");
 116     }
 117
 118     @Test
 119     public void testFilterNothing() {
 120         ToscaPolicyTypeFilter filter = ToscaPolicyTypeFilter.builder().build();
 121
 122         List<ToscaPolicyType> filteredList = filter.filter(typeList);
 123         assertTrue(filteredList.containsAll(typeList));
 124     }
 125
 126     @Test
 127     public void testFilterLatestVersion() {
 128         ToscaPolicyTypeFilter filter =
 129                 ToscaPolicyTypeFilter.builder().version(ToscaPolicyTypeFilter.LATEST_VERSION).build();
 130
 131         List<ToscaPolicyType> filteredList = filter.filter(typeList);
 132         assertEquals(13, filteredList.size());
 133         assertEquals("1.0.0", filteredList.get(0).getVersion());
 134         assertEquals("0.0.0", filteredList.get(4).getVersion());
 135
 136         typeList.get(12).setVersion("2.0.0");
 137         filteredList = filter.filter(typeList);
 138         assertEquals(13, filteredList.size());
 139         assertEquals("2.0.0", filteredList.get(0).getVersion());
 140         assertEquals("0.0.0", filteredList.get(4).getVersion());
 141
 142         typeList.get(12).setVersion("1.0.0");
 143         filteredList = filter.filter(typeList);
 144         assertEquals(13, filteredList.size());
 145         assertEquals("1.0.0", filteredList.get(0).getVersion());
 146         assertEquals("0.0.0", filteredList.get(4).getVersion());
 147     }
 148
 149     @Test
 150     public void testFilterNameVersion() {
 151         ToscaPolicyTypeFilter filter = ToscaPolicyTypeFilter.builder().name("onap.policies.Monitoring").build();
 152         List<ToscaPolicyType> filteredList = filter.filter(typeList);
 153         assertEquals(2, filteredList.size());
 154
 155         filter = ToscaPolicyTypeFilter.builder().name("onap.policy.monitoring.cdap.tca.hi.lo.app").build();
 156         filteredList = filter.filter(typeList);
 157         assertEquals(1, filteredList.size());
 158
 159         filter = ToscaPolicyTypeFilter.builder().name("onap.policies.optimization.LpaPolicy").build();
 160         filteredList = filter.filter(typeList);
 161         assertEquals(0, filteredList.size());
 162
 163         filter = ToscaPolicyTypeFilter.builder().version("0.0.0").build();
 164         filteredList = filter.filter(typeList);
 165         assertEquals(9, filteredList.size());
 166
 167         filter = ToscaPolicyTypeFilter.builder().name("onap.policies.optimization.Vim_fit").version("0.0.0").build();
 168         filteredList = filter.filter(typeList);
 169         assertEquals(1, filteredList.size());
 170
 171         filter = ToscaPolicyTypeFilter.builder().name("onap.policies.optimization.Vim_fit").version("0.0.1").build();
 172         filteredList = filter.filter(typeList);
 173         assertEquals(0, filteredList.size());
 174     }
 175 }