re base code

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

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
index d084bc2..b3d3dab 100644 (file)
--- 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
@@ -1,12 +1,12 @@
 package org.openecomp.sdc.be.components.utils;
 
 
-import java.util.List;
-import java.util.Map;
-
 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;
@@ -65,6 +65,16 @@ public class PolicyTypeBuilder {
         return this;
     }
 
+    public PolicyTypeBuilder setName(String name) {
+        policyTypeDataDefinition.setName(name);
+        return this;
+    }
+
+    public PolicyTypeBuilder setIcon(String icon) {
+        policyTypeDataDefinition.setIcon(icon);
+        return this;
+    }
+
     public PolicyTypeBuilder setHighestVersion(boolean isHighestVersion) {
         policyTypeDataDefinition.setHighestVersion(isHighestVersion);
         return this;