catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ObjectGenerator.java

   1 package org.openecomp.sdc.be.components.utils;
   2
   3 import org.openecomp.sdc.be.model.*;
   4
   5 import java.util.ArrayList;
   6 import java.util.Arrays;
   7 import java.util.List;
   8 import java.util.stream.Collectors;
   9 import java.util.stream.Stream;
  10
  11 public class ObjectGenerator {
  12
  13     public static Resource buildResourceWithInputs(String ... inputNames) {
  14         Resource resource = buildBasicResource();
  15         resource.setInputs(ObjectGenerator.buildInputs(inputNames));
  16         return resource;
  17     }
  18
  19     public static Resource buildResourceWithProperties(String ... propertiesNames) {
  20         ResourceBuilder resourceBuilder = new ResourceBuilder();
  21         resourceBuilder.setUniqueId("id");
  22         for (String propertyName : propertiesNames) {
  23             PropertyDefinition propertyDefinition = new PropertyDefinition();
  24             propertyDefinition.setName(propertyName);
  25             resourceBuilder.addProperty(propertyDefinition);
  26         }
  27         return resourceBuilder.build();
  28     }
  29
  30     public static List<ComponentInstanceProperty> buildInstanceProperties(String ... propertiesNames) {
  31         return Stream.of(propertiesNames).map(name ->  {
  32             ComponentInstanceProperty instProp = new ComponentInstanceProperty();
  33             instProp.setName(name);
  34             return instProp;
  35         }).collect(Collectors.toList());
  36     }
  37
  38     public static List<ComponentInstanceInput> buildInstanceInputs(String ... inputsNames) {
  39         return Stream.of(inputsNames).map(name ->  {
  40             ComponentInstanceInput instProp = new ComponentInstanceInput();
  41             instProp.setName(name);
  42             return instProp;
  43         }).collect(Collectors.toList());
  44     }
  45
  46     public static List<InputDefinition> buildInputs(String ... inputNames) {
  47         List<InputDefinition> inputs = new ArrayList<>();
  48         for (String inputName : inputNames) {
  49             InputDefinition inputDefinition = new InputDefinition();
  50             inputDefinition.setName(inputName);
  51             inputs.add(inputDefinition);
  52         }
  53         return inputs;
  54     }
  55
  56     public static Resource buildResourceWithComponentInstance(String ... instanceNames) {
  57         List<ComponentInstance> instances = new ArrayList<>();
  58         for (String instanceName : instanceNames) {
  59             ComponentInstance componentInstance = new ComponentInstanceBuilder().setName(instanceName).setComponentUid(instanceName).build();
  60             instances.add(componentInstance);
  61         }
  62         return buildResourceWithComponentInstances(instances);
  63     }
  64
  65     public static Resource buildResourceWithComponentInstances(List<ComponentInstance> instances) {
  66         Resource resource = buildBasicResource();
  67         resource.setComponentInstances(instances);
  68         return resource;
  69     }
  70
  71     public static Resource buildResourceWithComponentInstances(ComponentInstance ... instances) {
  72         return buildResourceWithComponentInstances(Arrays.asList(instances));
  73     }
  74
  75     public static Resource buildResourceWithRelationships(RequirementCapabilityRelDef ... relations) {
  76         Resource resource = buildBasicResource();
  77         ResourceBuilder resourceBuilder = new ResourceBuilder(resource);
  78         for (RequirementCapabilityRelDef relation : relations) {
  79             resourceBuilder.addRelationship(relation);
  80         }
  81         return resourceBuilder.build();
  82     }
  83
  84     public static Resource buildBasicResource() {
  85         Resource resource = new Resource();
  86         resource.setUniqueId("id");
  87         return resource;
  88     }
  89
  90     public static HeatParameterDefinition buildHeatParam(String defaultVal, String currValue) {
  91         return new HeatParameterBuilder().setDefaultValue(defaultVal).setCurrentValue(currValue).build();
  92     }
  93
  94 }