Sync Integ to Master

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

diff --git a/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/RelationsBuilder.java b/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/RelationsBuilder.java
index 8c457f1..e0d35c6 100644 (file)
--- a/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/RelationsBuilder.java
+++ b/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/RelationsBuilder.java
@@ -1,12 +1,12 @@
 package org.openecomp.sdc.be.components.utils;
 
-import java.util.Collections;
-
 import org.openecomp.sdc.be.model.CapabilityRequirementRelationship;
 import org.openecomp.sdc.be.model.RelationshipImpl;
 import org.openecomp.sdc.be.model.RelationshipInfo;
 import org.openecomp.sdc.be.model.RequirementCapabilityRelDef;
 
+import java.util.Collections;
+
 public class RelationsBuilder {
 
     private RequirementCapabilityRelDef relation;
@@ -16,8 +16,8 @@ public class RelationsBuilder {
         RelationshipInfo requirementAndRelationshipPair = new RelationshipInfo();
         RelationshipImpl relationship = new RelationshipImpl();
         requirementAndRelationshipPair.setRelationships(relationship);
-               CapabilityRequirementRelationship capReqRel = new CapabilityRequirementRelationship();
-               capReqRel.setRelation(requirementAndRelationshipPair);
+        CapabilityRequirementRelationship capReqRel = new CapabilityRequirementRelationship();
+        capReqRel.setRelation(requirementAndRelationshipPair);
         relation.setRelationships(Collections.singletonList(capReqRel));
     }
 
@@ -27,17 +27,17 @@ public class RelationsBuilder {
     }
 
     public RelationsBuilder setRequirementName(String reqName) {
-        relation.getSingleRelationship().getRelation().setRequirement(reqName);
+        relation.resolveSingleRelationship().getRelation().setRequirement(reqName);
         return this;
     }
 
     public RelationsBuilder setRelationType(String type) {
-        relation.getSingleRelationship().getRelation().getRelationship().setType(type);
+        relation.resolveSingleRelationship().getRelation().getRelationship().setType(type);
         return this;
     }
 
     public RelationsBuilder setCapabilityUID(String uid) {
-        relation.getSingleRelationship().getRelation().setCapabilityUid(uid);
+        relation.resolveSingleRelationship().getRelation().setCapabilityUid(uid);
         return this;
     }