push addional code
[sdc.git] / openecomp-be / lib / openecomp-sdc-translator-lib / openecomp-sdc-translator-core / src / test / java / org / openecomp / sdc / translator / TestUtils.java
1 package org.openecomp.sdc.translator;
2
3 import org.openecomp.sdc.common.utils.AsdcCommon;
4 import org.openecomp.core.translator.api.HeatToToscaTranslator;
5 import org.openecomp.core.utilities.file.FileUtils;
6 import org.junit.Assert;
7
8 import java.io.*;
9
10 public class TestUtils {
11   private static final String MANIFEST_NAME = AsdcCommon.MANIFEST_NAME;
12   private static String zipFilename = "VSP.zip";
13   private static String validationFilename = "validationOutput.json";
14
15   private TestUtils() {
16   }
17
18
19   public static void addFilesToTranslator(HeatToToscaTranslator heatToToscaTranslator, String path)
20       throws IOException {
21     File manifestFile = new File(path);
22     File[] files = manifestFile.listFiles();
23     byte[] fileContent;
24
25     Assert.assertNotNull("manifest files is empty", files);
26
27     for (File file : files) {
28
29       try (FileInputStream fis = new FileInputStream(file)) {
30
31         fileContent = FileUtils.toByteArray(fis);
32
33         if (file.getName().equals(MANIFEST_NAME)) {
34           heatToToscaTranslator.addManifest(MANIFEST_NAME, fileContent);
35         } else {
36           if (!file.getName().equals(zipFilename) && (!file.getName().equals(validationFilename))) {
37             heatToToscaTranslator.addFile(file.getName(), fileContent);
38           }
39         }
40       }
41     }
42   }
43 }