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;
32 import org.apache.tinkerpop.gremlin.structure.Direction;
33 import org.junit.Rule;
34 import org.junit.Test;
35 import org.junit.rules.ExpectedException;
36 import org.onap.aai.AAISetup;
37 import org.onap.aai.edges.EdgeIngestor;
38 import org.onap.aai.edges.EdgeRule;
39 import org.onap.aai.edges.EdgeRuleQuery;
40 import org.onap.aai.edges.enums.EdgeType;
41 import org.onap.aai.edges.exceptions.AmbiguousRuleChoiceException;
42 import org.onap.aai.edges.exceptions.EdgeRuleNotFoundException;
43 import org.onap.aai.setup.SchemaVersion;
44 import org.springframework.beans.factory.annotation.Autowired;
46 public class EdgeRulesTest extends AAISetup {
48 // set thrown.expect to whatever a specific test needs
49 // this establishes a default of expecting no exceptions to be thrown
51 public ExpectedException thrown = ExpectedException.none();
54 private EdgeIngestor edgeIngestor;
57 public void verifyOutDirection() throws EdgeRuleNotFoundException, AmbiguousRuleChoiceException {
59 EdgeRuleQuery ruleQuery = new EdgeRuleQuery.Builder("cloud-region", "flavor").edgeType(EdgeType.TREE).build();
61 EdgeRule rule = edgeIngestor.getRule(ruleQuery);
63 assertEquals("out direction", rule.getDirection(), Direction.IN);
67 public void verifyOutFlippedDirection() throws EdgeRuleNotFoundException, AmbiguousRuleChoiceException {
69 EdgeRuleQuery ruleQuery = new EdgeRuleQuery.Builder("flavor", "cloud-region").edgeType(EdgeType.TREE).build();
71 EdgeRule rule = edgeIngestor.getRule(ruleQuery);
73 assertEquals("in direction", rule.getDirection(), Direction.OUT);
77 public void verifyMultipleGet() throws EdgeRuleNotFoundException {
79 EdgeRuleQuery ruleQuery =
80 new EdgeRuleQuery.Builder("model-element", "model-ver").edgeType(EdgeType.TREE).build();
82 Multimap<String, EdgeRule> ruleMap = edgeIngestor.getRules(ruleQuery);
84 for (EdgeRule edgeRule : ruleMap.get("model|model-ver")) {
85 assertEquals("has isA rule", "org.onap.relationships.inventory.IsA", edgeRule.getLabel());
91 public void verifyAllRules() throws EdgeRuleNotFoundException {
93 // This will cause every rule in the real json files to be verified
94 // so if any required properties are missing, the verification builds
95 // will catch it and incorrect rules can't get merged in.
96 for (SchemaVersion v : schemaVersions.getVersions()) {
97 Multimap<String, EdgeRule> all = edgeIngestor.getAllRules(schemaVersions.getDefaultVersion());
99 // this part verifies the default properties
100 // 1) can have only at most 1 containment edge between same node type pair
101 // 2) if there is at least 1 cousin edge, there must be exactly 1 cousin edge with default=true
102 for (String key : all.keySet()) {
104 Collection<EdgeRule> edgeRuleCollection = all.get(key);
106 boolean foundContainment = false; // can have at most 1 containment rel btwn same pair of node types
107 boolean foundCousin = false;
108 boolean cousinDefault = false; // if there is a cousin edge there must be at least 1 default cousin
110 Set<String> labels = new HashSet<>(); // all edges between the same pair must have different labels
113 for (EdgeRule rule : edgeRuleCollection) {
114 EdgeRule match = rule;
116 if (!("NONE".equals(match.getContains()))) {
117 if (foundContainment) {
118 fail("more than one containment edge defined for " + v.toString() + " " + key);
120 foundContainment = true;
122 } else { // check cousin stuff
125 if (match.isDefault()) {
126 if (!cousinDefault) {
127 cousinDefault = true;
129 fail("more than one cousin edge defined as default for " + v.toString() + " " + key);
135 String label = match.getLabel();
136 if (labels.contains(label)) {
137 fail("same label found for multiple edges for " + v.toString() + " " + key);
142 if (foundCousin && !cousinDefault && cousinCount > 1) {
143 fail("there is at least one cousin edge but none are designated the default for " + v.toString()