added test case for listing multiple report definitions (opendistro-f…

…or-elasticsearch#276)
zhongnansu · Jan 5, 2021 · 3b6fcc7 · 3b6fcc7
1 parent c8d98e0
commit 3b6fcc7
Showing 1 changed file with 41 additions and 0 deletions.
diff --git a/.../kotlin/com/amazon/opendistroforelasticsearch/reportsscheduler/rest/ReportDefinitionIT.kt b/.../kotlin/com/amazon/opendistroforelasticsearch/reportsscheduler/rest/ReportDefinitionIT.kt
@@ -276,4 +276,45 @@ class ReportDefinitionIT : PluginRestTestCase() {
         )
         Thread.sleep(100)
     }
+
+    fun `test listing multiple report definitions`() {
+        val reportDefinitionRequest = constructReportDefinitionRequest()
+        val reportDefinitionResponse = executeRequest(
+            RestRequest.Method.POST.name,
+            "$BASE_REPORTS_URI/definition",
+            reportDefinitionRequest,
+            RestStatus.OK.status
+        )
+        val reportDefinitionId = reportDefinitionResponse.get("reportDefinitionId").asString
+        Assert.assertNotNull("reportDefinitionId should be generated", reportDefinitionId)
+        Thread.sleep(100)
+
+        val secondReportDefinitionRequest = constructReportDefinitionRequest(name = "new report definition")
+        val secondReportDefinitionResponse = executeRequest(
+            RestRequest.Method.POST.name,
+            "$BASE_REPORTS_URI/definition",
+            secondReportDefinitionRequest,
+            RestStatus.OK.status
+        )
+        val newReportDefinitionId = secondReportDefinitionResponse.get("reportDefinitionId").asString
+        Assert.assertNotNull("reportDefinitionId should be generated", newReportDefinitionId)
+        Thread.sleep(1000)
+        val listReportDefinitions = executeRequest(
+            RestRequest.Method.GET.name,
+            "$BASE_REPORTS_URI/definitions",
+            "",
+            RestStatus.OK.status
+        )
+        val totalHits = listReportDefinitions.get("totalHits").asInt
+        Assert.assertEquals(totalHits, 2)
+        val reportDefinitionsList = listReportDefinitions.get("reportDefinitionDetailsList").asJsonArray
+        Assert.assertEquals(
+            reportDefinitionId,
+            reportDefinitionsList[0].asJsonObject.get("id").asString
+        )
+        Assert.assertEquals(
+            newReportDefinitionId,
+            reportDefinitionsList[1].asJsonObject.get("id").asString
+        )
+    }
 }