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

   1 package org.openecomp.sdc.be.components.utils;
   2
   3 import org.openecomp.sdc.be.model.GroupDefinition;
   4
   5 import java.util.HashMap;
   6 import java.util.Map;
   7
   8 public class GroupDefinitionBuilder {
   9     private GroupDefinition groupDefinition;
  10
  11     private GroupDefinitionBuilder() {
  12         this.groupDefinition = new GroupDefinition();
  13     }
  14
  15     public static GroupDefinitionBuilder create() {
  16         return new GroupDefinitionBuilder();
  17     }
  18
  19     public GroupDefinitionBuilder setUniqueId(String uid) {
  20         groupDefinition.setUniqueId(uid);
  21         return this;
  22     }
  23
  24     public GroupDefinition build() {
  25         return groupDefinition;
  26     }
  27
  28     public GroupDefinitionBuilder addMember(String memberName, String memberId) {
  29         Map<String, String> members = groupDefinition.getMembers();
  30         if (members == null) {
  31             members = new HashMap<>();
  32             groupDefinition.setMembers(members);
  33         }
  34         members.put(memberId, memberId);
  35         return this;
  36     }
  37 }
  38