summaryrefslogtreecommitdiffstats
path: root/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ResourceBuilder.java
diff options
context:
space:
mode:
Diffstat (limited to 'catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ResourceBuilder.java')
-rw-r--r--catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ResourceBuilder.java10
1 files changed, 8 insertions, 2 deletions
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 e5a14852e8..c6f2c5b430 100644
--- 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();