1 files changed, 42 insertions, 0 deletions
diff --git a/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ArtifactBuilder.java b/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ArtifactBuilder.java
new file mode 100644
index 0000000000..55a119e078
--- /dev/null
+++ b/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ArtifactBuilder.java
@@ -0,0 +1,42 @@
+package org.openecomp.sdc.be.components.utils;
+
+import java.util.ArrayList;
+
+import org.openecomp.sdc.be.model.ArtifactDefinition;
+import org.openecomp.sdc.be.model.HeatParameterDefinition;
+
+public class ArtifactBuilder {
+
+    private ArtifactDefinition artifactDefinition;
+
+    public ArtifactBuilder() {
+        this.artifactDefinition = new ArtifactDefinition();
+    }
+
+    public ArtifactBuilder setType(String type) {
+        this.artifactDefinition.setArtifactType(type);
+        return this;
+    }
+
+    public ArtifactBuilder setName(String name) {
+        this.artifactDefinition.setArtifactName(name);
+        return this;
+    }
+
+    public ArtifactBuilder setLabel(String label) {
+        this.artifactDefinition.setArtifactLabel(label);
+        return this;
+    }
+
+    public ArtifactBuilder addHeatParam(HeatParameterDefinition heatParam) {
+        if (this.artifactDefinition.getHeatParameters() == null) {
+            this.artifactDefinition.setHeatParameters(new ArrayList<>());
+        }
+        this.artifactDefinition.getHeatParameters().add(heatParam);
+        return this;
+    }
+
+    public ArtifactDefinition build() {
+        return artifactDefinition;
+    }
+}