1 files changed, 84 insertions, 0 deletions
diff --git a/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/PolicyTypeBuilder.java b/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/PolicyTypeBuilder.java
new file mode 100644
index 0000000000..8c2b54f27e
--- /dev/null
+++ b/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/PolicyTypeBuilder.java
@@ -0,0 +1,84 @@
+package org.openecomp.sdc.be.components.utils;
+
+
+import org.openecomp.sdc.be.model.PolicyTypeDefinition;
+import org.openecomp.sdc.be.model.PropertyDefinition;
+
+import java.util.List;
+import java.util.Map;
+
+public class PolicyTypeBuilder {
+
+    private PolicyTypeDefinition policyTypeDataDefinition;
+
+    public PolicyTypeBuilder() {
+        this.policyTypeDataDefinition = new PolicyTypeDefinition();
+    }
+
+    public PolicyTypeBuilder setType(String type) {
+        policyTypeDataDefinition.setType(type);
+        return this;
+    }
+
+    public PolicyTypeBuilder setUniqueId(String uid) {
+        policyTypeDataDefinition.setUniqueId(uid);
+        return this;
+    }
+
+    public PolicyTypeBuilder setDerivedFrom(String derivedFrom) {
+        policyTypeDataDefinition.setDerivedFrom(derivedFrom);
+        return this;
+    }
+
+    public PolicyTypeBuilder setVersion(String version) {
+        policyTypeDataDefinition.setVersion(version);
+        return this;
+    }
+
+    public PolicyTypeBuilder setDescription(String version) {
+        policyTypeDataDefinition.setDescription(version);
+        return this;
+    }
+
+    public PolicyTypeBuilder setTargets(List<String> targets) {
+        policyTypeDataDefinition.setTargets(targets);
+        return this;
+    }
+
+    public PolicyTypeBuilder setMetadata(Map<String, String> metadata) {
+        policyTypeDataDefinition.setMetadata(metadata);
+        return this;
+    }
+
+    public PolicyTypeBuilder setModificationTime(long modificationTime) {
+        policyTypeDataDefinition.setModificationTime(modificationTime);
+        return this;
+    }
+
+    public PolicyTypeBuilder setCreationTime(long creationTime) {
+        policyTypeDataDefinition.setModificationTime(creationTime);
+        return this;
+    }
+
+    public PolicyTypeBuilder setOwner(String owner) {
+        policyTypeDataDefinition.setOwnerId(owner);
+        return this;
+    }
+
+    public PolicyTypeBuilder setHighestVersion(boolean isHighestVersion) {
+        policyTypeDataDefinition.setHighestVersion(isHighestVersion);
+        return this;
+    }
+
+    public PolicyTypeBuilder setProperties(List<PropertyDefinition> properties) {
+        policyTypeDataDefinition.setProperties(properties);
+        return this;
+    }
+
+    public PolicyTypeDefinition build() {
+        return policyTypeDataDefinition;
+    }
+
+
+
+}