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.model.PropertyDefinition;
   4 import org.openecomp.sdc.be.model.Resource;
   5
   6 import java.util.ArrayList;
   7
   8 public class ResourceBuilder extends ComponentBuilder<Resource, ResourceBuilder> {
   9
  10     private Resource resource;
  11
  12     @Override
  13     protected Resource component() {
  14         resource = new Resource();
  15         return resource;
  16     }
  17
  18     @Override
  19     protected ComponentBuilder<Resource, ResourceBuilder> self() {
  20         return this;
  21     }
  22
  23     ResourceBuilder addProperty(PropertyDefinition propertyDefinition) {
  24         if (resource.getProperties() == null) {
  25             resource.setProperties(new ArrayList<>());
  26         }
  27         resource.getProperties().add(propertyDefinition);
  28         return this;
  29     }
  30
  31
  32
  33     public ResourceBuilder() {
  34         super();
  35     }
  36
  37     public ResourceBuilder(Resource resource) {
  38         super(resource);
  39     }
  40
  41 }