run overflow forecast a 2nd time as regression test for elastic/ml-cp…

…p#110 (#30969) Improve test to run overflow forecast a 2nd time as regression test for elastic/ml-cpp#110
elastic · Jun 5, 2018 · 5e48ba7 · 5e48ba7
1 parent 500094f
commit 5e48ba7
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/...k/qa/ml-native-tests/src/test/java/org/elasticsearch/xpack/ml/integration/ForecastIT.java b/...k/qa/ml-native-tests/src/test/java/org/elasticsearch/xpack/ml/integration/ForecastIT.java
@@ -239,15 +239,28 @@ public void testOverflowToDisk() throws Exception {
             throw e;
         }
 
-        closeJob(job.getId());
-
         List<ForecastRequestStats> forecastStats = getForecastStats();
         assertThat(forecastStats.size(), equalTo(1));
         ForecastRequestStats forecastRequestStats = forecastStats.get(0);
         List<Forecast> forecasts = getForecasts(job.getId(), forecastRequestStats);
 
         assertThat(forecastRequestStats.getRecordCount(), equalTo(8000L));
         assertThat(forecasts.size(), equalTo(8000));
+
+        // run forecast a 2nd time
+        try {
+            String forecastId = forecast(job.getId(), TimeValue.timeValueHours(1), null);
+
+            waitForecastToFinish(job.getId(), forecastId);
+        } catch (ElasticsearchStatusException e) {
+            if (e.getMessage().contains("disk space")) {
+                throw new ElasticsearchStatusException(
+                        "Test likely fails due to insufficient disk space on test machine, please free up space.", e.status(), e);
+            }
+            throw e;
+        }
+
+        closeJob(job.getId());
     }
 
     private void createDataWithLotsOfClientIps(TimeValue bucketSpan, Job.Builder job) throws IOException {