2 * ============LICENSE_START=======================================================
\r
4 * ================================================================================
\r
5 * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
\r
6 * Copyright © 2017-2018 Amdocs
\r
7 * ================================================================================
\r
8 * Licensed under the Apache License, Version 2.0 (the "License");
\r
9 * you may not use this file except in compliance with the License.
\r
10 * You may obtain a copy of the License at
\r
12 * http://www.apache.org/licenses/LICENSE-2.0
\r
14 * Unless required by applicable law or agreed to in writing, software
\r
15 * distributed under the License is distributed on an "AS IS" BASIS,
\r
16 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
\r
17 * See the License for the specific language governing permissions and
\r
18 * limitations under the License.
\r
19 * ============LICENSE_END=========================================================
\r
21 package org.onap.aai.sa.searchdbabstraction.searchapi;
\r
23 import com.fasterxml.jackson.databind.ObjectMapper;
\r
24 import java.io.File;
\r
25 import java.io.IOException;
\r
26 import org.junit.Assert;
\r
27 import org.junit.Test;
\r
28 import org.onap.aai.sa.rest.TestUtils;
\r
30 public class RangeQueryTest {
\r
32 @Test(expected = IllegalArgumentException.class)
\r
33 public void testSetGt() {
\r
35 RangeQuery rq = new RangeQuery();
\r
36 rq.setLt(new String("2x"));
\r
37 Assert.assertEquals("2x", rq.getLt());
\r
38 Assert.assertNotNull(rq.toElasticSearch());
\r
39 Assert.assertNotNull(rq.toString());
\r
40 rq.setGt(new Integer(1));
\r
43 @Test(expected = IllegalArgumentException.class)
\r
44 public void testSetGte() {
\r
46 RangeQuery rq = new RangeQuery();
\r
47 rq.setGt(new Integer(1));
\r
48 Assert.assertNotNull(rq.toElasticSearch());
\r
49 Assert.assertNotNull(rq.toString());
\r
50 rq.setGte(new Integer(1));
\r
53 @Test(expected = IllegalArgumentException.class)
\r
54 public void testSetLt() {
\r
56 RangeQuery rq = new RangeQuery();
\r
57 rq.setLt(new Integer(1));
\r
58 rq.setFormat("format-1");
\r
59 Assert.assertEquals(rq.getFormat(), "format-1");
\r
60 Assert.assertNotNull(rq.toElasticSearch());
\r
61 Assert.assertNotNull(rq.toString());
\r
63 rq.setGt(new Integer(1));
\r
64 Assert.assertNotNull(rq.toElasticSearch());
\r
65 Assert.assertNotNull(rq.toString());
\r
66 rq.setLt(new String("10"));
\r
69 @Test(expected = IllegalArgumentException.class)
\r
70 public void testSetLte() {
\r
71 RangeQuery rq = new RangeQuery();
\r
72 rq.setGt(new Integer(1));
\r
73 rq.setTimeZone("CT");
\r
74 Assert.assertEquals(rq.getTimeZone(), "CT");
\r
75 Assert.assertNotNull(rq.toElasticSearch());
\r
76 Assert.assertNotNull(rq.toString());
\r
78 rq.setLte(new String("10"));
\r
82 public void testSearchStatementAggregations() throws IOException {
\r
83 File queryWithSubrangeFile = new File("src/test/resources/json/queries/query-with-subrange.json");
\r
84 String queryWithSubrangeStr = TestUtils.readFileToString(queryWithSubrangeFile);
\r
86 ObjectMapper mapper = new ObjectMapper();
\r
87 SearchStatement ss = mapper.readValue(queryWithSubrangeStr, SearchStatement.class);
\r
89 Aggregation a1 = getAggregationObject();
\r
90 Aggregation a2 = getAggregationObject();
\r
91 Aggregation[] aggs = new Aggregation[] {a1, a2};
\r
92 ss.setAggregations(aggs);
\r
93 Assert.assertNotNull(ss.toString());
\r
96 private Aggregation getAggregationObject() {
\r
97 Aggregation a = new Aggregation();
\r
99 AggregationStatement as = new AggregationStatement();
\r
100 DateHistogramAggregation dha = new DateHistogramAggregation();
\r
101 dha.setField("field-1");
\r
102 dha.setInterval("interval-1");
\r
103 Assert.assertEquals(dha.getInterval(), "interval-1");
\r
104 dha.setTimeZone("CT");
\r
105 Assert.assertEquals(dha.getTimeZone(), "CT");
\r
106 dha.setFormat("format-1");
\r
107 Assert.assertEquals(dha.getFormat(), "format-1");
\r
109 dha.setMinThreshold(1);
\r
110 Assert.assertNotNull(dha.toElasticSearch());
\r
111 Assert.assertNotNull(dha.toString());
\r
112 as.setDateHist(dha);
\r
115 as.getNestedPath();
\r
117 DateRangeAggregation dra = new DateRangeAggregation();
\r
118 dra.setField("field-1");
\r
119 dra.setMinThreshold(1);
\r
120 dra.setFormat("format-1");
\r
121 Assert.assertEquals(dra.getFormat(), "format-1");
\r
122 DateRange dr = new DateRange();
\r
123 dr.setFromDate("01-12-2017");
\r
124 Assert.assertEquals(dr.getFromDate(), "01-12-2017");
\r
125 dr.setToDate("21-12-2017");
\r
126 Assert.assertEquals(dr.getToDate(), "21-12-2017");
\r
127 DateRange[] drs = {dr};
\r
128 dra.setDateRanges(drs);
\r
129 Assert.assertTrue(dra.getDateRanges().length == 1);
\r
130 Assert.assertNotNull(dra.toElasticSearch());
\r
131 Assert.assertNotNull(dra.toString());
\r
132 as.setDateRange(dra);
\r
135 as.getNestedPath();
\r
137 GroupByAggregation gba = new GroupByAggregation();
\r
138 gba.setField("field-1");
\r
139 gba.setMinThreshold(1);
\r
140 Assert.assertNotNull(gba.toElasticSearch());
\r
141 Assert.assertNotNull(gba.toString());
\r
142 as.setGroupBy(gba);
\r
143 Assert.assertNotNull(as.toString());
\r
145 a.setStatement(as);
\r
146 Assert.assertNotNull(a.toString());
\r