ui-ci/src/main/java/org/openecomp/sdc/ci/tests/businesslogic/ArtifactBusinessLogic.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.ci.tests.businesslogic;
  22
  23 import static org.junit.Assert.assertTrue;
  24
  25 import java.io.File;
  26 import java.io.IOException;
  27 import java.util.ArrayList;
  28 import java.util.HashMap;
  29 import java.util.LinkedList;
  30 import java.util.List;
  31 import java.util.Map;
  32 import java.util.Map.Entry;
  33
  34 import org.openecomp.sdc.be.datatypes.elements.HeatParameterDataDefinition;
  35 import org.openecomp.sdc.be.model.ArtifactDefinition;
  36 import org.openecomp.sdc.ci.tests.datatypes.HeatAndHeatEnvNamesPair;
  37 import org.openecomp.sdc.ci.tests.datatypes.HeatMetaFirstLevelDefinition;
  38 import org.openecomp.sdc.ci.tests.datatypes.HeatWithParametersDefinition;
  39 import org.openecomp.sdc.ci.tests.datatypes.enums.ArtifactTypeEnum;
  40 import org.openecomp.sdc.ci.tests.execute.devCI.ArtifactFromCsar;
  41 import org.openecomp.sdc.ci.tests.pages.HomePage;
  42 import org.openecomp.sdc.ci.tests.utilities.FileHandling;
  43
  44 import com.clearspring.analytics.util.Pair;
  45
  46 public class ArtifactBusinessLogic {
  47
  48         private final static String[] okFileExtensions = new String[] {"yaml", "yml", "env"};
  49         private static final String PARAMETERS = "parameters";
  50         private static final String DEPLOYMENT = "Deployment";
  51
  52         public static synchronized Map<String, File> createEnvFilesListFromCsar(String vspName, String filePath) throws Exception{
  53                 Map<String, File> generatedEnvFiles = new HashMap<>();
  54                 File csarFile= HomePage.downloadVspCsarToDefaultDirectory(vspName);
  55                 FileHandling.unzip(csarFile.toString(), filePath);
  56                 List<File> yamlList = getHeatFilesCreatedFromCsar(csarFile, filePath);
  57                 Map<String, HeatAndHeatEnvNamesPair> filesPairMap = getFilesPairMap(yamlList);
  58                 generatedEnvFiles = generateDefaultEnvFiles(filesPairMap, filePath);
  59                 return generatedEnvFiles;
  60         }
  61
  62         public static synchronized List<File> getHeatFilesCreatedFromCsar(File pathToDirectory, String filePath) throws Exception {
  63                 List<File> fileList = new ArrayList<>();
  64                 String artifactsFilePath = filePath + "Artifacts" + File.separator;
  65                 List<File> fileListFromArtifactsDirectory = FileHandling.getHeatAndHeatEnvArtifactsFromZip(new File(artifactsFilePath), okFileExtensions);
  66                 Map<String, Object> combinedMap = ArtifactFromCsar.combineHeatArtifacstWithFolderArtifacsToMap(pathToDirectory.toString());
  67                 LinkedList<HeatMetaFirstLevelDefinition> deploymentArtifacts = ((LinkedList<HeatMetaFirstLevelDefinition>) combinedMap.get(DEPLOYMENT));
  68                 for(HeatMetaFirstLevelDefinition deploymentArtifact : deploymentArtifacts){
  69                         String type = deploymentArtifact.getType();
  70                         if(type.equals(ArtifactTypeEnum.HEAT.getType()) ||
  71                            type.equals(ArtifactTypeEnum.HEAT_ENV.getType()) ||
  72                            type.equals(ArtifactTypeEnum.HEAT_VOL.getType()) ||
  73                            type.equals(ArtifactTypeEnum.HEAT_NET.getType())){
  74                                 File file = (new File(artifactsFilePath + deploymentArtifact.getFileName()));
  75                                 if(fileListFromArtifactsDirectory.contains(file)){
  76                                         fileList.add(file);
  77                                 }else{
  78                                         assertTrue("File " + file + " does not exist", false);
  79                                 }
  80                         }
  81                 }
  82                 return fileList;
  83         }
  84         public static synchronized Map<String, HeatAndHeatEnvNamesPair> getFilesPairMap(List<File> generatedEnvFiles) {
  85
  86                 Map<String, HeatAndHeatEnvNamesPair> heatAndHeatEnvPairs= new HashMap<>();
  87                 for(File file : generatedEnvFiles){
  88                         String[] fileName = file.getName().split("\\.");
  89                         String currentKey = fileName[0];
  90                         String currentExtension = fileName[1];
  91                         HeatAndHeatEnvNamesPair pair;
  92                         if(!heatAndHeatEnvPairs.containsKey(currentKey)){
  93                                 pair = new HeatAndHeatEnvNamesPair();
  94                                 heatAndHeatEnvPairs.put(currentKey, pair);
  95                         }else{
  96                                 pair = heatAndHeatEnvPairs.get(currentKey);
  97                         }
  98                         setFileToPair(file, currentExtension, pair);
  99                 }
 100                 return heatAndHeatEnvPairs;
 101         }
 102
 103         /**
 104          * The method fill list of HeatWithParametersDefinition parameters
 105          * @param deploymentArtifacts
 106          * @return
 107          */
 108         public static synchronized List<HeatWithParametersDefinition> extractHeatWithParametersDefinition(Map<String, ArtifactDefinition> deploymentArtifacts) {
 109
 110                 List<HeatWithParametersDefinition> heatAndEnvLabelList = new ArrayList<>();
 111
 112                 for (Entry<String, ArtifactDefinition> artifactDefinitionChild : deploymentArtifacts.entrySet()){
 113                         if(artifactDefinitionChild.getValue().getArtifactType().equals(ArtifactTypeEnum.HEAT_ENV.getType())){
 114                                 for(Entry<String, ArtifactDefinition> artifactDefinitionParent : deploymentArtifacts.entrySet()){
 115                                         if(artifactDefinitionChild.getValue().getGeneratedFromId().equals(artifactDefinitionParent.getValue().getUniqueId())){
 116                                                 String heatLabel = artifactDefinitionParent.getValue().getArtifactLabel();
 117                                                 String heatArtifactType = artifactDefinitionParent.getValue().getArtifactType();
 118                                                 String heatArtifactDisplayName = artifactDefinitionParent.getValue().getArtifactDisplayName();
 119                                                 List<HeatParameterDataDefinition> heatParameterDefinition = artifactDefinitionParent.getValue().getHeatParameters();
 120                                                 String heatEnvLabel = artifactDefinitionChild.getValue().getArtifactLabel();
 121                                                 String heatEnvArtifactType = artifactDefinitionChild.getValue().getArtifactType();
 122                                                 heatAndEnvLabelList.add(new HeatWithParametersDefinition(heatLabel, heatEnvLabel, heatArtifactType, heatEnvArtifactType, heatArtifactDisplayName, heatParameterDefinition));
 123                                                 break;
 124                                         }
 125                                 }
 126                         }
 127                 }
 128                 return heatAndEnvLabelList;
 129         }
 130
 131
 132         public static synchronized void setFileToPair(File file, String currentExtension, HeatAndHeatEnvNamesPair pair) {
 133                 if(!currentExtension.equals("env")){
 134                         pair.setHeatFileName(file);
 135                 }else{
 136                         pair.setHeatEnvFileName(file);
 137                 }
 138         }
 139
 140         public static synchronized Map<String, File> generateDefaultEnvFiles(Map<String, HeatAndHeatEnvNamesPair> filesPairMap, String filePath) throws Exception {
 141
 142                 Map<String, File> generatedEnvFilesMap = new HashMap<>();
 143                 for(Entry<String, HeatAndHeatEnvNamesPair> pair : filesPairMap.entrySet()){
 144                         Map<String, Pair<String, Object>> envParametersMap = getEnvParametersMap(pair);
 145                         File generatedEnvFile = createEnvFile(envParametersMap, new File(filePath + pair.getKey() + ".env"));
 146                         generatedEnvFilesMap.put(pair.getKey(), generatedEnvFile);
 147                 }
 148                 return generatedEnvFilesMap;
 149         }
 150
 151         public static synchronized File createEnvFile(Map<String, Pair<String, Object>> envParametersMap, File fileToWrite) throws IOException {
 152
 153                 FileHandling.writeToFile(fileToWrite, PARAMETERS+":", 0);
 154                 FileHandling.writeToFile(fileToWrite, envParametersMap, 2);
 155                 return fileToWrite;
 156         }
 157
 158         public static synchronized Map<String, Pair<String, Object>> getEnvParametersMap(Entry<String, HeatAndHeatEnvNamesPair> pair) throws Exception {
 159                 File heatFileName = pair.getValue().getHeatFileName();
 160                 File heatEnvFileName = pair.getValue().getHeatEnvFileName();
 161                 Map<String, Pair<String, Object>> envParametersMap = new HashMap<>();
 162                 fillParametersMapFromHeatFile(heatFileName, envParametersMap);
 163                 fillParametersMapFromHeatEnvFile(heatEnvFileName, envParametersMap);
 164                 return envParametersMap;
 165         }
 166
 167         public static synchronized void fillParametersMapFromHeatEnvFile(File heatEnvFileName, Map<String, Pair<String, Object>> envParametersMap) throws Exception {
 168                 if(heatEnvFileName != null){
 169                         Map<String, Object> mapHeatEnvFileParameters = FileHandling.parseYamlFileToMapByPattern(heatEnvFileName, PARAMETERS);
 170                         for (Map.Entry<String, Object> parameter : mapHeatEnvFileParameters.entrySet()){
 171                                 String key = parameter.getKey();
 172                                 Pair<String, Object> pair;
 173                                 if(envParametersMap.containsKey(key)){
 174                                         if(envParametersMap.get(key).left.equals("string") && parameter.getValue() != null){
 175                                                 pair = Pair.create(envParametersMap.get(key).left, "\"" + parameter.getValue() + "\"");
 176                                         }else if(envParametersMap.get(key).left.equals("string") && parameter.getValue() == null){
 177                                                 pair = Pair.create(envParametersMap.get(key).left, "");
 178                                         }else if(parameter.getValue() == null){
 179                                                 pair = Pair.create(envParametersMap.get(key).left, "");
 180                                         }else{
 181                                                 pair = Pair.create(envParametersMap.get(key).left, parameter.getValue());
 182                                         }
 183                                         envParametersMap.put(key, pair);
 184                                 }
 185                         }
 186                 }
 187         }
 188
 189         @SuppressWarnings("unchecked")
 190         public static synchronized void fillParametersMapFromHeatFile(File heatFileName, Map<String, Pair<String, Object>> envParametersMap) throws Exception {
 191                 Map<String, Object> mapHeatFileParameters = FileHandling.parseYamlFileToMapByPattern(heatFileName, PARAMETERS);
 192                 for (Map.Entry<String, Object> parameter : mapHeatFileParameters.entrySet()){
 193                         Map<String, Object> value = (Map<String, Object>) parameter.getValue();
 194                         Pair<String, Object> pair;
 195                         if(value.get("type").toString().equals("string") && value.get("default") != null ){
 196                                 pair = Pair.create(value.get("type").toString(), "\"" + value.get("default") + "\"");
 197                         }else if(value.get("type").toString().equals("string") && value.get("default") == null){
 198                                 pair = Pair.create(value.get("type").toString(), "");
 199                         }else if(value.get("default") == null){
 200                                 pair = Pair.create(value.get("type").toString(), "");
 201                         }else{
 202                                 pair = Pair.create(value.get("type").toString(), value.get("default"));
 203                         }
 204                         envParametersMap.put(parameter.getKey(), pair);
 205                 }
 206         }
 207
 208 }