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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
package org.onap.schema;
import org.apache.commons.io.IOUtils;
import org.junit.Test;
import org.onap.crud.exception.CrudException;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Arrays;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.Comparator;
import java.util.concurrent.ConcurrentHashMap;
import java.util.HashMap;
import java.util.Map;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Collectors;
import static org.junit.Assert.*;
public class RelationshipSchemaTest {
final static Pattern rulesFilePattern = Pattern.compile("DbEdgeRules(.*).json");
final static Pattern propsFilePattern = Pattern.compile("edge_properties_(.*).json");
final static Pattern versionPattern = Pattern.compile(".*(v\\d+).json");
@Test
public void shouldLoadAllTheVersionsInDirectory() throws Exception {
Map<String, RelationshipSchema> versionContextMap = new ConcurrentHashMap<>();
loadRelations(versionContextMap);
assertTrue(versionContextMap.keySet().size() >= 0);
}
@Test
public void shouldContainValidTypes() throws Exception {
Map<String, RelationshipSchema> versionContextMap = new ConcurrentHashMap<>();
loadRelations(versionContextMap);
assertTrue(versionContextMap.get("v11").isValidType("groupsResourcesIn"));
assertTrue(versionContextMap.get("v11").isValidType("uses"));
assertFalse(versionContextMap.get("v11").isValidType("notValidType"));
}
@Test
public void shouldLookUpByRelation() throws Exception {
Map<String, RelationshipSchema> versionContextMap = new ConcurrentHashMap<>();
loadRelations(versionContextMap);
assertNotNull(versionContextMap.get("v11").lookupRelation("availability-zone:complex:groupsResourcesIn"));
assertTrue(versionContextMap.get("v11")
.lookupRelation("availability-zone:complex:groupsResourcesIn").containsKey("usesResource"));
}
@Test
public void shouldLookUpByRelationType() throws Exception {
Map<String, RelationshipSchema> versionContextMap = new ConcurrentHashMap<>();
loadRelations(versionContextMap);
assertNotNull(versionContextMap.get("v11").lookupRelationType("groupsResourcesIn"));
assertTrue(versionContextMap.get("v11")
.lookupRelation("availability-zone:complex:groupsResourcesIn").containsKey("usesResource"));
}
private void loadRelations(Map<String, RelationshipSchema> map){
ClassLoader classLoader = getClass().getClassLoader();
File dir = new File(classLoader.getResource("model").getFile());
File[] allFiles = dir.listFiles((d, name) ->
(propsFilePattern.matcher(name).matches() || rulesFilePattern.matcher(name).matches()));
Arrays.stream(allFiles).sorted(Comparator.comparing(File::getName))
.collect(Collectors.groupingBy(f -> myMatcher(versionPattern, f.getName())))
.forEach((e, f) -> map.put(e, jsonFilesLoader(f)));
}
private RelationshipSchema jsonFilesLoader (List<File> files) {
List<String> fileContents = new ArrayList<>();
RelationshipSchema rsSchema = null;
for (File f : files) {
fileContents.add(jsonToString(f));
}
try {
rsSchema = new RelationshipSchema(fileContents);
} catch (CrudException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
return rsSchema;
}
private String jsonToString (File file) {
InputStream inputStream = null;
String content = null;
HashMap<String,Object> result = null;
try {
inputStream = new FileInputStream(file);
content = IOUtils.toString(inputStream, "UTF-8");
} catch (IOException e) {
e.printStackTrace();
}
return content;
}
private String myMatcher (Pattern p, String s) {
Matcher m = p.matcher(s);
return m.matches() ? m.group(1) : "";
}
}
|
