aboutsummaryrefslogtreecommitdiffstats
path: root/catalog-be/src/test/java/org/openecomp/sdc/be/components/utils/ObjectGenerator.java
blob: 2f546ba0f7f5df447bfcfd528229092bdbf67535 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
/*-
 * ============LICENSE_START=======================================================
 * SDC
 * ================================================================================
 * Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
 * ================================================================================
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * ============LICENSE_END=========================================================
 */

package org.openecomp.sdc.be.components.utils;

import org.openecomp.sdc.be.model.*;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class ObjectGenerator {

    public static Resource buildResourceWithInputs(String ... inputNames) {
        Resource resource = buildBasicResource();
        resource.setInputs(ObjectGenerator.buildInputs(inputNames));
        return resource;
    }

    public static Resource buildResourceWithProperties(String ... propertiesNames) {
        ResourceBuilder resourceBuilder = new ResourceBuilder();
        resourceBuilder.setUniqueId("id");
        for (String propertyName : propertiesNames) {
            PropertyDefinition propertyDefinition = new PropertyDefinition();
            propertyDefinition.setName(propertyName);
            resourceBuilder.addProperty(propertyDefinition);
        }
        return resourceBuilder.build();
    }

    public static List<ComponentInstanceProperty> buildInstanceProperties(String ... propertiesNames) {
        return Stream.of(propertiesNames).map(name ->  {
            ComponentInstanceProperty instProp = new ComponentInstanceProperty();
            instProp.setName(name);
            return instProp;
        }).collect(Collectors.toList());
    }

    public static List<ComponentInstanceInput> buildInstanceInputs(String ... inputsNames) {
        return Stream.of(inputsNames).map(name ->  {
            ComponentInstanceInput instProp = new ComponentInstanceInput();
            instProp.setName(name);
            return instProp;
        }).collect(Collectors.toList());
    }

    public static List<InputDefinition> buildInputs(String ... inputNames) {
        List<InputDefinition> inputs = new ArrayList<>();
        for (String inputName : inputNames) {
            InputDefinition inputDefinition = new InputDefinition();
            inputDefinition.setName(inputName);
            inputs.add(inputDefinition);
        }
        return inputs;
    }

    public static Resource buildResourceWithComponentInstance(String ... instanceNames) {
        List<ComponentInstance> instances = new ArrayList<>();
        for (String instanceName : instanceNames) {
            ComponentInstance componentInstance = new ComponentInstanceBuilder().setName(instanceName).setComponentUid(instanceName).build();
            instances.add(componentInstance);
        }
        return buildResourceWithComponentInstances(instances);
    }

    public static Resource buildResourceWithComponentInstances(List<ComponentInstance> instances) {
        Resource resource = buildBasicResource();
        resource.setComponentInstances(instances);
        return resource;
    }

    public static Resource buildResourceWithComponentInstances(ComponentInstance ... instances) {
        return buildResourceWithComponentInstances(Arrays.asList(instances));
    }

    public static Resource buildResourceWithRelationships(RequirementCapabilityRelDef ... relations) {
        Resource resource = buildBasicResource();
        ResourceBuilder resourceBuilder = new ResourceBuilder(resource);
        for (RequirementCapabilityRelDef relation : relations) {
            resourceBuilder.addRelationship(relation);
        }
        return resourceBuilder.build();
    }

    public static Resource buildBasicResource() {
        Resource resource = new Resource();
        resource.setUniqueId("id");
        return resource;
    }

    public static HeatParameterDefinition buildHeatParam(String defaultVal, String currValue) {
        return new HeatParameterBuilder().setDefaultValue(defaultVal).setCurrentValue(currValue).build();
    }

}