3 * ============LICENSE_START=======================================================
4 * Copyright (C) 2020 Nordix Foundation
5 * ================================================================================
6 * Licensed under the Apache License, Version 2.0 (the "License");
7 * you may not use this file except in compliance with the License.
8 * You may obtain a copy of the License at
10 * http://www.apache.org/licenses/LICENSE-2.0
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.
17 * SPDX-License-Identifier: Apache-2.0
18 * ============LICENSE_END=========================================================
20 package org.openecomp.sdc.be.plugins.etsi.nfv.nsd.generator;
22 import static org.hamcrest.MatcherAssert.assertThat;
23 import static org.hamcrest.core.Is.is;
24 import static org.hamcrest.core.IsNull.notNullValue;
25 import static org.junit.jupiter.api.Assertions.assertThrows;
26 import static org.mockito.Mockito.mock;
27 import static org.mockito.Mockito.when;
28 import static org.openecomp.sdc.be.plugins.etsi.nfv.nsd.generator.EtsiNfvNsCsarEntryGenerator.ETSI_NS_COMPONENT_CATEGORY;
29 import static org.openecomp.sdc.common.api.ArtifactTypeEnum.ONBOARDED_PACKAGE;
31 import fj.data.Either;
32 import java.util.ArrayList;
33 import java.util.Collections;
34 import java.util.HashMap;
35 import java.util.List;
37 import java.util.Optional;
38 import org.junit.jupiter.api.BeforeEach;
39 import org.junit.jupiter.api.Test;
40 import org.mockito.Mock;
41 import org.mockito.MockitoAnnotations;
42 import org.openecomp.sdc.be.dao.cassandra.ArtifactCassandraDao;
43 import org.openecomp.sdc.be.datatypes.enums.ComponentTypeEnum;
44 import org.openecomp.sdc.be.datatypes.enums.JsonPresentationFields;
45 import org.openecomp.sdc.be.model.ArtifactDefinition;
46 import org.openecomp.sdc.be.model.ComponentInstance;
47 import org.openecomp.sdc.be.model.Service;
48 import org.openecomp.sdc.be.model.category.CategoryDefinition;
49 import org.openecomp.sdc.be.plugins.etsi.nfv.nsd.exception.NsdException;
50 import org.openecomp.sdc.be.plugins.etsi.nfv.nsd.exception.VnfDescriptorException;
51 import org.openecomp.sdc.be.plugins.etsi.nfv.nsd.factory.NsDescriptorGeneratorFactory;
52 import org.openecomp.sdc.be.plugins.etsi.nfv.nsd.generator.config.EtsiVersion;
53 import org.openecomp.sdc.be.plugins.etsi.nfv.nsd.generator.config.NsDescriptorConfig;
54 import org.openecomp.sdc.be.plugins.etsi.nfv.nsd.model.Nsd;
55 import org.openecomp.sdc.be.plugins.etsi.nfv.nsd.model.VnfDescriptor;
56 import org.openecomp.sdc.be.resources.data.DAOArtifactData;
58 class EtsiNfvNsdCsarGeneratorImplTest {
60 private static final String SERVICE_NORMALIZED_NAME = "normalizedName";
62 private VnfDescriptorGenerator vnfDescriptorGenerator;
64 private NsDescriptorGenerator nsDescriptorGeneratorImpl;
66 private NsDescriptorGeneratorFactory nsDescriptorGeneratorFactory;
68 private ArtifactCassandraDao artifactCassandraDao;
70 private Service service;
71 private EtsiNfvNsdCsarGeneratorImpl etsiNfvNsdCsarGenerator;
75 MockitoAnnotations.initMocks(this);
76 final EtsiVersion version2_5_1 = EtsiVersion.VERSION_2_5_1;
77 etsiNfvNsdCsarGenerator = new EtsiNfvNsdCsarGeneratorImpl(new NsDescriptorConfig(version2_5_1), vnfDescriptorGenerator,
78 nsDescriptorGeneratorFactory, artifactCassandraDao);
79 when(nsDescriptorGeneratorFactory.create()).thenReturn(nsDescriptorGeneratorImpl);
83 void generateNsdCsarSuccessfulTest() throws VnfDescriptorException, NsdException {
84 mockServiceComponent();
85 mockServiceComponentArtifacts();
86 final byte[] nsdCsar = etsiNfvNsdCsarGenerator.generateNsdCsar(service);
87 assertThat("", nsdCsar, is(notNullValue()));
91 void invalidComponentTest() {
92 assertThrows(NsdException.class, () -> etsiNfvNsdCsarGenerator.generateNsdCsar(null));
95 private void mockServiceComponent() throws VnfDescriptorException, NsdException {
96 when(service.getNormalizedName()).thenReturn(SERVICE_NORMALIZED_NAME);
97 when(service.getComponentType()).thenReturn(ComponentTypeEnum.SERVICE);
98 final String componentInstance1Name = "componentInstance1";
99 final ComponentInstance componentInstance1 = mockServiceComponentInstance(componentInstance1Name);
100 final ArtifactDefinition instanceArtifact1 = mockComponentInstanceArtifact(componentInstance1, "instanceArtifact1");
101 final VnfDescriptor vnfDescriptor1 = new VnfDescriptor();
102 final List<VnfDescriptor> vnfDescriptorList = Collections.singletonList(vnfDescriptor1);
103 final Nsd nsd = new Nsd();
104 when(vnfDescriptorGenerator.generate(componentInstance1Name, instanceArtifact1)).thenReturn(Optional.of(vnfDescriptor1));
105 when(nsDescriptorGeneratorImpl.generate(service, vnfDescriptorList)).thenReturn(Optional.of(nsd));
106 final List<CategoryDefinition> categoryDefinitionList = new ArrayList<>();
107 final CategoryDefinition nsComponentCategoryDefinition = new CategoryDefinition();
108 nsComponentCategoryDefinition.setName(ETSI_NS_COMPONENT_CATEGORY);
109 categoryDefinitionList.add(nsComponentCategoryDefinition);
110 when(service.getCategories()).thenReturn(categoryDefinitionList);
113 private void mockServiceComponentArtifacts() {
114 final Map<String, ArtifactDefinition> allArtifactsMap = new HashMap<>();
115 final String artifact1Id = "artifact1";
116 final ArtifactDefinition artifact1 = mockArtifactDefinition(artifact1Id);
117 final byte[] artifact1Bytes = new byte[1];
118 allArtifactsMap.put(artifact1Id, artifact1);
119 when(service.getAllArtifacts()).thenReturn(allArtifactsMap);
120 final DAOArtifactData artifact1Data = new DAOArtifactData();
121 artifact1Data.setDataAsArray(artifact1Bytes);
122 when(artifactCassandraDao.getArtifact(artifact1Id)).thenReturn(Either.left(artifact1Data));
125 private ComponentInstance mockServiceComponentInstance(final String componentInstanceName) {
126 final Map<String, ArtifactDefinition> deploymentArtifactMap = new HashMap<>();
127 final String instanceArtifact1Id = "instanceArtifact1";
128 final ArtifactDefinition instanceArtifact1 = mockArtifactDefinition(instanceArtifact1Id);
129 instanceArtifact1.setToscaPresentationValue(JsonPresentationFields.ARTIFACT_TYPE, ONBOARDED_PACKAGE.getType());
130 deploymentArtifactMap.put(instanceArtifact1Id, instanceArtifact1);
131 DAOArtifactData instanceArtifact1Data = new DAOArtifactData();
132 final byte[] instanceArtifact1Bytes = new byte[1];
133 instanceArtifact1Data.setDataAsArray(instanceArtifact1Bytes);
134 when(artifactCassandraDao.getArtifact(instanceArtifact1Id)).thenReturn(Either.left(instanceArtifact1Data));
135 final ComponentInstance componentInstance = mock(ComponentInstance.class);
136 when(componentInstance.getDeploymentArtifacts()).thenReturn(deploymentArtifactMap);
137 when(componentInstance.getName()).thenReturn(componentInstanceName);
138 final List<ComponentInstance> componentInstanceList = new ArrayList<>();
139 componentInstanceList.add(componentInstance);
140 when(service.getComponentInstances()).thenReturn(componentInstanceList);
141 return componentInstance;
144 private ArtifactDefinition mockComponentInstanceArtifact(final ComponentInstance componentInstance, final String instanceArtifactId) {
145 final Map<String, ArtifactDefinition> deploymentArtifactMap = new HashMap<>();
146 when(componentInstance.getDeploymentArtifacts()).thenReturn(deploymentArtifactMap);
147 final ArtifactDefinition instanceArtifact1 = mockArtifactDefinition(instanceArtifactId);
148 instanceArtifact1.setToscaPresentationValue(JsonPresentationFields.ARTIFACT_TYPE, ONBOARDED_PACKAGE.getType());
149 deploymentArtifactMap.put(instanceArtifactId, instanceArtifact1);
150 DAOArtifactData instanceArtifact1Data = new DAOArtifactData();
151 final byte[] instanceArtifact1Bytes = new byte[1];
152 instanceArtifact1Data.setDataAsArray(instanceArtifact1Bytes);
153 when(artifactCassandraDao.getArtifact(instanceArtifactId)).thenReturn(Either.left(instanceArtifact1Data));
154 return instanceArtifact1;
157 private ArtifactDefinition mockArtifactDefinition(final String artifactId) {
158 final ArtifactDefinition artifact = new ArtifactDefinition();
159 artifact.setEsId(artifactId);