2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright © 2017-18 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=========================================================
21 package org.onap.aai.edges;
23 import static org.junit.Assert.*;
25 import com.jayway.jsonpath.DocumentContext;
26 import com.jayway.jsonpath.JsonPath;
28 import java.io.InputStream;
29 import java.util.List;
30 import java.util.Scanner;
32 import org.junit.Before;
33 import org.junit.Test;
34 import org.onap.aai.edges.enums.EdgeType;
36 public class EdgeRuleQueryTest {
37 private DocumentContext testRules;
38 private String readStart = "$.rules.[?]";
40 /* **** DATA SETUP **** */
43 InputStream is = getClass().getResourceAsStream("/edgeRules/test.json");
45 Scanner scanner = new Scanner(is);
46 String json = scanner.useDelimiter("\\Z").next();
49 this.testRules = JsonPath.parse(json);
54 public void testFromToSingle() {
55 // rule defined from quux to foo
56 EdgeRuleQuery q = new EdgeRuleQuery.Builder("quux", "foo").build();
57 List<Object> results = testRules.read(readStart, q.getFilter());
59 assertTrue(results.size() == 1);
63 public void testToFromSingle() {
64 // rule defined from quux to foo, this is flipped
65 EdgeRuleQuery q = new EdgeRuleQuery.Builder("foo", "quux").build();
66 List<Object> results = testRules.read(readStart, q.getFilter());
68 assertTrue(results.size() == 1);
72 public void testFromToMultiple() {
73 // rules have two from foo to bar
74 EdgeRuleQuery q = new EdgeRuleQuery.Builder("foo", "bar").build();
75 List<Object> results = testRules.read(readStart, q.getFilter());
76 assertTrue(results.size() == 2);
80 public void testToFromMultiple() {
81 // rules have two from foo to bar
82 EdgeRuleQuery q = new EdgeRuleQuery.Builder("bar", "foo").build();
83 List<Object> results = testRules.read(readStart, q.getFilter());
85 assertTrue(results.size() == 2);
86 assertTrue(!(results.get(0).toString().equals(results.get(1).toString())));
90 public void testJustFrom() {
91 // there are 4 foo rules (foo>bar, foo>bar, foo>baz, quux>foo)
92 EdgeRuleQuery q = new EdgeRuleQuery.Builder("foo").build();
93 List<Object> results = testRules.read(readStart, q.getFilter());
94 assertTrue(results.size() == 4);
96 // there are 2 bar rules
97 EdgeRuleQuery q2 = new EdgeRuleQuery.Builder("bar").build();
98 List<Object> results2 = testRules.read(readStart, q2.getFilter());
99 assertTrue(results2.size() == 2);
103 public void testWithLabel() {
104 // there are foo >eats> bar and foo >eatz> bar
105 EdgeRuleQuery q = new EdgeRuleQuery.Builder("foo", "bar").label("eatz").build();
106 List<Object> results = testRules.read(readStart, q.getFilter());
107 assertTrue(results.size() == 1);
108 assertTrue(results.get(0).toString().contains("eatz"));
112 public void testCousin() {
113 EdgeRuleQuery q = new EdgeRuleQuery.Builder("foo").edgeType(EdgeType.COUSIN).build();
114 List<Object> results = testRules.read(readStart, q.getFilter());
115 assertTrue(results.size() == 2);
117 EdgeRuleQuery q2 = new EdgeRuleQuery.Builder("foo", "bar").edgeType(EdgeType.COUSIN).label("eats").build();
118 List<Object> results2 = testRules.read(readStart, q2.getFilter());
119 assertTrue(results2.size() == 1);
120 assertTrue(results2.get(0).toString().contains("eats"));
122 EdgeRuleQuery q3 = new EdgeRuleQuery.Builder("foo", "quux").edgeType(EdgeType.COUSIN).build();
123 List<Object> results3 = testRules.read(readStart, q3.getFilter());
124 assertTrue(results3.isEmpty());
128 public void testTree() {
129 EdgeRuleQuery q = new EdgeRuleQuery.Builder("foo").edgeType(EdgeType.TREE).build();
130 List<Object> results = testRules.read(readStart, q.getFilter());
131 assertTrue(results.size() == 2);