openecomp-be/backend/openecomp-sdc-vendor-software-product-manager/src/test/java/org/openecomp/sdc/vendorsoftwareproduct/tree/TreeBaseTest.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.tree;
  22
  23 import org.openecomp.core.utilities.file.FileUtils;
  24 import org.openecomp.sdc.heat.services.tree.HeatTreeManager;
  25
  26 import java.io.File;
  27 import java.net.URISyntaxException;
  28 import java.net.URL;
  29
  30 /**
  31  * Created by SHALOMB on 6/8/2016.
  32  */
  33 public class TreeBaseTest {
  34
  35   String INPUT_DIR;
  36
  37
  38   HeatTreeManager initHeatTreeManager() {
  39     HeatTreeManager heatTreeManager = new HeatTreeManager();
  40
  41     URL url = Thread.currentThread().getContextClassLoader().getResource(INPUT_DIR);
  42
  43     File inputDir = null;
  44     try {
  45       inputDir = new File(url.toURI());
  46     } catch (URISyntaxException exception) {
  47       exception.printStackTrace();
  48     }
  49     File[] files = inputDir.listFiles();
  50     for (File inputFile : files) {
  51       try {
  52         heatTreeManager.addFile(inputFile.getName(), FileUtils.loadFileToInputStream(
  53             INPUT_DIR.replace("/", File.separator) + File.separator + inputFile.getName()));
  54       } catch (Exception e) {
  55         e.printStackTrace();
  56         throw e;
  57       }
  58     }
  59     return heatTreeManager;
  60   }
  61 }