--- /dev/null
+/**\r
+ * ============LICENSE_START=======================================================\r
+ * org.onap.aai\r
+ * ================================================================================\r
+ * Copyright © 2017 AT&T Intellectual Property. All rights reserved.\r
+ * Copyright © 2017 Amdocs\r
+ * ================================================================================\r
+ * Licensed under the Apache License, Version 2.0 (the "License");\r
+ * you may not use this file except in compliance with the License.\r
+ * You may obtain a copy of the License at\r
+ *\r
+ * http://www.apache.org/licenses/LICENSE-2.0\r
+ *\r
+ * Unless required by applicable law or agreed to in writing, software\r
+ * distributed under the License is distributed on an "AS IS" BASIS,\r
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
+ * See the License for the specific language governing permissions and\r
+ * limitations under the License.\r
+ * ============LICENSE_END=========================================================\r
+ *\r
+ * ECOMP is a trademark and service mark of AT&T Intellectual Property.\r
+ */\r
+package org.onap.aai.sa.searchdbabstraction.searchapi;\r
+\r
+import com.fasterxml.jackson.databind.ObjectMapper;\r
+import org.junit.Assert;\r
+import org.junit.Test;\r
+import org.onap.aai.sa.rest.TestUtils;\r
+\r
+import java.io.File;\r
+import java.io.IOException;\r
+import java.util.Date;\r
+\r
+public class RangeQueryTest {\r
+\r
+ @Test(expected=IllegalArgumentException.class)\r
+ public void testSetGt(){\r
+\r
+ RangeQuery rq = new RangeQuery();\r
+ rq.setLt(new String("2x"));\r
+ Assert.assertEquals("2x", rq.getLt());\r
+ Assert.assertNotNull(rq.toElasticSearch());\r
+ Assert.assertNotNull(rq.toString());\r
+ rq.setGt(new Integer(1));\r
+ }\r
+\r
+ @Test(expected=IllegalArgumentException.class)\r
+ public void testSetGte(){\r
+\r
+ RangeQuery rq = new RangeQuery();\r
+ rq.setGt(new Integer(1));\r
+ Assert.assertNotNull(rq.toElasticSearch());\r
+ Assert.assertNotNull(rq.toString());\r
+ rq.setGte(new Integer(1));\r
+ }\r
+\r
+ @Test(expected=IllegalArgumentException.class)\r
+ public void testSetLt(){\r
+\r
+ RangeQuery rq = new RangeQuery();\r
+ rq.setLt(new Integer(1));\r
+ rq.setFormat("format-1");\r
+ Assert.assertEquals(rq.getFormat(), "format-1");\r
+ Assert.assertNotNull(rq.toElasticSearch());\r
+ Assert.assertNotNull(rq.toString());\r
+\r
+ rq.setGt(new Integer(1));\r
+ Assert.assertNotNull(rq.toElasticSearch());\r
+ Assert.assertNotNull(rq.toString());\r
+ rq.setLt(new String("10"));\r
+ }\r
+\r
+ @Test(expected=IllegalArgumentException.class)\r
+ public void testSetLte(){\r
+ RangeQuery rq = new RangeQuery();\r
+ rq.setGt(new Integer(1));\r
+ rq.setTimeZone("CT");\r
+ Assert.assertEquals(rq.getTimeZone(), "CT");\r
+ Assert.assertNotNull(rq.toElasticSearch());\r
+ Assert.assertNotNull(rq.toString());\r
+\r
+ rq.setLte(new String("10"));\r
+ }\r
+\r
+ @Test\r
+ public void testSearchStatementAggregations() throws IOException {\r
+ File queryWithSubrangeFile = new File("src/test/resources/json/queries/query-with-subrange.json");\r
+ String queryWithSubrangeStr = TestUtils.readFileToString(queryWithSubrangeFile);\r
+\r
+ ObjectMapper mapper = new ObjectMapper();\r
+ SearchStatement ss = mapper.readValue(queryWithSubrangeStr, SearchStatement.class);\r
+\r
+ Aggregation a1 = getAggregationObject();\r
+ Aggregation a2 = getAggregationObject();\r
+ Aggregation[] aggs= new Aggregation[] {a1, a2};\r
+ ss.setAggregations(aggs);\r
+ Assert.assertNotNull(ss.toString());\r
+ }\r
+\r
+ private Aggregation getAggregationObject(){\r
+ Aggregation a = new Aggregation();\r
+\r
+ AggregationStatement as = new AggregationStatement();\r
+ DateHistogramAggregation dha = new DateHistogramAggregation();\r
+ dha.setField("field-1");\r
+ dha.setInterval("interval-1");\r
+ Assert.assertEquals(dha.getInterval(), "interval-1");\r
+ dha.setTimeZone("CT");\r
+ Assert.assertEquals(dha.getTimeZone(), "CT");\r
+ dha.setFormat("format-1");\r
+ Assert.assertEquals(dha.getFormat(), "format-1");\r
+ dha.setSize(10);\r
+ dha.setMinThreshold(1);\r
+ Assert.assertNotNull(dha.toElasticSearch());\r
+ Assert.assertNotNull(dha.toString());\r
+ as.setDateHist(dha);\r
+ as.toString();\r
+\r
+ as.getNestedPath();\r
+\r
+ DateRangeAggregation dra = new DateRangeAggregation();\r
+ dra.setField("field-1");\r
+ dra.setMinThreshold(1);\r
+ dra.setFormat("format-1");\r
+ Assert.assertEquals(dra.getFormat(), "format-1");\r
+ DateRange dr = new DateRange();\r
+ dr.setFromDate("01-12-2017");\r
+ Assert.assertEquals(dr.getFromDate(), "01-12-2017");\r
+ dr.setToDate("21-12-2017");\r
+ Assert.assertEquals(dr.getToDate(), "21-12-2017");\r
+ DateRange[] drs = {dr};\r
+ dra.setDateRanges(drs);\r
+ Assert.assertTrue(dra.getDateRanges().length==1);\r
+ Assert.assertNotNull(dra.toElasticSearch());\r
+ Assert.assertNotNull(dra.toString());\r
+ as.setDateRange(dra);\r
+ as.toString();\r
+\r
+ as.getNestedPath();\r
+\r
+ GroupByAggregation gba = new GroupByAggregation();\r
+ gba.setField("field-1");\r
+ gba.setMinThreshold(1);\r
+ Assert.assertNotNull(gba.toElasticSearch());\r
+ Assert.assertNotNull(gba.toString());\r
+ as.setGroupBy(gba);\r
+ Assert.assertNotNull(as.toString());\r
+\r
+ a.setStatement(as);\r
+ Assert.assertNotNull(a.toString());\r
+ return a;\r
+ }\r
+}\r