diff options
Diffstat (limited to 'catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ComponentBuilder.java')
-rw-r--r-- | catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ComponentBuilder.java | 74 |
1 files changed, 57 insertions, 17 deletions
diff --git a/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ComponentBuilder.java b/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ComponentBuilder.java index f3cf208252..dd5849707a 100644 --- a/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ComponentBuilder.java +++ b/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ComponentBuilder.java @@ -1,21 +1,13 @@ package org.openecomp.sdc.be.components.utils; +import org.openecomp.sdc.be.dao.utils.MapUtil; +import org.openecomp.sdc.be.datatypes.enums.ComponentTypeEnum; +import org.openecomp.sdc.be.model.*; + import java.util.ArrayList; import java.util.HashMap; import java.util.List; -import org.openecomp.sdc.be.dao.utils.MapUtil; -import org.openecomp.sdc.be.datatypes.enums.ComponentTypeEnum; -import org.openecomp.sdc.be.model.Component; -import org.openecomp.sdc.be.model.ComponentInstance; -import org.openecomp.sdc.be.model.ComponentInstanceInput; -import org.openecomp.sdc.be.model.ComponentInstanceProperty; -import org.openecomp.sdc.be.model.GroupDefinition; -import org.openecomp.sdc.be.model.InputDefinition; -import org.openecomp.sdc.be.model.LifecycleStateEnum; -import org.openecomp.sdc.be.model.PolicyDefinition; -import org.openecomp.sdc.be.model.RequirementCapabilityRelDef; - public abstract class ComponentBuilder<T extends Component, B extends ComponentBuilder<T, B>> { private T component; @@ -67,12 +59,32 @@ public abstract class ComponentBuilder<T extends Component, B extends ComponentB return self(); } + public ComponentBuilder<T, B> addComponentInstance(String instanceName) { + ComponentInstance instance = new ComponentInstanceBuilder() + .setUniqueId(instanceName) + .setName(instanceName) + .build(); + return addComponentInstance(instance); + } + + public ComponentBuilder<T, B> addComponentInstance(String instanceName, String uniqueId) { + ComponentInstance instance = new ComponentInstanceBuilder() + .setUniqueId(uniqueId) + .setName(instanceName) + .build(); + return addComponentInstance(instance); + } + public ComponentBuilder<T, B> addComponentInstance(ComponentInstance componentInstance) { + initInstances(); + component.getComponentInstances().add(componentInstance); + return self(); + } + + private void initInstances() { if (component.getComponentInstances() == null) { component.setComponentInstances(new ArrayList<>()); } - component.getComponentInstances().add(componentInstance); - return self(); } public ComponentBuilder<T, B> addInput(InputDefinition input) { @@ -102,6 +114,7 @@ public abstract class ComponentBuilder<T extends Component, B extends ComponentB public ComponentBuilder<T, B> addInstanceProperty(String instanceId, String propName) { ComponentInstanceProperty componentInstanceProperty = new ComponentInstanceProperty(); componentInstanceProperty.setName(propName); + componentInstanceProperty.setUniqueId(propName); this.addInstanceProperty(instanceId, componentInstanceProperty); return self(); } @@ -117,6 +130,7 @@ public abstract class ComponentBuilder<T extends Component, B extends ComponentB public ComponentBuilder<T, B> addInstanceInput(String instanceId, String propName) { ComponentInstanceInput componentInstanceInput = new ComponentInstanceInput(); componentInstanceInput.setName(propName); + componentInstanceInput.setUniqueId(propName); this.addInstanceInput(instanceId, componentInstanceInput); return self(); } @@ -136,12 +150,33 @@ public abstract class ComponentBuilder<T extends Component, B extends ComponentB component.getPolicies().put(policyDefinition.getUniqueId(), policyDefinition); return self(); } - public ComponentBuilder<T, B> addGroups(GroupDefinition groupDefinition){ + + public ComponentBuilder<T, B> addPolicyProperty(String policyId, String propName){ + PolicyDefinition policyWithProp = PolicyDefinitionBuilder.create() + .addProperty(propName) + .setUniqueId(policyId) + .build(); + return addPolicy(policyWithProp); + } + + public ComponentBuilder<T, B> addGroup(GroupDefinition groupDefinition){ + initGroups(); + component.getGroups().add(groupDefinition); + return self(); + } + + public ComponentBuilder<T, B> addGroupProperty(String groupId, String propName){ + GroupDefinition groupWithProp = GroupDefinitionBuilder.create() + .addProperty(propName) + .setUniqueId(groupId) + .build(); + return addGroup(groupWithProp); + } + + private void initGroups() { if(component.getGroups() == null){ component.setGroups(new ArrayList<>()); } - component.getGroups().add(groupDefinition); - return self(); } public ComponentBuilder<T, B> setPolicies(List<PolicyDefinition> policies) { @@ -149,6 +184,11 @@ public abstract class ComponentBuilder<T extends Component, B extends ComponentB return self(); } + public ComponentBuilder<T, B> setGroups(List<GroupDefinition> groups) { + component.setGroups(groups); + return self(); + } + public T build() { return component; } |