gerrit.onap Code Review - sdc.git/blob

   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 import org.junit.Assert;
  25 import org.junit.Test;
  26 import org.onap.sdc.tosca.datatypes.model.CapabilityFilter;
  27 import org.onap.sdc.tosca.datatypes.model.Constraint;
  28 import org.onap.sdc.tosca.datatypes.model.NodeFilter;
  29 import org.onap.sdc.tosca.services.YamlUtil;
  30
  31 /**
  32  * @author KATYR
  33  * @since May 10, 2018
  34  */
  35
  36 public class NodeFilterTest {
  37
  38     private static final String GET_INPUT = "get_input";
  39
  40     @Test
  41     public void nodeFilterToYamlAndBack() {
  42         NodeFilter nodeFilter = new NodeFilter();
  43         nodeFilter.setProperties(createTestProperties());
  44         nodeFilter.setCapabilities(createTestCapabilities());
  45
  46         String yamlString = new YamlUtil().objectToYaml(nodeFilter);
  47
  48         NodeFilter nodeFilterFromYaml =
  49                 new YamlUtil().yamlToObject(yamlString, NodeFilter.class);
  50         Assert.assertNotNull(nodeFilterFromYaml);
  51     }
  52
  53
  54     private List<Map<String, List<Constraint>>> createTestProperties() {
  55         List<Map<String, List<Constraint>>> propertyFilterDefinitions = new ArrayList<>();
  56         Map<String, List<Constraint>> pfd1 = new HashMap<>();
  57         List<Constraint> constraints = new ArrayList<>();
  58         constraints.add(createConstraint("siteName", true));
  59         constraints.add(createConstraint("siteName1", false));
  60         pfd1.put("name1", constraints);
  61
  62
  63         Map<String, List<Constraint>> pfd2 = new HashMap<>();
  64         List<Constraint> constraints2 = new ArrayList<>();
  65         constraints2.add(createConstraint("input", true));
  66         constraints2.add(createConstraint("output", false));
  67         pfd2.put("name", constraints2);
  68
  69
  70         propertyFilterDefinitions.add(pfd1);
  71         propertyFilterDefinitions.add(pfd2);
  72
  73         return propertyFilterDefinitions;
  74
  75     }
  76
  77     private Constraint createConstraint(String constraintParam, boolean createEqualityConstraint) {
  78         Constraint constraint = new Constraint();
  79
  80         if (createEqualityConstraint) {
  81             constraint.setEqual(createValue(constraintParam));
  82         } else {
  83             constraint.setGreater_than(createValue(constraintParam));
  84
  85         }
  86         return constraint;
  87     }
  88
  89     private static Map<String, String> createValue(String propertyName) {
  90         Map<String, String> getPropertyMap = new HashMap<>();
  91         getPropertyMap.put(GET_INPUT, propertyName);
  92
  93         return getPropertyMap;
  94     }
  95
  96
  97
  98     private List<Map<String, CapabilityFilter>> createTestCapabilities() {
  99         List<Map<String, CapabilityFilter>> capabilities = new ArrayList<>();
 100         Map<String, CapabilityFilter> outerMap1 = new HashMap<>();
 101         Map<String, List<Constraint>> innerMap1 = new HashMap<>();
 102
 103         CapabilityFilter filterProperties1 = new CapabilityFilter();
 104         innerMap1.put("queue_model", createConstraints(3));
 105         List<Map<String, List<Constraint>>> innerList1 = new ArrayList<>();
 106         innerList1.add(innerMap1);
 107         filterProperties1.setProperties(innerList1);
 108         outerMap1.put("diffserv1", filterProperties1);
 109
 110         Map<String, CapabilityFilter> outerMap2 = new HashMap<>();
 111         CapabilityFilter filterProperties2 = new CapabilityFilter();
 112         Map<String, List<Constraint>> innerMap2 = new HashMap<>();
 113         Map<String, List<Constraint>> innerMap3 = new HashMap<>();
 114         innerMap2.put("queue_model", createConstraints(2));
 115         innerMap3.put("queue_model2", createConstraints(2));
 116         List<Map<String, List<Constraint>>> innerList2 = new ArrayList<>();
 117         innerList2.add(innerMap2);
 118         innerList2.add(innerMap3);
 119         filterProperties2.setProperties(innerList2);
 120         outerMap2.put("diffserv2", filterProperties2);
 121
 122         capabilities.add(outerMap1);
 123         capabilities.add(outerMap2);
 124
 125
 126         return capabilities;
 127     }
 128
 129     private List<Constraint> createConstraints(int length) {
 130         List<Constraint> constraints = new ArrayList<>();
 131         for (int i = 0; i < length; i++) {
 132             if (i / 2 == 0) {
 133                 constraints.add(createConstraint("siteName" + UUID.randomUUID().getMostSignificantBits(), true));
 134             } else {
 135                 constraints.add(createConstraint("siteName" + UUID.randomUUID().getLeastSignificantBits(), false));
 136             }
 137
 138         }
 139
 140         return constraints;
 141
 142     }
 143
 144 }