2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
6 * Modifications Copyright (c) 2019 Samsung
7 * ================================================================================
8 * Licensed under the Apache License, Version 2.0 (the "License");
9 * you may not use this file except in compliance with the License.
10 * You may obtain a copy of the License at
12 * http://www.apache.org/licenses/LICENSE-2.0
14 * Unless required by applicable law or agreed to in writing, software
15 * distributed under the License is distributed on an "AS IS" BASIS,
16 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17 * See the License for the specific language governing permissions and
18 * limitations under the License.
19 * ============LICENSE_END=========================================================
22 package org.openecomp.sdc.asdctool.impl.validator.tasks.artifacts;
24 import fj.data.Either;
25 import org.junit.After;
26 import org.junit.Assert;
27 import org.junit.Before;
28 import org.junit.Test;
29 import org.junit.runner.RunWith;
30 import org.mockito.InjectMocks;
31 import org.mockito.Mock;
32 import org.mockito.junit.MockitoJUnitRunner;
33 import org.openecomp.sdc.asdctool.impl.validator.config.ValidationConfigManager;
34 import org.openecomp.sdc.asdctool.impl.validator.utils.ReportManager;
35 import org.openecomp.sdc.asdctool.impl.validator.utils.ReportManagerHelper;
36 import org.openecomp.sdc.be.dao.cassandra.ArtifactCassandraDao;
37 import org.openecomp.sdc.be.dao.cassandra.CassandraOperationStatus;
38 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
39 import org.openecomp.sdc.be.datatypes.elements.ArtifactDataDefinition;
40 import org.openecomp.sdc.be.datatypes.elements.MapArtifactDataDefinition;
41 import org.openecomp.sdc.be.model.jsonjanusgraph.datamodel.TopologyTemplate;
42 import org.openecomp.sdc.be.model.jsonjanusgraph.operations.TopologyTemplateOperation;
43 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
44 import org.powermock.core.classloader.annotations.PrepareForTest;
45 import org.powermock.modules.junit4.PowerMockRunner;
46 import org.powermock.modules.junit4.PowerMockRunnerDelegate;
49 import java.util.ArrayList;
50 import java.util.HashMap;
51 import java.util.List;
53 import java.util.Objects;
54 import java.util.stream.IntStream;
56 import static org.junit.Assert.assertEquals;
57 import static org.junit.Assert.assertFalse;
58 import static org.junit.Assert.assertTrue;
59 import static org.mockito.ArgumentMatchers.any;
60 import static org.mockito.ArgumentMatchers.eq;
61 import static org.mockito.Mockito.when;
63 @RunWith(PowerMockRunner.class)
64 @PowerMockRunnerDelegate(MockitoJUnitRunner.class)
65 @PrepareForTest({ReportManager.class})
66 public class ArtifactValidationUtilsTest {
69 private ArtifactCassandraDao artifactCassandraDao;
71 private TopologyTemplateOperation topologyTemplateOperation;
73 private ArtifactValidationUtils testSubject;
76 private GraphVertex vertex;
78 private MapArtifactDataDefinition mapToscaDataDefinition;
80 private ArtifactDataDefinition artifactDataDefinition;
82 private ArtifactDataDefinition artifactDataDefinitionNotInCassandra;
84 private ArtifactDataDefinition artifactDataDefinitionDummy;
86 private TopologyTemplate topologyTemplate;
88 private static final String ES_ID = "testEsInCassandra";
89 private static final String ES_ID_NOT_IN_CASS = "testEsNotInCassandra";
90 private static final String TASK_NAME = "testTaskName";
91 private static final String UNIQUE_ID = "4321";
92 private static final String UNIQUE_ID_VERTEX = "321";
94 public void initReportManager() {
95 String resourcePath = new File(Objects
96 .requireNonNull(ArtifactValidationUtilsTest.class.getClassLoader().getResource(""))
97 .getFile()).getAbsolutePath();
98 ValidationConfigManager.setOutputFullFilePath(resourcePath);
103 public void setup() {
105 when(artifactCassandraDao.getCountOfArtifactById(ES_ID)).thenReturn(Either.left(1L));
106 when(artifactCassandraDao.getCountOfArtifactById(ES_ID_NOT_IN_CASS))
107 .thenReturn(Either.right(CassandraOperationStatus.NOT_FOUND));
109 when(artifactDataDefinition.getEsId()).thenReturn(ES_ID);
110 when(artifactDataDefinitionNotInCassandra.getEsId()).thenReturn(ES_ID_NOT_IN_CASS);
112 when(artifactDataDefinitionNotInCassandra.getUniqueId()).thenReturn(UNIQUE_ID);
113 when(vertex.getUniqueId()).thenReturn(UNIQUE_ID_VERTEX);
117 public void clean() {
118 ReportManagerHelper.cleanReports();
122 public void testValidateArtifactsAreInCassandra() {
124 List<ArtifactDataDefinition> artifacts = new ArrayList<>();
125 artifacts.add(artifactDataDefinition);
128 ArtifactsVertexResult result =
129 testSubject.validateArtifactsAreInCassandra(vertex, TASK_NAME, artifacts);
131 List reportOutputFile = ReportManagerHelper.getReportOutputFileAsList();
134 assertTrue(result.getStatus());
135 assertEquals(0, result.notFoundArtifacts.size());
136 assertEquals("Artifact " + ES_ID + " is in Cassandra", reportOutputFile.get(2));
140 public void testValidateArtifactsNotInCassandra() {
142 List<ArtifactDataDefinition> artifacts = new ArrayList<>();
143 artifacts.add(artifactDataDefinition);
144 artifacts.add(artifactDataDefinitionNotInCassandra);
147 ArtifactsVertexResult result =
148 testSubject.validateArtifactsAreInCassandra(vertex, TASK_NAME, artifacts);
149 ReportManager.reportEndOfToolRun();
151 List reportOutputFile = ReportManagerHelper.getReportOutputFileAsList();
154 assertFalse(result.getStatus());
155 assertEquals(1, result.notFoundArtifacts.size());
156 assertEquals(UNIQUE_ID, result.notFoundArtifacts.iterator().next());
158 assertEquals("Artifact " + ES_ID + " is in Cassandra", reportOutputFile.get(2));
159 assertEquals("Artifact " + ES_ID_NOT_IN_CASS + " doesn't exist in Cassandra",
160 reportOutputFile.get(3));
161 assertEquals("Task: " + TASK_NAME, reportOutputFile.get(5));
162 assertEquals("FailedVertices: [" + UNIQUE_ID_VERTEX + "]", reportOutputFile.get(6));
166 public void testIsArtifactsInCassandra() {
168 boolean notInCass = testSubject.isArtifactInCassandra(ES_ID_NOT_IN_CASS);
169 boolean inCass = testSubject.isArtifactInCassandra(ES_ID);
172 assertFalse(notInCass);
177 public void testAddRelevantArtifacts() {
179 Map<String, ArtifactDataDefinition> artifactsMap = new HashMap<>();
180 artifactsMap.put(ES_ID_NOT_IN_CASS, artifactDataDefinitionNotInCassandra);
181 artifactsMap.put(ES_ID, artifactDataDefinition);
184 List<ArtifactDataDefinition> result = testSubject.addRelevantArtifacts(artifactsMap);
187 result.forEach(Assert::assertNotNull);
191 public void testAddRelevantArtifactsWithNullEsId() {
193 Map<String, ArtifactDataDefinition> artifactsMap = new HashMap<>();
194 artifactsMap.put("", artifactDataDefinitionDummy);
197 List<ArtifactDataDefinition> result = testSubject.addRelevantArtifacts(artifactsMap);
200 assertEquals(0, result.size());
204 public void testValidateTopologyTemplateArtifacts() {
206 Map<String, ArtifactDataDefinition> artifacts = new HashMap<>();
207 artifacts.put(ES_ID, artifactDataDefinition);
209 when(topologyTemplate.getDeploymentArtifacts()).thenReturn(artifacts);
210 when(topologyTemplate.getArtifacts()).thenReturn(artifacts);
211 when(topologyTemplate.getServiceApiArtifacts()).thenReturn(artifacts);
213 when(mapToscaDataDefinition.getMapToscaDataDefinition()).thenReturn(artifacts);
214 Map<String, MapArtifactDataDefinition> artifactsMap = new HashMap<>();
215 artifactsMap.put(ES_ID, mapToscaDataDefinition);
217 when(topologyTemplate.getInstanceArtifacts()).thenReturn(artifactsMap);
218 when(topologyTemplate.getInstDeploymentArtifacts()).thenReturn(artifactsMap);
220 when(topologyTemplateOperation.getToscaElement(eq(vertex.getUniqueId()), any()))
221 .thenReturn(Either.left(topologyTemplate));
224 ArtifactsVertexResult result =
225 testSubject.validateTopologyTemplateArtifacts(vertex, TASK_NAME);
227 List reportOutputFile = ReportManagerHelper.getReportOutputFileAsList();
230 assertTrue(result.getStatus());
231 assertEquals(0, result.notFoundArtifacts.size());
233 IntStream.range(2, reportOutputFile.size()).forEach(
234 i -> assertEquals("Artifact " + ES_ID + " is in Cassandra", reportOutputFile.get(i)));
238 public void testValidateTopologyTemplateArtifactsNotFoundToscaElement() {
240 when(topologyTemplateOperation.getToscaElement(eq(vertex.getUniqueId()), any()))
241 .thenReturn(Either.right(StorageOperationStatus.NOT_FOUND));
244 ArtifactsVertexResult result =
245 testSubject.validateTopologyTemplateArtifacts(vertex, TASK_NAME);
248 assertFalse(result.getStatus());