2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright © 2017-2018 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 package org.onap.aai.edges;
22 import static org.junit.Assert.*;
24 import java.io.InputStream;
25 import java.util.List;
26 import java.util.Scanner;
28 import org.junit.Before;
29 import org.junit.Test;
30 import org.onap.aai.edges.enums.EdgeType;
32 import com.jayway.jsonpath.DocumentContext;
33 import com.jayway.jsonpath.JsonPath;
35 public class EdgeRuleQueryTest {
36 private DocumentContext testRules;
37 private String readStart = "$.rules.[?]";
39 /* **** DATA SETUP **** */
42 InputStream is = getClass().getResourceAsStream("/edgeRules/test.json");
44 Scanner scanner = new Scanner(is);
45 String json = scanner.useDelimiter("\\Z").next();
48 this.testRules = JsonPath.parse(json);
53 public void testFromToSingle() {
54 //rule defined from quux to foo
55 EdgeRuleQuery q = new EdgeRuleQuery.Builder("quux", "foo").build();
56 List<Object> results = testRules.read(readStart, q.getFilter());
58 assertTrue(results.size() == 1);
62 public void testToFromSingle() {
63 //rule defined from quux to foo, this is flipped
64 EdgeRuleQuery q = new EdgeRuleQuery.Builder("foo", "quux").build();
65 List<Object> results = testRules.read(readStart, q.getFilter());
67 assertTrue(results.size() == 1);
71 public void testFromToMultiple() {
72 //rules have two from foo to bar
73 EdgeRuleQuery q = new EdgeRuleQuery.Builder("foo", "bar").build();
74 List<Object> results = testRules.read(readStart, q.getFilter());
75 assertTrue(results.size() == 2);
79 public void testToFromMultiple() {
80 //rules have two from foo to bar
81 EdgeRuleQuery q = new EdgeRuleQuery.Builder("bar", "foo").build();
82 List<Object> results = testRules.read(readStart, q.getFilter());
84 assertTrue(results.size() == 2);
85 assertTrue(!(results.get(0).toString().equals(results.get(1).toString())));
89 public void testJustFrom() {
90 //there are 4 foo rules (foo>bar, foo>bar, foo>baz, quux>foo)
91 EdgeRuleQuery q = new EdgeRuleQuery.Builder("foo").build();
92 List<Object> results = testRules.read(readStart, q.getFilter());
93 assertTrue(results.size() == 4);
95 //there are 2 bar rules
96 EdgeRuleQuery q2 = new EdgeRuleQuery.Builder("bar").build();
97 List<Object> results2 = testRules.read(readStart, q2.getFilter());
98 assertTrue(results2.size() == 2);
102 public void testWithLabel() {
103 //there are foo >eats> bar and foo >eatz> bar
104 EdgeRuleQuery q = new EdgeRuleQuery.Builder("foo","bar").label("eatz").build();
105 List<Object> results = testRules.read(readStart, q.getFilter());
106 assertTrue(results.size() == 1);
107 assertTrue(results.get(0).toString().contains("eatz"));
111 public void testCousin() {
112 EdgeRuleQuery q = new EdgeRuleQuery.Builder("foo").edgeType(EdgeType.COUSIN).build();
113 List<Object> results = testRules.read(readStart, q.getFilter());
114 assertTrue(results.size() == 2);
116 EdgeRuleQuery q2 = new EdgeRuleQuery.Builder("foo", "bar").edgeType(EdgeType.COUSIN).label("eats").build();
117 List<Object> results2 = testRules.read(readStart, q2.getFilter());
118 assertTrue(results2.size() == 1);
119 assertTrue(results2.get(0).toString().contains("eats"));
121 EdgeRuleQuery q3 = new EdgeRuleQuery.Builder("foo","quux").edgeType(EdgeType.COUSIN).build();
122 List<Object> results3 = testRules.read(readStart, q3.getFilter());
123 assertTrue(results3.isEmpty());
127 public void testTree() {
128 EdgeRuleQuery q = new EdgeRuleQuery.Builder("foo").edgeType(EdgeType.TREE).build();
129 List<Object> results = testRules.read(readStart, q.getFilter());
130 assertTrue(results.size() == 2);