diff options
Diffstat (limited to 'catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/InputsBuilder.java')
-rw-r--r-- | catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/InputsBuilder.java | 48 |
1 files changed, 48 insertions, 0 deletions
diff --git a/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/InputsBuilder.java b/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/InputsBuilder.java new file mode 100644 index 0000000000..d2c2af35d7 --- /dev/null +++ b/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/InputsBuilder.java @@ -0,0 +1,48 @@ +package org.openecomp.sdc.be.components.utils; + +import org.openecomp.sdc.be.datatypes.elements.Annotation; +import org.openecomp.sdc.be.model.InputDefinition; + +import java.util.ArrayList; +import java.util.List; + +public class InputsBuilder { + + private InputDefinition input; + + private InputsBuilder() { + this.input = new InputDefinition(); + } + + public static InputsBuilder create() { + return new InputsBuilder(); + } + + public InputsBuilder setName(String name) { + input.setName(name); + return this; + } + + public InputsBuilder setPropertyId(String propertyId) { + input.setPropertyId(propertyId); + return this; + } + + public InputsBuilder addAnnotation(Annotation annotation) { + List<Annotation> annotations = getAnnotations(); + annotations.add(annotation); + return this; + } + + private List<Annotation> getAnnotations() { + if (input.getAnnotations() == null) { + input.setAnnotations(new ArrayList<>()); + } + return input.getAnnotations(); + } + + public InputDefinition build() { + return input; + } + +} |