Catalog alignment
[sdc.git] / catalog-be / src / test / java / org / openecomp / sdc / be / tosca / GroupExportParserImplTest.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * SDC
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
10  * 
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  * 
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=========================================================
19  */
20
21 package org.openecomp.sdc.be.tosca;
22
23 import fj.data.Either;
24 import org.apache.commons.lang3.tuple.Pair;
25 import org.junit.Before;
26 import org.junit.Test;
27 import org.junit.runner.RunWith;
28 import org.mockito.Mock;
29 import org.mockito.junit.MockitoJUnitRunner;
30 import org.openecomp.sdc.be.components.impl.exceptions.SdcResourceNotFoundException;
31 import org.openecomp.sdc.be.components.utils.GroupDefinitionBuilder;
32 import org.openecomp.sdc.be.components.utils.ResourceBuilder;
33 import org.openecomp.sdc.be.model.CapabilityDefinition;
34 import org.openecomp.sdc.be.model.Component;
35 import org.openecomp.sdc.be.model.ComponentInstanceProperty;
36 import org.openecomp.sdc.be.model.GroupDefinition;
37 import org.openecomp.sdc.be.model.Resource;
38 import org.openecomp.sdc.be.model.cache.ApplicationDataTypeCache;
39 import org.openecomp.sdc.be.tosca.model.ToscaGroupTemplate;
40 import org.openecomp.sdc.be.tosca.model.ToscaTemplateCapability;
41
42 import java.util.ArrayList;
43 import java.util.HashMap;
44 import java.util.List;
45 import java.util.Map;
46 import java.util.stream.Stream;
47
48 import static java.util.Collections.singletonList;
49 import static org.assertj.core.api.Assertions.assertThat;
50 import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
51 import static org.mockito.Mockito.when;
52
53 @RunWith(MockitoJUnitRunner.class)
54 public class GroupExportParserImplTest {
55
56         private static final String GROUP_DEFINITION_NAME = "groupDefinitionName";
57         private static final String GROUP_DEFINITION_INVARIANT_UUID = "groupDefinitionInvariantUUID";
58         private static final String GROUP_DEFINITION_GROUP_UUID = "groupDefinitionGroupUUID";
59         private static final String GROUP_DEFINITION_VERSION = "groupDefinitionVersion";
60         private static final String GROUP_DEFINITION_TYPE = "groupDefinitionType";
61
62         private static final String CAPABILITIE_NAME = "capabilitieName";
63         private static final String COMPONENT_INSTANCE_PROPERTY_NAME = "componentInstancePropertyName";
64         private static final String COMPONENT_INSTANCE_PROPERTY_VALUE = "componentInstancePropertyValue";
65
66
67         private GroupExportParser groupExportParser;
68
69         @Mock
70         private ApplicationDataTypeCache dataTypeCache;
71
72         @Mock
73         private Component component;
74
75         @Before
76         public void setUp() throws Exception {
77                 initGroupExportParser();
78         }
79
80         private void initGroupExportParser() {
81                 when(dataTypeCache.getAll()).thenReturn(Either.left(null));
82                 groupExportParser = new GroupExportParserImpl(dataTypeCache);
83         }
84
85         @Test
86         public void failToGetAllDataTypes() {
87                 when(dataTypeCache.getAll()).thenReturn(Either.right(null));
88                 assertThatExceptionOfType(SdcResourceNotFoundException.class).isThrownBy(() -> groupExportParser = new GroupExportParserImpl(dataTypeCache));
89         }
90
91         @Test
92         public void noGroupsInComponent() {
93                 when(component.getGroups()).thenReturn(null);
94                 Map<String, ToscaGroupTemplate> groups = groupExportParser.getGroups(component);
95                 assertThat(groups).isNull();
96         }
97
98         @Test
99         public void oneGroupInComponent() {
100                 when(component.getGroups()).thenReturn(genOneGroup());
101                 Map<String, ToscaGroupTemplate> groups = groupExportParser.getGroups(component);
102                 assertThat(groups).isNotNull();
103                 assertThat(groups.size()).isOne();
104                 ToscaGroupTemplate toscaGroupTemplate = groups.get(GROUP_DEFINITION_NAME);
105                 Map<String, ToscaTemplateCapability> capabilities = toscaGroupTemplate.getCapabilities();
106                 ToscaTemplateCapability toscaTemplateCapability = capabilities.get(CAPABILITIE_NAME);
107                 Map<String, Object> properties = toscaTemplateCapability.getProperties();               
108                 assertThat(properties).containsEntry(COMPONENT_INSTANCE_PROPERTY_NAME, COMPONENT_INSTANCE_PROPERTY_VALUE);
109                 
110         }
111
112         @Test
113         public void whenExportingGroupMembers_takeNamesFromTheMembersAndNotFromTheGroupMembersMap() {
114                 GroupDefinition group1 = getGroup("group1", "type1", Pair.of("instance1Name", "inst1"));
115                 GroupDefinition group2 = getGroup("group2", "type2", Pair.of("instance2Name", "inst2"));
116                 Resource resource = new ResourceBuilder()
117                                 .addComponentInstance("inst1")
118                                 .addComponentInstance("inst2")
119                                 .addGroup(group1)
120                                 .addGroup(group2)
121                                 .build();
122                 Map<String, ToscaGroupTemplate> groups = groupExportParser.getGroups(resource);
123                 assertThat(groups.values())
124                                 .extracting("members")
125                                 .containsExactlyInAnyOrder(singletonList("inst1"), singletonList("inst2"));
126         }
127
128         @SafeVarargs
129         private final GroupDefinition getGroup(String name, String type, Pair<String, String>... members) {
130                 GroupDefinitionBuilder groupBuilder = GroupDefinitionBuilder.create()
131                                 .setName(name)
132                                 .setType(type);
133                 Stream.of(members).forEach(member -> groupBuilder.addMember(member.getKey(), member.getValue()));
134                 return groupBuilder.build();
135         }
136
137         private List<GroupDefinition> genOneGroup() {
138                 List<GroupDefinition> group = new ArrayList<>();
139                 GroupDefinitionBuilder groupBuilder = GroupDefinitionBuilder.create()
140                                 .setName(GROUP_DEFINITION_NAME) 
141                                 .setInvariantUUID(GROUP_DEFINITION_INVARIANT_UUID)
142                                 .setGroupUUID(GROUP_DEFINITION_GROUP_UUID)
143                                 .setVersion(GROUP_DEFINITION_VERSION)
144                                 .setType(GROUP_DEFINITION_TYPE);
145                 
146                 GroupDefinition groupDefinition = groupBuilder.build();
147
148                 Map<String, List<CapabilityDefinition>> capabilities = genCapabilities();
149                 groupDefinition.setCapabilities(capabilities);
150
151                 group.add(groupDefinition);
152                 return group;
153         }
154
155         private Map<String, List<CapabilityDefinition>> genCapabilities() {
156
157                 Map<String, List<CapabilityDefinition>> capabilities = new HashMap<>();
158                 List<CapabilityDefinition> capabilityDefinitions = new ArrayList<>();
159                 CapabilityDefinition capabilityDefinition = new CapabilityDefinition();
160
161                 List<ComponentInstanceProperty> capabilityProperties = new ArrayList<>();
162                 ComponentInstanceProperty componentInstanceProperty = new ComponentInstanceProperty();
163                 componentInstanceProperty.setName(COMPONENT_INSTANCE_PROPERTY_NAME);
164                 componentInstanceProperty.setValue(COMPONENT_INSTANCE_PROPERTY_VALUE);
165
166                 capabilityProperties.add(componentInstanceProperty);
167                 capabilityDefinition.setProperties(capabilityProperties);
168                 capabilityDefinition.setName(CAPABILITIE_NAME);
169
170                 capabilityDefinitions.add(capabilityDefinition);
171                 capabilities.put("JustSoneName", capabilityDefinitions);
172                 return capabilities;
173         }
174
175 }