re base code

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

diff --git a/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ResourceBuilder.java b/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ResourceBuilder.java
index e5a1485..c6f2c5b 100644 (file)
--- a/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ResourceBuilder.java
+++ b/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ResourceBuilder.java
@@ -1,10 +1,11 @@
 package org.openecomp.sdc.be.components.utils;
 
-import java.util.ArrayList;
-
+import org.openecomp.sdc.be.datatypes.enums.ResourceTypeEnum;
 import org.openecomp.sdc.be.model.PropertyDefinition;
 import org.openecomp.sdc.be.model.Resource;
 
+import java.util.ArrayList;
+
 public class ResourceBuilder extends ComponentBuilder<Resource, ResourceBuilder> {
 
     private Resource resource;
@@ -29,6 +30,11 @@ public class ResourceBuilder extends ComponentBuilder<Resource, ResourceBuilder>
     }
 
 
+    public ResourceBuilder setResourceType(ResourceTypeEnum resourceType) {
+        resource.setResourceType(resourceType);
+        return this;
+    }
+
 
     public ResourceBuilder() {
         super();