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

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * SDC
   4  * ================================================================================
   5  * Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.openecomp.sdc.be.components.utils;
  22
  23 import org.openecomp.sdc.be.model.CapabilityRequirementRelationship;
  24 import org.openecomp.sdc.be.model.RelationshipImpl;
  25 import org.openecomp.sdc.be.model.RelationshipInfo;
  26 import org.openecomp.sdc.be.model.RequirementCapabilityRelDef;
  27
  28 import java.util.Collections;
  29
  30 public class RelationsBuilder {
  31
  32     private RequirementCapabilityRelDef relation;
  33
  34     public RelationsBuilder() {
  35         relation = new RequirementCapabilityRelDef();
  36         RelationshipInfo requirementAndRelationshipPair = new RelationshipInfo();
  37         RelationshipImpl relationship = new RelationshipImpl();
  38         requirementAndRelationshipPair.setRelationships(relationship);
  39         CapabilityRequirementRelationship capReqRel = new CapabilityRequirementRelationship();
  40         capReqRel.setRelation(requirementAndRelationshipPair);
  41         relation.setRelationships(Collections.singletonList(capReqRel));
  42     }
  43
  44     public RelationsBuilder setFromNode(String fromNode) {
  45         relation.setFromNode(fromNode);
  46         return this;
  47     }
  48
  49     public RelationsBuilder setRequirementName(String reqName) {
  50         relation.resolveSingleRelationship().getRelation().setRequirement(reqName);
  51         return this;
  52     }
  53
  54     public RelationsBuilder setRelationType(String type) {
  55         relation.resolveSingleRelationship().getRelation().getRelationship().setType(type);
  56         return this;
  57     }
  58
  59     public RelationsBuilder setCapabilityUID(String uid) {
  60         relation.resolveSingleRelationship().getRelation().setCapabilityUid(uid);
  61         return this;
  62     }
  63
  64     public RelationsBuilder setToNode(String toNode) {
  65         relation.setToNode(toNode);
  66         return this;
  67     }
  68
  69     public RequirementCapabilityRelDef build() {
  70         return relation;
  71     }
  72
  73 }