2c01e98876040b2f9a1665d19b38e99c481821da
[policy/clamp.git] /
1 /*-
2  * ============LICENSE_START=======================================================
3  *  Copyright (C) 2021-2022 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.fail;
26 import static org.mockito.ArgumentMatchers.any;
27 import static org.mockito.Mockito.mock;
28 import static org.mockito.Mockito.when;
29
30 import java.util.List;
31 import java.util.Optional;
32 import java.util.UUID;
33 import org.junit.jupiter.api.BeforeAll;
34 import org.junit.jupiter.api.Test;
35 import org.onap.policy.clamp.models.acm.concepts.AutomationCompositionDefinition;
36 import org.onap.policy.clamp.models.acm.persistence.concepts.JpaAutomationCompositionDefinition;
37 import org.onap.policy.clamp.models.acm.persistence.repository.AutomationCompositionDefinitionRepository;
38 import org.onap.policy.clamp.models.acm.persistence.repository.ToscaServiceTemplateRepository;
39 import org.onap.policy.common.utils.coder.CoderException;
40 import org.onap.policy.common.utils.coder.StandardYamlCoder;
41 import org.onap.policy.common.utils.resources.ResourceUtils;
42 import org.onap.policy.models.base.PfModelException;
43 import org.onap.policy.models.tosca.authorative.concepts.ToscaServiceTemplate;
44 import org.onap.policy.models.tosca.simple.concepts.JpaToscaServiceTemplate;
45
46 class AcDefinitionProviderTest {
47
48     private static final String TOSCA_SERVICE_TEMPLATE_YAML = "clamp/acm/pmsh/funtional-pmsh-usecase.yaml";
49
50     private static final StandardYamlCoder YAML_TRANSLATOR = new StandardYamlCoder();
51
52     private static ToscaServiceTemplate inputServiceTemplate;
53
54     @BeforeAll
55     static void loadServiceTemplate() {
56         inputServiceTemplate = getToscaServiceTemplate(TOSCA_SERVICE_TEMPLATE_YAML);
57     }
58
59     @Test
60     void testCreateServiceTemplate() throws PfModelException {
61         var acmDefinitionRepository = mock(AutomationCompositionDefinitionRepository.class);
62
63         var jpaServiceTemplate = ProviderUtils.getJpaAndValidate(inputServiceTemplate, JpaToscaServiceTemplate::new,
64                 "toscaServiceTemplate");
65         var acmDefinition = new AutomationCompositionDefinition();
66         acmDefinition.setCompositionId(UUID.randomUUID());
67         acmDefinition.setServiceTemplate(jpaServiceTemplate.toAuthorative());
68
69         when(acmDefinitionRepository.save(any(JpaAutomationCompositionDefinition.class)))
70                 .thenReturn(new JpaAutomationCompositionDefinition(acmDefinition));
71
72         var serviceTemplateRepository = mock(ToscaServiceTemplateRepository.class);
73         var acDefinitionProvider = new AcDefinitionProvider(serviceTemplateRepository, acmDefinitionRepository);
74         var result = acDefinitionProvider.createAutomationCompositionDefinition(inputServiceTemplate);
75
76         assertThat(result.getServiceTemplate()).isEqualTo(jpaServiceTemplate.toAuthorative());
77     }
78
79     @Test
80     void testDeleteAcDefintion() throws PfModelException {
81         var jpaServiceTemplate = ProviderUtils.getJpaAndValidate(inputServiceTemplate, JpaToscaServiceTemplate::new,
82                 "toscaServiceTemplate");
83         var acmDefinition = new AutomationCompositionDefinition();
84         acmDefinition.setCompositionId(UUID.randomUUID());
85         acmDefinition.setServiceTemplate(jpaServiceTemplate.toAuthorative());
86
87         var serviceTemplateRepository = mock(ToscaServiceTemplateRepository.class);
88         var acmDefinitionRepository = mock(AutomationCompositionDefinitionRepository.class);
89         when(acmDefinitionRepository.findById(acmDefinition.getCompositionId().toString()))
90                 .thenReturn(Optional.of(new JpaAutomationCompositionDefinition(acmDefinition)));
91
92         var acDefinitionProvider = new AcDefinitionProvider(serviceTemplateRepository, acmDefinitionRepository);
93         var result = acDefinitionProvider.deleteAcDefintion(acmDefinition.getCompositionId());
94
95         assertThat(result).isEqualTo(jpaServiceTemplate.toAuthorative());
96     }
97
98     @Test
99     void testDeleteServiceTemplateEmpty() throws PfModelException {
100         var serviceTemplateRepository = mock(ToscaServiceTemplateRepository.class);
101         when(serviceTemplateRepository.findAll()).thenReturn(List.of());
102
103         var compositionId = UUID.randomUUID();
104         var acmDefinitionRepository = mock(AutomationCompositionDefinitionRepository.class);
105         var acDefinitionProvider = new AcDefinitionProvider(serviceTemplateRepository, acmDefinitionRepository);
106         assertThatThrownBy(() -> acDefinitionProvider.deleteAcDefintion(compositionId))
107                 .hasMessage("delete of Automation Composition Definition \"" + compositionId
108                         + "\" failed, Automation Composition Definition does not exist");
109     }
110
111     @Test
112     void testGetServiceTemplate() throws PfModelException {
113         var jpaServiceTemplate = ProviderUtils.getJpaAndValidate(inputServiceTemplate, JpaToscaServiceTemplate::new,
114                 "toscaServiceTemplate");
115         var serviceTemplateRepository = mock(ToscaServiceTemplateRepository.class);
116         when(serviceTemplateRepository.getFiltered(JpaToscaServiceTemplate.class, inputServiceTemplate.getName(),
117                 inputServiceTemplate.getVersion())).thenReturn(List.of(jpaServiceTemplate));
118
119         var acmDefinitionRepository = mock(AutomationCompositionDefinitionRepository.class);
120         var acDefinitionProvider = new AcDefinitionProvider(serviceTemplateRepository, acmDefinitionRepository);
121         var result = acDefinitionProvider.getServiceTemplateList(inputServiceTemplate.getName(),
122                 inputServiceTemplate.getVersion());
123
124         assertThat(result).hasSize(1);
125         assertThat(result.get(0)).isEqualTo(jpaServiceTemplate.toAuthorative());
126     }
127
128     /**
129      * Get ToscaServiceTemplate from resource.
130      *
131      * @param path path of the resource
132      */
133     public static ToscaServiceTemplate getToscaServiceTemplate(String path) {
134
135         try {
136             return YAML_TRANSLATOR.decode(ResourceUtils.getResourceAsStream(path), ToscaServiceTemplate.class);
137         } catch (CoderException e) {
138             fail("Cannot read or decode " + path);
139             return null;
140         }
141     }
142 }