aai-schema-ingest/src/test/java/org/onap/aai/edges/EdgeRuleQueryTest.java

   1 /**
   2  * ============LICENSE_START=======================================================
   3  * org.onap.aai
   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
  10  *
  11  *   http://www.apache.org/licenses/LICENSE-2.0
  12  *
  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=========================================================
  19  */
  20
  21 package org.onap.aai.edges;
  22
  23 import static org.junit.Assert.*;
  24
  25 import java.io.InputStream;
  26 import java.util.List;
  27 import java.util.Scanner;
  28
  29 import org.junit.Before;
  30 import org.junit.Test;
  31 import org.onap.aai.edges.enums.EdgeType;
  32
  33 import com.jayway.jsonpath.DocumentContext;
  34 import com.jayway.jsonpath.JsonPath;
  35
  36
  37 public class EdgeRuleQueryTest {
  38     private DocumentContext testRules;
  39     private String readStart = "$.rules.[?]";
  40
  41     /* **** DATA SETUP **** */
  42     @Before
  43     public void setup() {
  44         InputStream is = getClass().getResourceAsStream("/edgeRules/test.json");
  45
  46         Scanner scanner = new Scanner(is);
  47         String json = scanner.useDelimiter("\\Z").next();
  48         scanner.close();
  49
  50         this.testRules = JsonPath.parse(json);
  51     }
  52
  53     /* **** TESTS **** */
  54     @Test
  55     public void testFromToSingle() {
  56         //rule defined from quux to foo
  57         EdgeRuleQuery q = new EdgeRuleQuery.Builder("quux", "foo").build();
  58         List<Object> results = testRules.read(readStart, q.getFilter());
  59
  60         assertTrue(results.size() == 1);
  61     }
  62
  63     @Test
  64     public void testToFromSingle() {
  65         //rule defined from quux to foo, this is flipped
  66         EdgeRuleQuery q = new EdgeRuleQuery.Builder("foo", "quux").build();
  67         List<Object> results = testRules.read(readStart, q.getFilter());
  68
  69         assertTrue(results.size() == 1);
  70     }
  71
  72     @Test
  73     public void testFromToMultiple() {
  74         //rules have two from foo to bar
  75         EdgeRuleQuery q = new EdgeRuleQuery.Builder("foo", "bar").build();
  76         List<Object> results = testRules.read(readStart, q.getFilter());
  77         assertTrue(results.size() == 2);
  78     }
  79
  80     @Test
  81     public void testToFromMultiple() {
  82         //rules have two from foo to bar
  83         EdgeRuleQuery q = new EdgeRuleQuery.Builder("bar", "foo").build();
  84         List<Object> results = testRules.read(readStart, q.getFilter());
  85
  86         assertTrue(results.size() == 2);
  87         assertTrue(!(results.get(0).toString().equals(results.get(1).toString())));
  88     }
  89
  90     @Test
  91     public void testJustFrom() {
  92         //there are 4 foo rules (foo>bar, foo>bar, foo>baz, quux>foo)
  93         EdgeRuleQuery q = new EdgeRuleQuery.Builder("foo").build();
  94         List<Object> results = testRules.read(readStart, q.getFilter());
  95         assertTrue(results.size() == 4);
  96
  97         //there are 2 bar rules
  98         EdgeRuleQuery q2 = new EdgeRuleQuery.Builder("bar").build();
  99         List<Object> results2 = testRules.read(readStart, q2.getFilter());
 100         assertTrue(results2.size() == 2);
 101     }
 102
 103     @Test
 104     public void testWithLabel() {
 105         //there are foo >eats> bar and foo >eatz> bar
 106         EdgeRuleQuery q = new EdgeRuleQuery.Builder("foo","bar").label("eatz").build();
 107         List<Object> results = testRules.read(readStart, q.getFilter());
 108         assertTrue(results.size() == 1);
 109         assertTrue(results.get(0).toString().contains("eatz"));
 110     }
 111
 112     @Test
 113     public void testCousin() {
 114         EdgeRuleQuery q = new EdgeRuleQuery.Builder("foo").edgeType(EdgeType.COUSIN).build();
 115         List<Object> results = testRules.read(readStart, q.getFilter());
 116         assertTrue(results.size() == 2);
 117
 118         EdgeRuleQuery q2 = new EdgeRuleQuery.Builder("foo", "bar").edgeType(EdgeType.COUSIN).label("eats").build();
 119         List<Object> results2 = testRules.read(readStart, q2.getFilter());
 120         assertTrue(results2.size() == 1);
 121         assertTrue(results2.get(0).toString().contains("eats"));
 122
 123         EdgeRuleQuery q3 = new EdgeRuleQuery.Builder("foo","quux").edgeType(EdgeType.COUSIN).build();
 124         List<Object> results3 = testRules.read(readStart, q3.getFilter());
 125         assertTrue(results3.isEmpty());
 126     }
 127
 128     @Test
 129     public void testTree() {
 130         EdgeRuleQuery q = new EdgeRuleQuery.Builder("foo").edgeType(EdgeType.TREE).build();
 131         List<Object> results = testRules.read(readStart, q.getFilter());
 132         assertTrue(results.size() == 2);
 133     }
 134 }