src/test/java/org/onap/aai/sa/searchdbabstraction/searchapi/RangeQueryTest.java

   1 /**\r
   2  * ============LICENSE_START=======================================================\r
   3  * org.onap.aai\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
  11  *\r
  12  *       http://www.apache.org/licenses/LICENSE-2.0\r
  13  *\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
  20  */\r
  21 package org.onap.aai.sa.searchdbabstraction.searchapi;\r
  22 \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
  29 \r
  30 public class RangeQueryTest {\r
  31 \r
  32     @Test(expected=IllegalArgumentException.class)\r
  33     public void testSetGt(){\r
  34 \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
  41     }\r
  42 \r
  43     @Test(expected=IllegalArgumentException.class)\r
  44     public void testSetGte(){\r
  45 \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
  51     }\r
  52 \r
  53     @Test(expected=IllegalArgumentException.class)\r
  54     public void testSetLt(){\r
  55 \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
  62 \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
  67     }\r
  68 \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
  77 \r
  78         rq.setLte(new String("10"));\r
  79     }\r
  80 \r
  81     @Test\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
  85 \r
  86         ObjectMapper mapper = new ObjectMapper();\r
  87         SearchStatement ss = mapper.readValue(queryWithSubrangeStr, SearchStatement.class);\r
  88 \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
  94     }\r
  95 \r
  96     private Aggregation getAggregationObject(){\r
  97         Aggregation a = new Aggregation();\r
  98 \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
 108         dha.setSize(10);\r
 109         dha.setMinThreshold(1);\r
 110         Assert.assertNotNull(dha.toElasticSearch());\r
 111         Assert.assertNotNull(dha.toString());\r
 112         as.setDateHist(dha);\r
 113         as.toString();\r
 114 \r
 115         as.getNestedPath();\r
 116 \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
 133         as.toString();\r
 134 \r
 135         as.getNestedPath();\r
 136 \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
 144 \r
 145         a.setStatement(as);\r
 146         Assert.assertNotNull(a.toString());\r
 147         return a;\r
 148     }\r
 149 }\r