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

   1 package org.openecomp.sdc.be.components.utils;
   2
   3 import org.openecomp.sdc.be.model.CapabilityDefinition;
   4 import org.openecomp.sdc.be.model.ComponentInstanceProperty;
   5
   6 import java.util.ArrayList;
   7
   8 public class CapabilityDefinitionBuilder {
   9
  10     private CapabilityDefinition capabilityDefinition;
  11
  12     public CapabilityDefinitionBuilder() {
  13         capabilityDefinition = new CapabilityDefinition();
  14         capabilityDefinition.setProperties(new ArrayList<>());
  15     }
  16
  17     public CapabilityDefinitionBuilder addProperty(ComponentInstanceProperty property) {
  18         capabilityDefinition.getProperties().add(property);
  19         return this;
  20     }
  21
  22     public CapabilityDefinitionBuilder setOwnerId(String ownerId) {
  23         capabilityDefinition.setOwnerId(ownerId);
  24         return this;
  25     }
  26
  27     public CapabilityDefinitionBuilder setOwnerName(String ownerName) {
  28         capabilityDefinition.setOwnerName(ownerName);
  29         return this;
  30     }
  31
  32     public CapabilityDefinitionBuilder setType(String type) {
  33         capabilityDefinition.setType(type);
  34         return this;
  35     }
  36
  37     public CapabilityDefinitionBuilder setId(String ownerId) {
  38         capabilityDefinition.setUniqueId(ownerId);
  39         return this;
  40     }
  41
  42     public CapabilityDefinitionBuilder setName(String name) {
  43         capabilityDefinition.setName(name);
  44         return this;
  45     }
  46
  47
  48     public CapabilityDefinition build() {
  49         return capabilityDefinition;
  50     }
  51 }