2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright © 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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
20 * ECOMP is a trademark and service mark of AT&T Intellectual Property.
23 package org.onap.aai.serialization.db;
25 import static junit.framework.TestCase.fail;
26 import static org.junit.Assert.assertEquals;
28 import com.google.common.collect.Multimap;
31 import java.util.stream.Collectors;
33 import org.apache.tinkerpop.gremlin.structure.Direction;
34 import org.junit.Rule;
35 import org.junit.Test;
36 import org.junit.rules.ExpectedException;
37 import org.onap.aai.AAISetup;
38 import org.onap.aai.edges.EdgeIngestor;
39 import org.onap.aai.edges.EdgeRule;
40 import org.onap.aai.edges.EdgeRuleQuery;
41 import org.onap.aai.edges.enums.EdgeType;
42 import org.onap.aai.edges.exceptions.AmbiguousRuleChoiceException;
43 import org.onap.aai.edges.exceptions.EdgeRuleNotFoundException;
44 import org.onap.aai.setup.SchemaVersion;
45 import org.springframework.beans.factory.annotation.Autowired;
47 public class EdgeRulesTest extends AAISetup {
49 // set thrown.expect to whatever a specific test needs
50 // this establishes a default of expecting no exceptions to be thrown
52 public ExpectedException thrown = ExpectedException.none();
55 private EdgeIngestor edgeIngestor;
58 public void verifyOutDirection() throws EdgeRuleNotFoundException, AmbiguousRuleChoiceException {
60 EdgeRuleQuery ruleQuery = new EdgeRuleQuery.Builder("cloud-region", "flavor").edgeType(EdgeType.TREE).build();
62 EdgeRule rule = edgeIngestor.getRule(ruleQuery);
64 assertEquals("out direction", rule.getDirection(), Direction.IN);
68 public void verifyOutFlippedDirection() throws EdgeRuleNotFoundException, AmbiguousRuleChoiceException {
70 EdgeRuleQuery ruleQuery = new EdgeRuleQuery.Builder("flavor", "cloud-region").edgeType(EdgeType.TREE).build();
72 EdgeRule rule = edgeIngestor.getRule(ruleQuery);
74 assertEquals("in direction", rule.getDirection(), Direction.OUT);
78 public void verifyMultipleGet() throws EdgeRuleNotFoundException {
80 EdgeRuleQuery ruleQuery =
81 new EdgeRuleQuery.Builder("model-element", "model-ver").edgeType(EdgeType.TREE).build();
83 Multimap<String, EdgeRule> ruleMap = edgeIngestor.getRules(ruleQuery);
85 for (EdgeRule edgeRule : ruleMap.get("model|model-ver")) {
86 assertEquals("has isA rule", "org.onap.relationships.inventory.IsA", edgeRule.getLabel());
92 public void verifyAllRules() throws EdgeRuleNotFoundException {
94 // This will cause every rule in the real json files to be verified
95 // so if any required properties are missing, the verification builds
96 // will catch it and incorrect rules can't get merged in.
97 for (SchemaVersion v : schemaVersions.getVersions()) {
98 Multimap<String, EdgeRule> all = edgeIngestor.getAllRules(schemaVersions.getDefaultVersion());
100 // this part verifies the default properties
101 // 1) can have only at most 1 containment edge between same node type pair
102 // 2) if there is at least 1 cousin edge, there must be exactly 1 cousin edge with default=true
103 for (String key : all.keySet()) {
105 Collection<EdgeRule> edgeRuleCollection = all.get(key);
107 boolean foundContainment = false; // can have at most 1 containment rel btwn same pair of node types
108 boolean foundCousin = false;
109 boolean cousinDefault = false; // if there is a cousin edge there must be at least 1 default cousin
111 Set<String> labels = new HashSet<>(); // all edges between the same pair must have different labels
114 for (EdgeRule rule : edgeRuleCollection) {
115 EdgeRule match = rule;
117 if (!("NONE".equals(match.getContains()))) {
118 if (foundContainment) {
119 fail("more than one containment edge defined for " + v.toString() + " " + key);
121 foundContainment = true;
123 } else { // check cousin stuff
126 if (match.isDefault()) {
127 if (!cousinDefault) {
128 cousinDefault = true;
130 fail("more than one cousin edge defined as default for " + v.toString() + " " + key);
136 String label = match.getLabel();
137 if (labels.contains(label)) {
138 fail("same label found for multiple edges for " + v.toString() + " " + key);
143 if (foundCousin && !cousinDefault && cousinCount > 1) {
144 fail("there is at least one cousin edge but none are designated the default for " + v.toString()