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

   1 package org.openecomp.sdc.be.components.utils;
   2
   3 import org.openecomp.sdc.be.datatypes.enums.ResourceTypeEnum;
   4 import org.openecomp.sdc.be.model.PropertyDefinition;
   5 import org.openecomp.sdc.be.model.Resource;
   6
   7 import java.util.ArrayList;
   8
   9 public class ResourceBuilder extends ComponentBuilder<Resource, ResourceBuilder> {
  10
  11     private Resource resource;
  12
  13     @Override
  14     protected Resource component() {
  15         resource = new Resource();
  16         return resource;
  17     }
  18
  19     @Override
  20     protected ComponentBuilder<Resource, ResourceBuilder> self() {
  21         return this;
  22     }
  23
  24     ResourceBuilder addProperty(PropertyDefinition propertyDefinition) {
  25         if (resource.getProperties() == null) {
  26             resource.setProperties(new ArrayList<>());
  27         }
  28         resource.getProperties().add(propertyDefinition);
  29         return this;
  30     }
  31
  32
  33     public ResourceBuilder setResourceType(ResourceTypeEnum resourceType) {
  34         resource.setResourceType(resourceType);
  35         return this;
  36     }
  37
  38
  39     public ResourceBuilder() {
  40         super();
  41     }
  42
  43     public ResourceBuilder(Resource resource) {
  44         super(resource);
  45     }
  46
  47 }