1 files changed, 49 insertions, 0 deletions
diff --git a/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ComponentInstanceBuilder.java b/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ComponentInstanceBuilder.java
new file mode 100644
index 0000000000..a99f08c20d
--- /dev/null
+++ b/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ComponentInstanceBuilder.java
@@ -0,0 +1,49 @@
+package org.openecomp.sdc.be.components.utils;
+
+import java.util.HashMap;
+import java.util.Map;
+
+import org.openecomp.sdc.be.model.ArtifactDefinition;
+import org.openecomp.sdc.be.model.ComponentInstance;
+
+public class ComponentInstanceBuilder {
+
+    private ComponentInstance componentInstance;
+
+    public ComponentInstanceBuilder() {
+        componentInstance = new ComponentInstance();
+    }
+
+    public ComponentInstanceBuilder setName(String name) {
+        componentInstance.setName(name);
+        return this;
+    }
+
+    public ComponentInstanceBuilder setComponentUid(String componentUid) {
+        componentInstance.setComponentUid(componentUid);
+        return this;
+    }
+
+    public ComponentInstanceBuilder setId(String id) {
+        componentInstance.setUniqueId(id);
+        return this;
+    }
+
+    public ComponentInstanceBuilder setToscaName(String toscaName) {
+        componentInstance.setToscaComponentName(toscaName);
+        return this;
+    }
+
+    public ComponentInstanceBuilder addDeploymentArtifact(ArtifactDefinition artifactDefinition) {
+        Map<String, ArtifactDefinition> deploymentArtifacts = componentInstance.getDeploymentArtifacts();
+        if (deploymentArtifacts == null) {
+            componentInstance.setDeploymentArtifacts(new HashMap<>());
+        }
+        componentInstance.getDeploymentArtifacts().put(artifactDefinition.getArtifactName(), artifactDefinition);
+        return this;
+    }
+
+    public ComponentInstance build() {
+        return componentInstance;
+    }
+}