diff options
Diffstat (limited to 'catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/PolicyDefinitionBuilder.java')
-rw-r--r-- | catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/PolicyDefinitionBuilder.java | 29 |
1 files changed, 23 insertions, 6 deletions
diff --git a/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/PolicyDefinitionBuilder.java b/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/PolicyDefinitionBuilder.java index 2826e2044e..f16486f776 100644 --- a/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/PolicyDefinitionBuilder.java +++ b/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/PolicyDefinitionBuilder.java @@ -1,14 +1,11 @@ package org.openecomp.sdc.be.components.utils; -import java.util.ArrayList; -import java.util.Arrays; -import java.util.HashMap; -import java.util.List; -import java.util.Map; - import org.openecomp.sdc.be.datatypes.elements.PolicyTargetType; import org.openecomp.sdc.be.datatypes.elements.PropertyDataDefinition; import org.openecomp.sdc.be.model.PolicyDefinition; +import org.openecomp.sdc.be.model.PropertyDefinition; + +import java.util.*; public class PolicyDefinitionBuilder { @@ -55,6 +52,15 @@ public class PolicyDefinitionBuilder { return this; } + public PolicyDefinitionBuilder addProperty(String propName) { + initPolicyProperties(); + PropertyDefinition prop = new PropertyDataDefinitionBuilder() + .setName(propName) + .build(); + policyDefinition.getProperties().add(prop); + return this; + } + public PolicyDefinition build() { return policyDefinition; } @@ -67,4 +73,15 @@ public class PolicyDefinitionBuilder { } return targets; } + + private void initPolicyProperties() { + if (policyDefinition.getProperties() == null) { + policyDefinition.setProperties(new ArrayList<>()); + } + } + + public PolicyDefinitionBuilder setType(String type) { + policyDefinition.setPolicyTypeName(type); + return this; + } } |