openecomp-be/lib/openecomp-tosca-lib/src/test/java/org/openecomp/sdc/tosca/datatypes/model/NodeFilterTest.java

   1 /*
   2  * Copyright © 2016-2018 European Support Limited
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package org.openecomp.sdc.tosca.datatypes.model;
  18
  19 import java.util.ArrayList;
  20 import java.util.HashMap;
  21 import java.util.List;
  22 import java.util.Map;
  23 import java.util.UUID;
  24
  25 import org.junit.Assert;
  26 import org.junit.Test;
  27 import org.onap.sdc.tosca.datatypes.model.CapabilityFilter;
  28 import org.onap.sdc.tosca.datatypes.model.Constraint;
  29 import org.onap.sdc.tosca.datatypes.model.NodeFilter;
  30 import org.onap.sdc.tosca.services.YamlUtil;
  31
  32 /**
  33  * @author KATYR
  34  * @since May 10, 2018
  35  */
  36
  37 public class NodeFilterTest {
  38
  39     private static final String GET_INPUT = "get_input";
  40
  41     @Test
  42     public void nodeFilterToYamlAndBack() {
  43         NodeFilter nodeFilter = new NodeFilter();
  44         nodeFilter.setProperties(createTestProperties());
  45         nodeFilter.setCapabilities(createTestCapabilities());
  46
  47         String yamlString = new YamlUtil().objectToYaml(nodeFilter);
  48
  49         NodeFilter nodeFilterFromYaml =
  50                 new YamlUtil().yamlToObject(yamlString, NodeFilter.class);
  51         Assert.assertNotNull(nodeFilterFromYaml);
  52     }
  53
  54
  55     private List<Map<String, List<Constraint>>> createTestProperties() {
  56         List<Map<String, List<Constraint>>> propertyFilterDefinitions = new ArrayList<>();
  57         Map<String, List<Constraint>> pfd1 = new HashMap<>();
  58         List<Constraint> constraints = new ArrayList<>();
  59         constraints.add(createConstraint("siteName", true));
  60         constraints.add(createConstraint("siteName1", false));
  61         pfd1.put("name1", constraints);
  62
  63
  64         Map<String, List<Constraint>> pfd2 = new HashMap<>();
  65         List<Constraint> constraints2 = new ArrayList<>();
  66         constraints2.add(createConstraint("input", true));
  67         constraints2.add(createConstraint("output", false));
  68         pfd2.put("name", constraints2);
  69
  70
  71         propertyFilterDefinitions.add(pfd1);
  72         propertyFilterDefinitions.add(pfd2);
  73
  74         return propertyFilterDefinitions;
  75
  76     }
  77
  78     private Constraint createConstraint(String constraintParam, boolean createEqualityConstraint) {
  79         Constraint constraint = new Constraint();
  80
  81         if (createEqualityConstraint) {
  82             constraint.setEqual(createValue(constraintParam));
  83         } else {
  84             constraint.setGreater_than(createValue(constraintParam));
  85
  86         }
  87         return constraint;
  88     }
  89
  90     private static Map<String, String> createValue(String propertyName) {
  91         Map<String, String> getPropertyMap = new HashMap<>();
  92         getPropertyMap.put(GET_INPUT, propertyName);
  93
  94         return getPropertyMap;
  95     }
  96
  97
  98
  99     private List<Map<String, CapabilityFilter>> createTestCapabilities() {
 100         List<Map<String, CapabilityFilter>> capabilities = new ArrayList<>();
 101         Map<String, CapabilityFilter> outerMap1 = new HashMap<>();
 102         Map<String, List<Constraint>> innerMap1 = new HashMap<>();
 103
 104         CapabilityFilter filterProperties1 = new CapabilityFilter();
 105         innerMap1.put("queue_model", createConstraints(3));
 106         List<Map<String, List<Constraint>>> innerList1 = new ArrayList<>();
 107         innerList1.add(innerMap1);
 108         filterProperties1.setProperties(innerList1);
 109         outerMap1.put("diffserv1", filterProperties1);
 110
 111         Map<String, CapabilityFilter> outerMap2 = new HashMap<>();
 112         CapabilityFilter filterProperties2 = new CapabilityFilter();
 113         Map<String, List<Constraint>> innerMap2 = new HashMap<>();
 114         Map<String, List<Constraint>> innerMap3 = new HashMap<>();
 115         innerMap2.put("queue_model", createConstraints(2));
 116         innerMap3.put("queue_model2", createConstraints(2));
 117         List<Map<String, List<Constraint>>> innerList2 = new ArrayList<>();
 118         innerList2.add(innerMap2);
 119         innerList2.add(innerMap3);
 120         filterProperties2.setProperties(innerList2);
 121         outerMap2.put("diffserv2", filterProperties2);
 122
 123         capabilities.add(outerMap1);
 124         capabilities.add(outerMap2);
 125
 126
 127         return capabilities;
 128     }
 129
 130     private List<Constraint> createConstraints(int length) {
 131         List<Constraint> constraints = new ArrayList<>();
 132         for (int i = 0; i < length; i++) {
 133             if (i / 2 == 0) {
 134                 constraints.add(createConstraint("siteName" + UUID.randomUUID().getMostSignificantBits(), true));
 135             } else {
 136                 constraints.add(createConstraint("siteName" + UUID.randomUUID().getLeastSignificantBits(), false));
 137             }
 138
 139         }
 140
 141         return constraints;
 142
 143     }
 144
 145 }