523a4fb885f51fc528895f4573ed31e1dabbc1a3
[policy/clamp.git] /
1 /*-
2  * ============LICENSE_START=======================================================
3  *  Copyright (C) 2021 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.controlloop.models.controlloop.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.assertNotNull;
26 import static org.junit.jupiter.api.Assertions.fail;
27 import static org.mockito.ArgumentMatchers.any;
28 import static org.mockito.Mockito.mock;
29 import static org.mockito.Mockito.when;
30
31 import java.util.List;
32 import java.util.Optional;
33 import org.junit.jupiter.api.BeforeAll;
34 import org.junit.jupiter.api.Test;
35 import org.onap.policy.clamp.controlloop.models.controlloop.persistence.repository.ToscaServiceTemplateRepository;
36 import org.onap.policy.common.utils.coder.CoderException;
37 import org.onap.policy.common.utils.coder.StandardYamlCoder;
38 import org.onap.policy.common.utils.resources.ResourceUtils;
39 import org.onap.policy.models.base.PfConceptKey;
40 import org.onap.policy.models.base.PfModelException;
41 import org.onap.policy.models.tosca.authorative.concepts.ToscaServiceTemplate;
42 import org.onap.policy.models.tosca.simple.concepts.JpaToscaServiceTemplate;
43
44 class ServiceTemplateProviderTest {
45
46     private static final String TOSCA_SERVICE_TEMPLATE_YAML =
47             "src/test/resources/providers/tosca-for-smoke-testing.yaml";
48
49     private static final StandardYamlCoder YAML_TRANSLATOR = new StandardYamlCoder();
50
51     private static ToscaServiceTemplate inputServiceTemplate;
52
53     @BeforeAll
54     static void loadServiceTemplate() {
55         inputServiceTemplate = getToscaServiceTemplate(TOSCA_SERVICE_TEMPLATE_YAML);
56     }
57
58     @Test
59     void testGetCommonOrInstancePropertiesFromNodeTypes() throws PfModelException {
60         var serviceTemplateRepository = mock(ToscaServiceTemplateRepository.class);
61         var serviceTemplateProvider = new ServiceTemplateProvider(serviceTemplateRepository);
62
63         var result = serviceTemplateProvider.getCommonOrInstancePropertiesFromNodeTypes(true, inputServiceTemplate);
64         assertNotNull(result);
65         assertThat(result).hasSize(8);
66     }
67
68     @Test
69     void testGetDerivedCommonOrInstanceNodeTemplates() throws PfModelException {
70         var serviceTemplateRepository = mock(ToscaServiceTemplateRepository.class);
71         var serviceTemplateProvider = new ServiceTemplateProvider(serviceTemplateRepository);
72
73         var commonOrInstanceNodeTypeProps =
74                 serviceTemplateProvider.getCommonOrInstancePropertiesFromNodeTypes(true, inputServiceTemplate);
75
76         var result = serviceTemplateProvider.getDerivedCommonOrInstanceNodeTemplates(
77                 inputServiceTemplate.getToscaTopologyTemplate().getNodeTemplates(), commonOrInstanceNodeTypeProps);
78
79         assertNotNull(result);
80         assertThat(result).hasSize(8);
81     }
82
83     @Test
84     void testCreateServiceTemplate() throws PfModelException {
85         var serviceTemplateRepository = mock(ToscaServiceTemplateRepository.class);
86         var serviceTemplateProvider = new ServiceTemplateProvider(serviceTemplateRepository);
87
88         var jpaServiceTemplate = ProviderUtils.getJpaAndValidate(inputServiceTemplate, JpaToscaServiceTemplate::new,
89                 "toscaServiceTemplate");
90         when(serviceTemplateRepository.save(jpaServiceTemplate)).thenReturn(jpaServiceTemplate);
91
92         var result = serviceTemplateProvider.createServiceTemplate(inputServiceTemplate);
93
94         assertThat(result).isEqualTo(jpaServiceTemplate.toAuthorative());
95     }
96
97     @Test
98     void testDeleteServiceTemplate() throws PfModelException {
99         var jpaServiceTemplate = ProviderUtils.getJpaAndValidate(inputServiceTemplate, JpaToscaServiceTemplate::new,
100                 "toscaServiceTemplate");
101         var serviceTemplateRepository = mock(ToscaServiceTemplateRepository.class);
102         when(serviceTemplateRepository
103                 .findById(new PfConceptKey(inputServiceTemplate.getName(), inputServiceTemplate.getVersion())))
104                         .thenReturn(Optional.of(jpaServiceTemplate));
105
106         var serviceTemplateProvider = new ServiceTemplateProvider(serviceTemplateRepository);
107         var result = serviceTemplateProvider.deleteServiceTemplate(inputServiceTemplate.getName(),
108                 inputServiceTemplate.getVersion());
109
110         assertThat(result).isEqualTo(jpaServiceTemplate.toAuthorative());
111     }
112
113     @Test
114     void testGetServiceTemplateListEmpty() throws PfModelException {
115         var serviceTemplateRepository = mock(ToscaServiceTemplateRepository.class);
116         when(serviceTemplateRepository.findById(any())).thenReturn(Optional.empty());
117
118         var serviceTemplateProvider = new ServiceTemplateProvider(serviceTemplateRepository);
119         assertThatThrownBy(() -> serviceTemplateProvider.getToscaServiceTemplate("Name", "1.0.0"))
120                 .hasMessage("Control Loop definitions not found");
121     }
122
123     @Test
124     void testGetServiceTemplateList() throws PfModelException {
125         var jpaServiceTemplate = ProviderUtils.getJpaAndValidate(inputServiceTemplate, JpaToscaServiceTemplate::new,
126                 "toscaServiceTemplate");
127         var serviceTemplateRepository = mock(ToscaServiceTemplateRepository.class);
128         when(serviceTemplateRepository
129                 .findById(new PfConceptKey(inputServiceTemplate.getName(), inputServiceTemplate.getVersion())))
130                         .thenReturn(Optional.of(jpaServiceTemplate));
131
132         var serviceTemplateProvider = new ServiceTemplateProvider(serviceTemplateRepository);
133         var result = serviceTemplateProvider.getToscaServiceTemplate(inputServiceTemplate.getName(),
134                 inputServiceTemplate.getVersion());
135
136         assertThat(result).isEqualTo(jpaServiceTemplate.toAuthorative());
137     }
138
139     @Test
140     void testGetServiceTemplate() throws PfModelException {
141         var jpaServiceTemplate = ProviderUtils.getJpaAndValidate(inputServiceTemplate, JpaToscaServiceTemplate::new,
142                 "toscaServiceTemplate");
143         var serviceTemplateRepository = mock(ToscaServiceTemplateRepository.class);
144         when(serviceTemplateRepository.getFiltered(JpaToscaServiceTemplate.class, inputServiceTemplate.getName(),
145                 inputServiceTemplate.getVersion())).thenReturn(List.of(jpaServiceTemplate));
146
147         var serviceTemplateProvider = new ServiceTemplateProvider(serviceTemplateRepository);
148         var result = serviceTemplateProvider.getServiceTemplateList(inputServiceTemplate.getName(),
149                 inputServiceTemplate.getVersion());
150
151         assertThat(result).hasSize(1);
152         assertThat(result.get(0)).isEqualTo(jpaServiceTemplate.toAuthorative());
153     }
154
155     /**
156      * Get ToscaServiceTemplate from resource.
157      *
158      * @param path path of the resource
159      */
160     public static ToscaServiceTemplate getToscaServiceTemplate(String path) {
161
162         try {
163             return YAML_TRANSLATOR.decode(ResourceUtils.getResourceAsStream(path), ToscaServiceTemplate.class);
164         } catch (CoderException e) {
165             fail("Cannot read or decode " + path);
166             return null;
167         }
168     }
169 }