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


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

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

import org.openecomp.sdc.be.datatypes.elements.PolicyTargetType;
import org.openecomp.sdc.be.datatypes.elements.PropertyDataDefinition;
import org.openecomp.sdc.be.model.PolicyDefinition;
import org.openecomp.sdc.be.model.PropertyDefinition;

import java.util.*;

public class PolicyDefinitionBuilder {

    private PolicyDefinition policyDefinition;

    private PolicyDefinitionBuilder() {
        this.policyDefinition = new PolicyDefinition();
    }

    public static PolicyDefinitionBuilder create() {
        return new PolicyDefinitionBuilder();
    }

    public PolicyDefinitionBuilder setName(String name) {
        policyDefinition.setName(name);
        return this;
    }

    public PolicyDefinitionBuilder setUniqueId(String uid) {
        policyDefinition.setUniqueId(uid);
        return this;
    }

    public PolicyDefinitionBuilder setProperties(PropertyDataDefinition ... props) {
        policyDefinition.setProperties(Arrays.asList(props));
        return this;
    }

    public PolicyDefinitionBuilder setTargets(Map<PolicyTargetType, List<String>> targets ){
        policyDefinition.setTargets(targets);
        return this;
    }

    public PolicyDefinitionBuilder addComponentInstanceTarget(String instId) {
        Map<PolicyTargetType, List<String>> targets = getTargets();
        targets.computeIfAbsent(PolicyTargetType.COMPONENT_INSTANCES, k -> new ArrayList<>())
               .add(instId);
        return this;
    }

    public PolicyDefinitionBuilder addGroupTarget(String groupId) {
        Map<PolicyTargetType, List<String>> targets = getTargets();
        targets.computeIfAbsent(PolicyTargetType.GROUPS, k -> new ArrayList<>()).add(groupId);
        return this;
    }

     public PolicyDefinitionBuilder addProperty(String propName) {
        initPolicyProperties();
        PropertyDefinition prop = new PropertyDataDefinitionBuilder()
                .setName(propName)
                .build();
        policyDefinition.getProperties().add(prop);
        return this;
    }

    public PolicyDefinition build() {
        return policyDefinition;
    }

    private Map<PolicyTargetType, List<String>> getTargets() {
        Map<PolicyTargetType, List<String>> targets = policyDefinition.getTargets();
        if (targets == null) {
            targets = new HashMap<>();
            policyDefinition.setTargets(targets);
        }
        return targets;
    }

    private void initPolicyProperties() {
        if (policyDefinition.getProperties() == null) {
            policyDefinition.setProperties(new ArrayList<>());
        }
    }

    public PolicyDefinitionBuilder setType(String type) {
        policyDefinition.setPolicyTypeName(type);
        return this;
    }
}