openecomp-be/backend/openecomp-sdc-vendor-software-product-manager/src/test/java/org/openecomp/sdc/vendorsoftwareproduct/services/impl/composition/CompositionEntityDataManagerImplTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * SDC
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.openecomp.sdc.vendorsoftwareproduct.services.impl.composition;
  22
  23 import org.apache.commons.collections.CollectionUtils;
  24 import org.mockito.InjectMocks;
  25 import org.mockito.MockitoAnnotations;
  26 import org.mockito.Spy;
  27 import org.openecomp.core.utilities.file.FileUtils;
  28 import org.openecomp.sdc.common.errors.CoreException;
  29 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ComponentEntity;
  30 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.NetworkEntity;
  31 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.NicEntity;
  32 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.VspQuestionnaireEntity;
  33 import org.openecomp.sdc.vendorsoftwareproduct.types.composition.CompositionEntityId;
  34 import org.openecomp.sdc.vendorsoftwareproduct.types.composition.CompositionEntityType;
  35 import org.openecomp.sdc.vendorsoftwareproduct.types.composition.CompositionEntityValidationData;
  36 import org.openecomp.sdc.vendorsoftwareproduct.types.composition.Network;
  37 import org.openecomp.sdc.vendorsoftwareproduct.types.schemagenerator.NetworkCompositionSchemaInput;
  38 import org.openecomp.sdc.vendorsoftwareproduct.types.schemagenerator.SchemaTemplateContext;
  39 import org.openecomp.sdc.versioning.dao.types.Version;
  40 import org.testng.Assert;
  41 import org.testng.annotations.BeforeMethod;
  42 import org.testng.annotations.Test;
  43
  44 import java.util.ArrayList;
  45 import java.util.Collection;
  46 import java.util.List;
  47 import java.util.Map;
  48
  49 import static org.mockito.Mockito.doReturn;
  50
  51 public class CompositionEntityDataManagerImplTest {
  52   private static final String VSP1 = "vsp1";
  53   private static final Version VERSION = new Version(0, 1);
  54   private static final String COMPONENT1 = "component1";
  55   private static final String NIC1 = "nic1";
  56   private static final String SIMPLE_SCHEMA = "{\n" +
  57       "  \"$schema\": \"http://json-schema.org/draft-04/schema#\",\n" +
  58       "  \"type\": \"object\",\n" +
  59       "  \"properties\": {\n" +
  60       "    \"a\": {\n" +
  61       "      \"type\": \"number\"\n" +
  62       "    }\n" +
  63       "  }\n" +
  64       "}";
  65   private static final String NETWORK_COMPOSITION_SCHEMA = "{\n" +
  66       "  \"$schema\": \"http://json-schema.org/draft-04/schema#\",\n" +
  67       "  \"type\": \"object\",\n" +
  68       "  \"properties\": {\n" +
  69       "    \"name\": {\n" +
  70       "      \"type\": \"string\",\n" +
  71       "      \"enum\": [\n" +
  72       "        \"network1 name\"\n" +
  73       "      ],\n" +
  74       "      \"default\": \"network1 name\"\n" +
  75       "    },\n" +
  76       "    \"dhcp\": {\n" +
  77       "      \"type\": \"boolean\",\n" +
  78       "      \"enum\": [\n" +
  79       "        true\n" +
  80       "      ],\n" +
  81       "      \"default\": true\n" +
  82       "    }\n" +
  83       "  },\n" +
  84       "  \"additionalProperties\": false,\n" +
  85       "  \"required\": [\n" +
  86       "    \"name\",\n" +
  87       "    \"dhcp\"\n" +
  88       "  ]\n" +
  89       "}";
  90
  91   private Map<CompositionEntityId, Collection<String>> errorsById;
  92   @InjectMocks
  93   @Spy
  94   private CompositionEntityDataManagerImpl compositionEntityDataManager;
  95
  96   @BeforeMethod
  97   public void setUp() throws Exception {
  98     MockitoAnnotations.initMocks(this);
  99   }
 100
 101   @Test(expectedExceptions = CoreException.class)
 102   public void testAddNullEntity_negative() {
 103     compositionEntityDataManager.addEntity(null, null);
 104   }
 105
 106   @Test
 107   public void testAddEntity() {
 108     compositionEntityDataManager
 109         .addEntity(new VspQuestionnaireEntity(VSP1, VERSION), null);
 110
 111     String invalidQuestionnaireData = "{\"a\": \"b\"}";
 112
 113     ComponentEntity component = new ComponentEntity(VSP1, VERSION, COMPONENT1);
 114     component.setQuestionnaireData(invalidQuestionnaireData);
 115     compositionEntityDataManager.addEntity(component, null);
 116
 117     NicEntity nic = new NicEntity(VSP1, VERSION, COMPONENT1, NIC1);
 118     nic.setQuestionnaireData(invalidQuestionnaireData);
 119     compositionEntityDataManager.addEntity(nic, null);
 120   }
 121
 122   // TODO: 3/15/2017 fix and enable
 123   //@Test(dependsOnMethods = "testAddEntity")
 124   public void testValidateEntitiesQuestionnaire() {
 125     doReturn(SIMPLE_SCHEMA).when(compositionEntityDataManager)
 126         .generateSchema(SchemaTemplateContext.questionnaire, CompositionEntityType.vsp, null);
 127     doReturn(SIMPLE_SCHEMA).when(compositionEntityDataManager)
 128         .generateSchema(SchemaTemplateContext.questionnaire, CompositionEntityType.component, null);
 129     doReturn(SIMPLE_SCHEMA).when(compositionEntityDataManager)
 130         .generateSchema(SchemaTemplateContext.questionnaire, CompositionEntityType.nic, null);
 131
 132     errorsById = compositionEntityDataManager.validateEntitiesQuestionnaire();
 133     Assert.assertNotNull(errorsById);
 134     Assert.assertEquals(errorsById.size(), 2); // both component and nic data don't mach schemas
 135     CompositionEntityId nicId =
 136         new NicEntity(VSP1, VERSION, COMPONENT1, NIC1).getCompositionEntityId();
 137     Assert.assertTrue(errorsById.containsKey(nicId));
 138     Assert.assertTrue(errorsById.containsKey(nicId.getParentId()));
 139   }
 140
 141   @Test(dependsOnMethods = "testAddEntity")
 142   public void testBuildTrees() {
 143     compositionEntityDataManager.buildTrees();
 144   }
 145
 146   // TODO: 3/15/2017 fix and enable
 147   //@Test(dependsOnMethods = "testBuildTrees")
 148   public void testAddErrorsToTrees() {
 149     compositionEntityDataManager.addErrorsToTrees(errorsById);
 150   }
 151
 152   // TODO: 3/15/2017 fix and enable
 153   //@Test(dependsOnMethods = "testAddErrorsToTrees")
 154   public void testGetTrees() {
 155     Collection<CompositionEntityValidationData> trees = compositionEntityDataManager.getTrees();
 156     Assert.assertNotNull(trees);
 157     Assert.assertEquals(trees.size(), 1);
 158
 159     CompositionEntityValidationData vspValidationData = trees.iterator().next();
 160     assertValidationData(vspValidationData, VSP1, CompositionEntityType.vsp, false);
 161     Assert.assertEquals(vspValidationData.getSubEntitiesValidationData().size(), 1);
 162
 163     CompositionEntityValidationData componentValidationData =
 164         vspValidationData.getSubEntitiesValidationData().iterator().next();
 165     assertValidationData(componentValidationData, COMPONENT1, CompositionEntityType.component,
 166         true);
 167     Assert.assertEquals(componentValidationData.getSubEntitiesValidationData().size(), 1);
 168
 169     CompositionEntityValidationData nicValidationData =
 170         componentValidationData.getSubEntitiesValidationData().iterator().next();
 171     assertValidationData(nicValidationData, NIC1, CompositionEntityType.nic, true);
 172     Assert.assertNull(nicValidationData.getSubEntitiesValidationData());
 173   }
 174
 175   @Test
 176   public void testValidateValidEntity() {
 177     NetworkEntity networkEntity = new NetworkEntity(VSP1, VERSION, "network1");
 178     Network network = new Network();
 179     network.setName("network1 name");
 180     network.setDhcp(true);
 181     networkEntity.setNetworkCompositionData(network);
 182
 183     NetworkCompositionSchemaInput schemaTemplateInput = new NetworkCompositionSchemaInput();
 184     schemaTemplateInput.setManual(false);
 185     schemaTemplateInput.setNetwork(network);
 186
 187     doReturn(NETWORK_COMPOSITION_SCHEMA).when(compositionEntityDataManager)
 188         .generateSchema(SchemaTemplateContext.composition, CompositionEntityType.network,
 189             schemaTemplateInput);
 190
 191     CompositionEntityValidationData validationData = compositionEntityDataManager
 192         .validateEntity(networkEntity, SchemaTemplateContext.composition, schemaTemplateInput);
 193     assertValidationData(validationData, "network1", CompositionEntityType.network, false);
 194   }
 195
 196   @Test
 197   public void testValidateInvalidEntity() {
 198     NetworkEntity networkEntity = new NetworkEntity(VSP1, VERSION, "network1");
 199     Network network = new Network();
 200     network.setName("network1 name changed");
 201     network.setDhcp(false);
 202     networkEntity.setNetworkCompositionData(network);
 203
 204     NetworkCompositionSchemaInput schemaTemplateInput = new NetworkCompositionSchemaInput();
 205     schemaTemplateInput.setManual(false);
 206     Network origNetwork = new Network();
 207     origNetwork.setName("network1 name");
 208     origNetwork.setDhcp(true);
 209     schemaTemplateInput.setNetwork(origNetwork);
 210
 211     doReturn(NETWORK_COMPOSITION_SCHEMA).when(compositionEntityDataManager)
 212         .generateSchema(SchemaTemplateContext.composition, CompositionEntityType.network,
 213             schemaTemplateInput);
 214
 215     CompositionEntityValidationData validationData = compositionEntityDataManager
 216         .validateEntity(networkEntity, SchemaTemplateContext.composition, schemaTemplateInput);
 217     assertValidationData(validationData, "network1", CompositionEntityType.network, true);
 218     Assert.assertEquals(validationData.getErrors().size(), 2);
 219   }
 220
 221   @Test
 222   public void testNicAndComponentValidQuestionnaire() {
 223     compositionEntityDataManager
 224         .addEntity(new VspQuestionnaireEntity(VSP1, VERSION), null);
 225
 226     ComponentEntity componentEntity = new ComponentEntity(VSP1, VERSION, COMPONENT1);
 227     componentEntity.setQuestionnaireData(loadFileToString("quesionnaire/validComponent.json"));
 228     compositionEntityDataManager.addEntity(componentEntity, null);
 229
 230     NicEntity nicEntity = new NicEntity(VSP1, VERSION, COMPONENT1, NIC1);
 231     nicEntity.setQuestionnaireData(loadFileToString("quesionnaire/validNic.json"));
 232     compositionEntityDataManager.addEntity(nicEntity, null);
 233
 234     doReturn(SIMPLE_SCHEMA)
 235         .when(compositionEntityDataManager)
 236         .generateSchema(SchemaTemplateContext.questionnaire, CompositionEntityType.vsp, null);
 237
 238     doReturn(loadFileToString("quesionnaire/schema/componentQuestionnaire.json"))
 239         .when(compositionEntityDataManager)
 240         .generateSchema(SchemaTemplateContext.questionnaire, CompositionEntityType.component, null);
 241
 242     doReturn(loadFileToString("quesionnaire/schema/nicQuestionnaire.json"))
 243         .when(compositionEntityDataManager)
 244         .generateSchema(SchemaTemplateContext.questionnaire, CompositionEntityType.nic, null);
 245
 246     Map<CompositionEntityId, Collection<String>> errorsById =
 247         compositionEntityDataManager.validateEntitiesQuestionnaire();
 248     Assert.assertEquals(errorsById.size(), 1);
 249   }
 250
 251   @Test(dependsOnMethods = "testNicAndComponentValidQuestionnaire")
 252   public void testComponentInvalidQuestionnaire() {
 253     ComponentEntity componentEntity = new ComponentEntity(VSP1, VERSION, COMPONENT1);
 254     componentEntity.setQuestionnaireData(loadFileToString("quesionnaire/invalidComponent.json"));
 255     compositionEntityDataManager.addEntity(componentEntity, null);
 256
 257     Map<CompositionEntityId, Collection<String>> errorsById =
 258         compositionEntityDataManager.validateEntitiesQuestionnaire();
 259     Assert.assertEquals(errorsById.size(), 2);
 260
 261     CompositionEntityId component = new ArrayList<>(errorsById.keySet()).get(1);
 262     List<String> errors = (List<String>) errorsById.get(component);
 263     Assert.assertEquals(errors.size(), 1);
 264     Assert.assertEquals(errors.get(0),
 265         "#/general/recovery/pointObjective: 20.0 is not lower or equal to 15");
 266   }
 267
 268   private static void assertValidationData(CompositionEntityValidationData validationData,
 269                                            String id, CompositionEntityType type,
 270                                            boolean hasErrors) {
 271     Assert.assertNotNull(validationData);
 272     Assert.assertEquals(validationData.getEntityId(), id);
 273     Assert.assertEquals(validationData.getEntityType(), type);
 274     Assert.assertTrue(CollectionUtils.isNotEmpty(validationData.getErrors()) == hasErrors);
 275   }
 276
 277   private static String loadFileToString(String path) {
 278     return new String(FileUtils.toByteArray(FileUtils.loadFileToInputStream(path)));
 279   }
 280 }