catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/RelationsBuilder.java

   1 package org.openecomp.sdc.be.components.utils;
   2
   3 import org.openecomp.sdc.be.model.CapabilityRequirementRelationship;
   4 import org.openecomp.sdc.be.model.RelationshipImpl;
   5 import org.openecomp.sdc.be.model.RelationshipInfo;
   6 import org.openecomp.sdc.be.model.RequirementCapabilityRelDef;
   7
   8 import java.util.Collections;
   9
  10 public class RelationsBuilder {
  11
  12     private RequirementCapabilityRelDef relation;
  13
  14     public RelationsBuilder() {
  15         relation = new RequirementCapabilityRelDef();
  16         RelationshipInfo requirementAndRelationshipPair = new RelationshipInfo();
  17         RelationshipImpl relationship = new RelationshipImpl();
  18         requirementAndRelationshipPair.setRelationships(relationship);
  19         CapabilityRequirementRelationship capReqRel = new CapabilityRequirementRelationship();
  20         capReqRel.setRelation(requirementAndRelationshipPair);
  21         relation.setRelationships(Collections.singletonList(capReqRel));
  22     }
  23
  24     public RelationsBuilder setFromNode(String fromNode) {
  25         relation.setFromNode(fromNode);
  26         return this;
  27     }
  28
  29     public RelationsBuilder setRequirementName(String reqName) {
  30         relation.resolveSingleRelationship().getRelation().setRequirement(reqName);
  31         return this;
  32     }
  33
  34     public RelationsBuilder setRelationType(String type) {
  35         relation.resolveSingleRelationship().getRelation().getRelationship().setType(type);
  36         return this;
  37     }
  38
  39     public RelationsBuilder setCapabilityUID(String uid) {
  40         relation.resolveSingleRelationship().getRelation().setCapabilityUid(uid);
  41         return this;
  42     }
  43
  44     public RelationsBuilder setToNode(String toNode) {
  45         relation.setToNode(toNode);
  46         return this;
  47     }
  48
  49     public RequirementCapabilityRelDef build() {
  50         return relation;
  51     }
  52
  53 }