2 * ============LICENSE_START=======================================================
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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
21 package org.openecomp.sdc.be.tosca;
23 import static org.assertj.core.api.Assertions.assertThat;
24 import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
25 import static org.junit.jupiter.api.Assertions.assertNull;
26 import static org.mockito.Mockito.when;
28 import fj.data.Either;
29 import java.util.ArrayList;
30 import java.util.Arrays;
31 import java.util.HashMap;
32 import java.util.List;
34 import org.junit.jupiter.api.Test;
35 import org.junit.jupiter.api.extension.ExtendWith;
36 import org.mockito.Mock;
37 import org.mockito.junit.jupiter.MockitoExtension;
38 import org.openecomp.sdc.be.components.impl.exceptions.SdcResourceNotFoundException;
39 import org.openecomp.sdc.be.datatypes.elements.PolicyTargetType;
40 import org.openecomp.sdc.be.model.Component;
41 import org.openecomp.sdc.be.model.ComponentInstance;
42 import org.openecomp.sdc.be.model.GroupDefinition;
43 import org.openecomp.sdc.be.model.PolicyDefinition;
44 import org.openecomp.sdc.be.model.cache.ApplicationDataTypeCache;
45 import org.openecomp.sdc.be.tosca.model.ToscaMetadata;
46 import org.openecomp.sdc.be.tosca.model.ToscaPolicyTemplate;
48 @ExtendWith(MockitoExtension.class)
49 class PolicyExportParserImplTest {
51 private static final String[] POLICY_KEYS = {"policy_key_1", "policy_key_2"};
52 private static final String[] VERSIONS = {"version_1", "version_1"};
53 private static final String[] POLICY_NAMES = {"name_1", "name_2"};
54 private static final String[] POLICY_UUIDS = {"policyUUID_1", "policyUUID_2"};
55 private static final String[] INVARIANT_UUIDS = {"invariantUUID_1", "invariantUUID_2"};
56 private static final String[] POLICY_TYPE_NAMES = {"policyTypeName_1", "policyTypeName_2"};
57 private static final String[] POLICY_COMPONENT_INSTANCES = {"policyComponentInstanceId"};
58 private static final String POLICY_COMPONENT_INSTANCES_NAME = "policyComponentInstanceName";
59 private static final String[] POLICY_GROUPS = {"policyGroupId"};
60 private static final String POLICY_GROUP_NAME = "PolicyGroupName";
62 private PolicyExportParser policiyExportParser;
65 private ApplicationDataTypeCache applicationDataTypeCache;
67 private PropertyConvertor propertyConvertor;
70 private Component component;
73 void failToGetAllDataTypes() {
75 when(applicationDataTypeCache.getAll(null)).thenReturn(Either.right(null));
76 assertThatExceptionOfType(SdcResourceNotFoundException.class).isThrownBy(() -> policiyExportParser = new PolicyExportParserImpl(
77 applicationDataTypeCache,
82 void noPoliciesInComponent() {
84 when(applicationDataTypeCache.getAll(null)).thenReturn(Either.left(null));
85 when(component.getPolicies()).thenReturn(null);
86 policiyExportParser = new PolicyExportParserImpl(applicationDataTypeCache, propertyConvertor);
87 Map<String, ToscaPolicyTemplate> policies = policiyExportParser.getPolicies(component);
92 void onePoliciesInComponent() {
94 List<Integer> constIndexes = Arrays.asList(new Integer[]{0});
95 testPoliciesInComponent(constIndexes);
99 void twoPoliciesInComponent() {
101 List<Integer> constIndexes = Arrays.asList(new Integer[]{0, 1});
102 testPoliciesInComponent(constIndexes);
105 private void testPoliciesInComponent(List<Integer> constIndexes) {
106 when(applicationDataTypeCache.getAll(null)).thenReturn(Either.left(null));
107 Map<String, PolicyDefinition> policiesToAdd = getPolicies(constIndexes);
109 when(component.getPolicies()).thenReturn(policiesToAdd);
110 when(component.getComponentInstances()).thenReturn(getComponentInstances());
111 when(component.getGroups()).thenReturn(getGroups());
112 policiyExportParser = new PolicyExportParserImpl(applicationDataTypeCache, propertyConvertor);
114 Map<String, ToscaPolicyTemplate> policies = policiyExportParser.getPolicies(component);
116 for (Integer i : constIndexes) {
118 ToscaPolicyTemplate toscaPolicyTemplate = policies.get(POLICY_NAMES[i]);
119 ToscaMetadata metadata = (ToscaMetadata) toscaPolicyTemplate.getMetadata();
121 assertThat(metadata.getInvariantUUID()).isEqualTo(INVARIANT_UUIDS[i]);
122 assertThat(metadata.getUUID()).isEqualTo(POLICY_UUIDS[i]);
123 assertThat(metadata.getName()).isEqualTo(POLICY_NAMES[i]);
124 assertThat(metadata.getVersion()).isEqualTo(VERSIONS[i]);
126 String type = toscaPolicyTemplate.getType();
127 assertThat(type).isEqualTo(POLICY_TYPE_NAMES[i]);
129 List<String> targets = toscaPolicyTemplate.getTargets();
130 assertThat(targets.get(0)).isEqualTo(POLICY_COMPONENT_INSTANCES_NAME);
131 assertThat(targets.get(1)).isEqualTo(POLICY_GROUP_NAME);
135 private List<GroupDefinition> getGroups() {
136 List<GroupDefinition> groups = new ArrayList<>();
137 GroupDefinition groupDefinition = new GroupDefinition();
138 groupDefinition.setUniqueId(POLICY_GROUPS[0]);
139 groupDefinition.setName(POLICY_GROUP_NAME);
140 groups.add(groupDefinition);
144 private List<ComponentInstance> getComponentInstances() {
145 List<ComponentInstance> componentInstances = new ArrayList<>();
146 ComponentInstance componentInstance = new ComponentInstance();
147 componentInstance.setUniqueId(POLICY_COMPONENT_INSTANCES[0]);
148 componentInstance.setName(POLICY_COMPONENT_INSTANCES_NAME);
149 componentInstances.add(componentInstance);
150 return componentInstances;
153 private Map<String, PolicyDefinition> getPolicies(List<Integer> indexes) {
154 Map<String, PolicyDefinition> policies = new HashMap<>();
156 for (int index : indexes) {
158 PolicyDefinition policyDefinition = new PolicyDefinition();
161 policyDefinition.setPolicyTypeName(POLICY_TYPE_NAMES[index]);
164 policyDefinition.setInvariantUUID(INVARIANT_UUIDS[index]);
165 policyDefinition.setPolicyUUID(POLICY_UUIDS[index]);
166 policyDefinition.setName(POLICY_NAMES[index]);
167 policyDefinition.setVersion(VERSIONS[index]);
170 policyDefinition.setTargets(getTargets());
172 policies.put(POLICY_KEYS[index], policyDefinition);
177 private Map<PolicyTargetType, List<String>> getTargets() {
178 Map<PolicyTargetType, List<String>> targets = new HashMap<>();
179 targets.put(PolicyTargetType.COMPONENT_INSTANCES, Arrays.asList(POLICY_COMPONENT_INSTANCES));
180 targets.put(PolicyTargetType.GROUPS, Arrays.asList(POLICY_GROUPS));