blob: 6a9f8df12d7c16893a47f0e6d084af97ef1652c8 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
package org.openecomp.sdc.heat.services.tree;
import org.openecomp.sdc.heat.datatypes.structure.HeatStructureTree;
import org.openecomp.core.utilities.file.FileContentHandler;
import org.openecomp.core.utilities.file.FileUtils;
import org.testng.Assert;
import org.testng.annotations.Test;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.net.URL;
public class HeatTreeManagerTest {
@Test
public void testHeatTreeCreation() {
FileContentHandler fileContentMap = new FileContentHandler();
URL url = this.getClass().getResource("/heatTreeValidationOutput");
File templateDir = new File(url.getFile());
File[] files = templateDir.listFiles();
if (files == null || files.length == 0) {
return;
}
for (File file : files) {
fileContentMap.addFile(file.getName(), getFileContent(file));
}
HeatTreeManager heatTreeManager = HeatTreeManagerUtil.initHeatTreeManager(fileContentMap);
heatTreeManager.createTree();
HeatStructureTree tree = heatTreeManager.getTree();
Assert.assertNotNull(tree);
Assert.assertEquals(tree.getHEAT().size(), 2);
}
private byte[] getFileContent(File file) {
try {
return FileUtils.toByteArray(new FileInputStream(file));
} catch (IOException e) {
e.printStackTrace();
}
return new byte[0];
}
}
|