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