a27a74be4fa0b5b1f83210ac4c952cf670efdfba
[policy/clamp.git] /
1 /*-
2  * ============LICENSE_START=======================================================
3  *  Copyright (C) 2021-2023 Nordix Foundation.
4  * ================================================================================
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  *      http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  *
17  * SPDX-License-Identifier: Apache-2.0
18  * ============LICENSE_END=========================================================
19  */
20
21 package org.onap.policy.clamp.models.acm.persistence.provider;
22
23 import static org.assertj.core.api.Assertions.assertThat;
24 import static org.assertj.core.api.Assertions.assertThatThrownBy;
25 import static org.mockito.ArgumentMatchers.any;
26 import static org.mockito.Mockito.mock;
27 import static org.mockito.Mockito.verify;
28 import static org.mockito.Mockito.when;
29
30 import java.util.List;
31 import java.util.Map;
32 import java.util.Optional;
33 import java.util.UUID;
34 import org.junit.jupiter.api.BeforeAll;
35 import org.junit.jupiter.api.Test;
36 import org.mockito.Mockito;
37 import org.onap.policy.clamp.models.acm.concepts.AcTypeState;
38 import org.onap.policy.clamp.models.acm.concepts.AutomationCompositionDefinition;
39 import org.onap.policy.clamp.models.acm.concepts.NodeTemplateState;
40 import org.onap.policy.clamp.models.acm.document.concepts.DocToscaServiceTemplate;
41 import org.onap.policy.clamp.models.acm.persistence.concepts.JpaAutomationCompositionDefinition;
42 import org.onap.policy.clamp.models.acm.persistence.repository.AutomationCompositionDefinitionRepository;
43 import org.onap.policy.clamp.models.acm.utils.CommonTestData;
44 import org.onap.policy.models.tosca.authorative.concepts.ToscaConceptIdentifier;
45 import org.onap.policy.models.tosca.authorative.concepts.ToscaServiceTemplate;
46 import org.springframework.data.domain.Example;
47
48 class AcDefinitionProviderTest {
49
50     private static final String TOSCA_SERVICE_TEMPLATE_YAML = "clamp/acm/pmsh/funtional-pmsh-usecase.yaml";
51     private static final String TOSCA_SERVICE_TEMPLATE_YAML_PROP =
52             "clamp/acm/test/tosca-template-additional-properties.yaml";
53
54     private static ToscaServiceTemplate inputServiceTemplate;
55
56     @BeforeAll
57     static void loadServiceTemplate() {
58         inputServiceTemplate = CommonTestData.getToscaServiceTemplate(TOSCA_SERVICE_TEMPLATE_YAML);
59     }
60
61     @Test
62     void testDocCopyCompare() {
63
64         var inputServiceTemplateProperties = CommonTestData.getToscaServiceTemplate(TOSCA_SERVICE_TEMPLATE_YAML_PROP);
65         var docServiceTemplate = new DocToscaServiceTemplate(inputServiceTemplateProperties);
66         var docServiceTemplateCopy = new DocToscaServiceTemplate(docServiceTemplate);
67
68         assertThat(docServiceTemplate.compareTo(docServiceTemplateCopy)).isEqualByComparingTo(0);
69         assertThat(docServiceTemplate.compareToWithoutEntities(docServiceTemplateCopy)).isZero();
70
71         var acmDefinition = getAcDefinition(docServiceTemplate);
72         var acmDefinitionCopy = getAcDefinition(docServiceTemplateCopy);
73
74         assertThat(acmDefinition.getServiceTemplate().getName())
75                 .isEqualTo(acmDefinitionCopy.getServiceTemplate().getName());
76
77     }
78
79     @Test
80     void testCreateServiceTemplate() {
81         var docServiceTemplate = new DocToscaServiceTemplate(inputServiceTemplate);
82         var acmDefinition = getAcDefinition(docServiceTemplate);
83
84         var acmDefinitionRepository = mock(AutomationCompositionDefinitionRepository.class);
85         when(acmDefinitionRepository.save(any(JpaAutomationCompositionDefinition.class)))
86                 .thenReturn(new JpaAutomationCompositionDefinition(acmDefinition));
87
88         var acDefinitionProvider = new AcDefinitionProvider(acmDefinitionRepository);
89         var result = acDefinitionProvider.createAutomationCompositionDefinition(inputServiceTemplate);
90
91         assertThat(result.getServiceTemplate()).isEqualTo(docServiceTemplate.toAuthorative());
92     }
93
94     @Test
95     void testUpdateServiceTemplate() {
96         var acmDefinitionRepository = mock(AutomationCompositionDefinitionRepository.class);
97         var acDefinitionProvider = new AcDefinitionProvider(acmDefinitionRepository);
98         acDefinitionProvider.updateServiceTemplate(UUID.randomUUID(), inputServiceTemplate);
99         verify(acmDefinitionRepository).save(any(JpaAutomationCompositionDefinition.class));
100     }
101
102     @Test
103     void testUpdateAcDefinition() {
104         var acmDefinitionRepository = mock(AutomationCompositionDefinitionRepository.class);
105         var acDefinitionProvider = new AcDefinitionProvider(acmDefinitionRepository);
106         var acmDefinition = getAcDefinition(new DocToscaServiceTemplate(inputServiceTemplate));
107         acDefinitionProvider.updateAcDefinition(acmDefinition);
108         verify(acmDefinitionRepository).save(any(JpaAutomationCompositionDefinition.class));
109     }
110
111     @Test
112     void testGetAcDefinition() {
113         var jpa = new JpaAutomationCompositionDefinition();
114         jpa.fromAuthorative(getAcDefinition(new DocToscaServiceTemplate(inputServiceTemplate)));
115         var acmDefinitionRepository = mock(AutomationCompositionDefinitionRepository.class);
116         when(acmDefinitionRepository.findById(jpa.getCompositionId())).thenReturn(Optional.of(jpa));
117         var acDefinitionProvider = new AcDefinitionProvider(acmDefinitionRepository);
118         var result = acDefinitionProvider.getAcDefinition(UUID.fromString(jpa.getCompositionId()));
119         assertThat(result).isNotNull();
120     }
121
122     @Test
123     void testGetAcDefinitionNotFound() {
124         var acmDefinitionRepository = mock(AutomationCompositionDefinitionRepository.class);
125         var acDefinitionProvider = new AcDefinitionProvider(acmDefinitionRepository);
126         var compositionId = UUID.randomUUID();
127         assertThatThrownBy(() -> acDefinitionProvider.getAcDefinition(compositionId))
128                 .hasMessage("Get serviceTemplate \"" + compositionId + "\" failed, serviceTemplate does not exist");
129     }
130
131     @Test
132     void testFindAcDefinition() {
133         var jpa = new JpaAutomationCompositionDefinition();
134         jpa.fromAuthorative(getAcDefinition(new DocToscaServiceTemplate(inputServiceTemplate)));
135         var acmDefinitionRepository = mock(AutomationCompositionDefinitionRepository.class);
136         when(acmDefinitionRepository.findById(jpa.getCompositionId())).thenReturn(Optional.of(jpa));
137         var acDefinitionProvider = new AcDefinitionProvider(acmDefinitionRepository);
138         var result = acDefinitionProvider.findAcDefinition(UUID.fromString(jpa.getCompositionId()));
139         assertThat(result).isNotNull();
140     }
141
142     @Test
143     void testGetAllAcDefinitions() {
144         var jpa = new JpaAutomationCompositionDefinition();
145         jpa.fromAuthorative(getAcDefinition(new DocToscaServiceTemplate(inputServiceTemplate)));
146         var acmDefinitionRepository = mock(AutomationCompositionDefinitionRepository.class);
147         when(acmDefinitionRepository.findAll()).thenReturn(List.of(jpa));
148         var acDefinitionProvider = new AcDefinitionProvider(acmDefinitionRepository);
149         var result = acDefinitionProvider.getAllAcDefinitions();
150         assertThat(result).hasSize(1);
151     }
152
153     @Test
154     void testDeleteAcDefintion() {
155         var docServiceTemplate = new DocToscaServiceTemplate(inputServiceTemplate);
156         var acmDefinition = getAcDefinition(docServiceTemplate);
157
158         var acmDefinitionRepository = mock(AutomationCompositionDefinitionRepository.class);
159         when(acmDefinitionRepository.findById(acmDefinition.getCompositionId().toString()))
160                 .thenReturn(Optional.of(new JpaAutomationCompositionDefinition(acmDefinition)));
161
162         var acDefinitionProvider = new AcDefinitionProvider(acmDefinitionRepository);
163         var result = acDefinitionProvider.deleteAcDefintion(acmDefinition.getCompositionId());
164
165         assertThat(result).isEqualTo(docServiceTemplate.toAuthorative());
166     }
167
168     @Test
169     void testDeleteServiceTemplateEmpty() {
170         var compositionId = UUID.randomUUID();
171         var acmDefinitionRepository = mock(AutomationCompositionDefinitionRepository.class);
172         var acDefinitionProvider = new AcDefinitionProvider(acmDefinitionRepository);
173         assertThatThrownBy(() -> acDefinitionProvider.deleteAcDefintion(compositionId))
174                 .hasMessage("delete of Automation Composition Definition \"" + compositionId
175                         + "\" failed, Automation Composition Definition does not exist");
176     }
177
178     @Test
179     void testGetServiceTemplate() {
180         var docServiceTemplate = new DocToscaServiceTemplate(inputServiceTemplate);
181         var acmDefinition = getAcDefinition(docServiceTemplate);
182         var acmDefinitionRepository = mock(AutomationCompositionDefinitionRepository.class);
183         when(acmDefinitionRepository.findAll(Mockito.<Example<JpaAutomationCompositionDefinition>>any()))
184                 .thenReturn(List.of(new JpaAutomationCompositionDefinition(acmDefinition)));
185
186         var acDefinitionProvider = new AcDefinitionProvider(acmDefinitionRepository);
187         var result = acDefinitionProvider.getServiceTemplateList(inputServiceTemplate.getName(),
188                 inputServiceTemplate.getVersion());
189
190         assertThat(result).hasSize(1);
191         assertThat(result.get(0)).isEqualTo(acmDefinition.getServiceTemplate());
192     }
193
194     private AutomationCompositionDefinition getAcDefinition(DocToscaServiceTemplate docServiceTemplate) {
195         var acmDefinition = new AutomationCompositionDefinition();
196         acmDefinition.setCompositionId(UUID.randomUUID());
197         acmDefinition.setState(AcTypeState.COMMISSIONED);
198         acmDefinition.setServiceTemplate(docServiceTemplate.toAuthorative());
199         var nodeTemplateState = new NodeTemplateState();
200         nodeTemplateState.setNodeTemplateStateId(UUID.randomUUID());
201         nodeTemplateState.setNodeTemplateId(new ToscaConceptIdentifier("name", "1.0.0"));
202         nodeTemplateState.setState(AcTypeState.COMMISSIONED);
203         acmDefinition.setElementStateMap(Map.of(nodeTemplateState.getNodeTemplateId().getName(), nodeTemplateState));
204         return acmDefinition;
205     }
206 }