openecomp-be/backend/openecomp-sdc-vendor-software-product-manager/src/test/java/org/openecomp/sdc/vendorsoftwareproduct/tree/TreeBaseTest.java

   1 package org.openecomp.sdc.vendorsoftwareproduct.tree;
   2
   3 import org.openecomp.sdc.heat.services.tree.HeatTreeManager;
   4 import org.openecomp.core.utilities.file.FileUtils;
   5
   6 import java.io.File;
   7 import java.net.URISyntaxException;
   8 import java.net.URL;
   9
  10 public class TreeBaseTest {
  11
  12   String INPUT_DIR;
  13
  14
  15   HeatTreeManager initHeatTreeManager() {
  16     HeatTreeManager heatTreeManager = new HeatTreeManager();
  17
  18     URL url = TreeBaseTest.class.getResource(INPUT_DIR);
  19
  20     File inputDir = null;
  21     try {
  22       inputDir = new File(url.toURI());
  23     } catch (URISyntaxException e) {
  24       e.printStackTrace();
  25     }
  26     File[] files = inputDir.listFiles();
  27     for (File inputFile : files) {
  28       heatTreeManager.addFile(inputFile.getName(), FileUtils.loadFileToInputStream(
  29           INPUT_DIR.replace("/", File.separator) + File.separator + inputFile.getName()));
  30     }
  31     return heatTreeManager;
  32   }
  33 }