b09ddda74534b28ab4e573a23081ab6d62fcb9dc
[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.junit.jupiter.api.Assertions.assertDoesNotThrow;
26 import static org.junit.jupiter.api.Assertions.assertEquals;
27 import static org.mockito.ArgumentMatchers.any;
28 import static org.mockito.ArgumentMatchers.anyIterable;
29 import static org.mockito.ArgumentMatchers.anyList;
30 import static org.mockito.Mockito.mock;
31 import static org.mockito.Mockito.verify;
32 import static org.mockito.Mockito.when;
33
34 import java.util.List;
35 import java.util.Optional;
36 import java.util.UUID;
37 import org.junit.jupiter.api.BeforeEach;
38 import org.junit.jupiter.api.Test;
39 import org.mockito.Mockito;
40 import org.onap.policy.clamp.models.acm.concepts.AutomationCompositionElementInfo;
41 import org.onap.policy.clamp.models.acm.concepts.AutomationCompositionInfo;
42 import org.onap.policy.clamp.models.acm.concepts.AutomationCompositions;
43 import org.onap.policy.clamp.models.acm.persistence.concepts.JpaAutomationComposition;
44 import org.onap.policy.clamp.models.acm.persistence.concepts.JpaAutomationCompositionElement;
45 import org.onap.policy.clamp.models.acm.persistence.repository.AutomationCompositionElementRepository;
46 import org.onap.policy.clamp.models.acm.persistence.repository.AutomationCompositionRepository;
47 import org.onap.policy.common.utils.coder.Coder;
48 import org.onap.policy.common.utils.coder.StandardCoder;
49 import org.onap.policy.common.utils.resources.ResourceUtils;
50 import org.springframework.data.domain.Example;
51
52 class AutomationCompositionProviderTest {
53
54     private static final String AC_IS_NULL = "automationComposition is marked non-null but is null";
55     private static final String ACELEMENT_IS_NULL = "element is marked non-null but is null";
56
57     private static final Coder CODER = new StandardCoder();
58     private static final String AUTOMATION_COMPOSITION_JSON =
59             "src/test/resources/providers/TestAutomationCompositions.json";
60
61     private AutomationCompositions inputAutomationCompositions;
62     private List<JpaAutomationComposition> inputAutomationCompositionsJpa;
63     private final String originalJson = ResourceUtils.getResourceAsString(AUTOMATION_COMPOSITION_JSON);
64
65     @BeforeEach
66     void beforeSetupDao() throws Exception {
67         inputAutomationCompositions = CODER.decode(originalJson, AutomationCompositions.class);
68         inputAutomationCompositionsJpa =
69                 ProviderUtils.getJpaAndValidateList(inputAutomationCompositions.getAutomationCompositionList(),
70                         JpaAutomationComposition::new, "automation compositions");
71     }
72
73     @Test
74     void testAutomationCompositionCreate() {
75         var automationCompositionRepository = mock(AutomationCompositionRepository.class);
76         var automationCompositionProvider = new AutomationCompositionProvider(automationCompositionRepository,
77                 mock(AutomationCompositionElementRepository.class));
78
79         when(automationCompositionRepository.save(any(JpaAutomationComposition.class)))
80                 .thenReturn(inputAutomationCompositionsJpa.get(0));
81         var inputAc = inputAutomationCompositions.getAutomationCompositionList().get(0);
82
83         var createdAutomationComposition = automationCompositionProvider.createAutomationComposition(inputAc);
84         inputAc.setInstanceId(createdAutomationComposition.getInstanceId());
85         assertEquals(inputAc, createdAutomationComposition);
86     }
87
88     @Test
89     void testAutomationCompositionUpdate() {
90         var automationCompositionRepository = mock(AutomationCompositionRepository.class);
91         var automationCompositionProvider = new AutomationCompositionProvider(automationCompositionRepository,
92                 mock(AutomationCompositionElementRepository.class));
93
94         assertThatThrownBy(() -> automationCompositionProvider.updateAutomationComposition(null))
95                 .hasMessageMatching(AC_IS_NULL);
96
97         when(automationCompositionRepository.save(inputAutomationCompositionsJpa.get(0)))
98                 .thenReturn(inputAutomationCompositionsJpa.get(0));
99
100         var createdAutomationComposition = automationCompositionProvider
101                 .updateAutomationComposition(inputAutomationCompositions.getAutomationCompositionList().get(0));
102
103         assertEquals(inputAutomationCompositions.getAutomationCompositionList().get(0), createdAutomationComposition);
104     }
105
106     @Test
107     void testGetAutomationCompositions() {
108         var automationCompositionRepository = mock(AutomationCompositionRepository.class);
109         var automationCompositionProvider = new AutomationCompositionProvider(automationCompositionRepository,
110                 mock(AutomationCompositionElementRepository.class));
111
112         var automationComposition = inputAutomationCompositions.getAutomationCompositionList().get(0);
113         var acList = automationCompositionProvider.getAutomationCompositions(UUID.randomUUID(),
114                 automationComposition.getName(), automationComposition.getVersion());
115         assertThat(acList).isEmpty();
116
117         when(automationCompositionRepository.findAll(Mockito.<Example<JpaAutomationComposition>>any()))
118                 .thenReturn(inputAutomationCompositionsJpa);
119         acList = automationCompositionProvider.getAutomationCompositions(automationComposition.getCompositionId(), null,
120                 null);
121         assertThat(acList).hasSize(2);
122     }
123
124     @Test
125     void testGetAutomationComposition() {
126         var automationCompositionRepository = mock(AutomationCompositionRepository.class);
127         var automationCompositionProvider = new AutomationCompositionProvider(automationCompositionRepository,
128                 mock(AutomationCompositionElementRepository.class));
129
130         var automationComposition = inputAutomationCompositions.getAutomationCompositionList().get(0);
131         assertThatThrownBy(
132                 () -> automationCompositionProvider.getAutomationComposition(automationComposition.getInstanceId()))
133                         .hasMessageMatching("AutomationComposition not found");
134
135         when(automationCompositionRepository.findById(automationComposition.getInstanceId().toString()))
136                 .thenReturn(Optional.of(inputAutomationCompositionsJpa.get(0)));
137         var ac = automationCompositionProvider.getAutomationComposition(automationComposition.getInstanceId());
138         assertEquals(automationComposition, ac);
139     }
140
141     @Test
142     void testFindAutomationComposition() {
143         var automationCompositionRepository = mock(AutomationCompositionRepository.class);
144         var automationCompositionProvider = new AutomationCompositionProvider(automationCompositionRepository,
145                 mock(AutomationCompositionElementRepository.class));
146
147         var automationComposition = inputAutomationCompositions.getAutomationCompositionList().get(0);
148         var acOpt = automationCompositionProvider.findAutomationComposition(automationComposition.getInstanceId());
149         assertThat(acOpt).isEmpty();
150
151         acOpt = automationCompositionProvider.findAutomationComposition(automationComposition.getKey().asIdentifier());
152         assertThat(acOpt).isEmpty();
153
154         when(automationCompositionRepository.findById(automationComposition.getInstanceId().toString()))
155                 .thenReturn(Optional.of(inputAutomationCompositionsJpa.get(0)));
156         acOpt = automationCompositionProvider.findAutomationComposition(automationComposition.getInstanceId());
157         assertEquals(automationComposition, acOpt.get());
158
159         when(automationCompositionRepository.findOne(Mockito.<Example<JpaAutomationComposition>>any()))
160                 .thenReturn(Optional.of(inputAutomationCompositionsJpa.get(0)));
161         acOpt = automationCompositionProvider.findAutomationComposition(automationComposition.getKey().asIdentifier());
162         assertEquals(automationComposition, acOpt.get());
163     }
164
165     @Test
166     void testGetAcInstancesByCompositionId() {
167         var automationCompositionRepository = mock(AutomationCompositionRepository.class);
168         var automationCompositionProvider = new AutomationCompositionProvider(automationCompositionRepository,
169                 mock(AutomationCompositionElementRepository.class));
170
171         var automationComposition = inputAutomationCompositions.getAutomationCompositionList().get(0);
172         when(automationCompositionRepository.findByCompositionId(automationComposition.getCompositionId().toString()))
173                 .thenReturn(inputAutomationCompositionsJpa);
174         var acList =
175                 automationCompositionProvider.getAcInstancesByCompositionId(automationComposition.getCompositionId());
176         assertEquals(inputAutomationCompositions.getAutomationCompositionList(), acList);
177     }
178
179     @Test
180     void testDeleteAutomationComposition() {
181         var automationCompositionRepository = mock(AutomationCompositionRepository.class);
182         var automationCompositionProvider = new AutomationCompositionProvider(automationCompositionRepository,
183                 mock(AutomationCompositionElementRepository.class));
184
185         assertThatThrownBy(() -> automationCompositionProvider.deleteAutomationComposition(UUID.randomUUID()))
186                 .hasMessageMatching(".*.failed, automation composition does not exist");
187
188         var automationComposition = inputAutomationCompositions.getAutomationCompositionList().get(0);
189         when(automationCompositionRepository.findById(automationComposition.getInstanceId().toString()))
190                 .thenReturn(Optional.of(inputAutomationCompositionsJpa.get(0)));
191
192         var deletedAc =
193                 automationCompositionProvider.deleteAutomationComposition(automationComposition.getInstanceId());
194         assertEquals(automationComposition, deletedAc);
195     }
196
197     @Test
198     void testAutomationCompositionElementUpdate() {
199         var acElementRepository = mock(AutomationCompositionElementRepository.class);
200         var automationCompositionProvider = new AutomationCompositionProvider(
201             mock(AutomationCompositionRepository.class), acElementRepository);
202
203         assertThatThrownBy(() -> automationCompositionProvider.updateAutomationCompositionElement(null, null))
204             .hasMessageMatching(ACELEMENT_IS_NULL);
205
206         var acElement = inputAutomationCompositions.getAutomationCompositionList().get(0).getElements().values()
207             .iterator().next();
208         automationCompositionProvider.updateAutomationCompositionElement(acElement, UUID.randomUUID());
209         verify(acElementRepository).save(any());
210     }
211
212     @Test
213     void testValidateElementIds() {
214         var acElementRepository = mock(AutomationCompositionElementRepository.class);
215         var automationCompositionProvider = new AutomationCompositionProvider(
216             mock(AutomationCompositionRepository.class), acElementRepository);
217
218         var ac = inputAutomationCompositions.getAutomationCompositionList().get(0);
219
220         var result = automationCompositionProvider.validateElementIds(ac);
221         assertThat(result.isValid()).isTrue();
222
223         var jpaElement = new JpaAutomationCompositionElement(ac.getElements().values().iterator().next());
224         when(acElementRepository.findAllById(anyIterable()))
225             .thenReturn(List.of(jpaElement));
226
227         ac.setInstanceId(null);
228         result = automationCompositionProvider.validateElementIds(ac);
229         assertThat(result.isValid()).isFalse();
230
231         ac.setInstanceId(UUID.randomUUID());
232         jpaElement.setInstanceId(UUID.randomUUID().toString());
233         result = automationCompositionProvider.validateElementIds(ac);
234         assertThat(result.isValid()).isFalse();
235
236         ac.setInstanceId(UUID.randomUUID());
237         jpaElement.setInstanceId(ac.getInstanceId().toString());
238         result = automationCompositionProvider.validateElementIds(ac);
239         assertThat(result.isValid()).isTrue();
240     }
241
242     @Test
243     void testUpgradeStates() {
244         var acElementRepository = mock(AutomationCompositionElementRepository.class);
245         var automationCompositionProvider = new AutomationCompositionProvider(
246             mock(AutomationCompositionRepository.class), acElementRepository);
247
248         assertDoesNotThrow(() -> automationCompositionProvider.upgradeStates(List.of()));
249         var acElement = inputAutomationCompositions.getAutomationCompositionList().get(0).getElements().values()
250             .iterator().next();
251
252         var acInfo = new AutomationCompositionInfo();
253         var acElementInfo = new AutomationCompositionElementInfo();
254         acInfo.setElements(List.of(acElementInfo));
255         acElementInfo.setAutomationCompositionElementId(acElement.getId());
256
257         when(acElementRepository.getReferenceById(acElement.getId().toString()))
258             .thenReturn(new JpaAutomationCompositionElement(acElement));
259
260         automationCompositionProvider.upgradeStates(List.of(acInfo));
261         verify(acElementRepository).saveAll(anyList());
262     }
263 }