1 package org.openecomp.sdc.heat.services.tree;
3 import org.openecomp.sdc.heat.datatypes.structure.HeatStructureTree;
4 import org.openecomp.core.utilities.file.FileContentHandler;
5 import org.openecomp.core.utilities.file.FileUtils;
6 import org.testng.Assert;
7 import org.testng.annotations.Test;
10 import java.io.FileInputStream;
11 import java.io.IOException;
14 public class HeatTreeManagerTest {
17 public void testHeatTreeCreation() {
19 FileContentHandler fileContentMap = new FileContentHandler();
20 URL url = this.getClass().getResource("/heatTreeValidationOutput");
22 File templateDir = new File(url.getFile());
23 File[] files = templateDir.listFiles();
25 if (files == null || files.length == 0) {
29 for (File file : files) {
30 fileContentMap.addFile(file.getName(), getFileContent(file));
33 HeatTreeManager heatTreeManager = HeatTreeManagerUtil.initHeatTreeManager(fileContentMap);
34 heatTreeManager.createTree();
35 HeatStructureTree tree = heatTreeManager.getTree();
36 Assert.assertNotNull(tree);
37 Assert.assertEquals(tree.getHEAT().size(), 2);
40 private byte[] getFileContent(File file) {
42 return FileUtils.toByteArray(new FileInputStream(file));
43 } catch (IOException e) {