[ML] Adds anomaly detectors forecast API test

x-pack/test/api_integration/apis/ml/anomaly_detectors/forecast_with_spaces.ts

@@ -0,0 +1,95 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0; you may not use this file except in compliance with the Elastic License
+ * 2.0.
+ */
+
+import { FtrProviderContext } from '../../../ftr_provider_context';
+import { USER } from '../../../../functional/services/ml/security_common';
+import { JOB_STATE, DATAFEED_STATE } from '../../../../../plugins/ml/common/constants/states';
+import { COMMON_REQUEST_HEADERS } from '../../../../functional/services/ml/common_api';
+
+export default ({ getService }: FtrProviderContext) => {
+  const esArchiver = getService('esArchiver');
+  const supertest = getService('supertestWithoutAuth');
+  const ml = getService('ml');
+  const spacesService = getService('spaces');
+
+  const forecastJobId = 'fq_single_forecast';
+  const idSpace1 = 'space1';
+  const idSpace2 = 'space2';
+  const jobConfig = ml.commonConfig.getADFqSingleMetricJobConfig(forecastJobId);
+  const datafeedConfig = ml.commonConfig.getADFqDatafeedConfig(forecastJobId);
+
+  async function runForecast(
+    jobId: string,
+    space: string,
+    duration: string | number,
+    user: USER,
+    expectedStatusCode: number
+  ) {
+    const { body } = await supertest
+      .post(`${space ? `/s/${space}` : ''}/api/ml/anomaly_detectors/${jobId}/_forecast`)
+      .auth(user, ml.securityCommon.getPasswordForUser(user))
+      .set(COMMON_REQUEST_HEADERS)
+      .send({ duration })
+      .expect(expectedStatusCode);
+
+    return body;
+  }
+
+  describe('POST anomaly_detectors _forecast with spaces', function () {
+    before(async () => {
+      await esArchiver.loadIfNeeded('x-pack/test/functional/es_archives/ml/farequote');
+      await ml.testResources.setKibanaTimeZoneToUTC();
+
+      await spacesService.create({ id: idSpace1, name: 'space_one', disabledFeatures: [] });
+      await spacesService.create({ id: idSpace2, name: 'space_two', disabledFeatures: [] });
+
+      await ml.api.createAnomalyDetectionJob(jobConfig, idSpace1);
+      await ml.api.createDatafeed(datafeedConfig, idSpace1);
+    });
+
+    after(async () => {
+      await ml.api.closeAnomalyDetectionJob(forecastJobId);
+      await spacesService.delete(idSpace1);
+      await spacesService.delete(idSpace2);
+      await ml.api.cleanMlIndices();
+      await ml.testResources.cleanMLSavedObjects();
+    });
+
+    it('should not forecast for a job that has not been opened', async () => {
+      await runForecast(forecastJobId, idSpace1, '1d', USER.ML_POWERUSER, 409);
+    });
+
+    it('should not forecast for a job that has not been run', async () => {
+      await ml.api.openAnomalyDetectionJob(forecastJobId);
+      await runForecast(forecastJobId, idSpace1, '1d', USER.ML_POWERUSER, 400);
+    });
+
+    it('should not forecast for a job with an invalid job ID', async () => {
+      await runForecast(`${forecastJobId}_invalid`, idSpace1, '1d', USER.ML_POWERUSER, 404);
+    });
+
+    it('should run forecast for open job with valid job ID', async () => {
+      await ml.api.startDatafeed(datafeedConfig.datafeed_id, { start: '0', end: `${Date.now()}` });
+      await ml.api.waitForDatafeedState(datafeedConfig.datafeed_id, DATAFEED_STATE.STOPPED);
+      await ml.api.waitForJobState(jobConfig.job_id, JOB_STATE.CLOSED);
+      await ml.api.openAnomalyDetectionJob(forecastJobId);
+      await runForecast(forecastJobId, idSpace1, '1d', USER.ML_POWERUSER, 200);
+    });
+
+    it('should not run forecast for open job with invalid duration', async () => {
+      await runForecast(forecastJobId, idSpace1, 3600000, USER.ML_VIEWER, 403);
+    });
+
+    it('should not run forecast for open job with valid job ID as ML Viewer', async () => {
+      await runForecast(forecastJobId, idSpace1, '1d', USER.ML_VIEWER, 403);
+    });
+
+    it('should not run forecast for open job with valid job ID in wrong space', async () => {
+      await runForecast(forecastJobId, idSpace2, '1d', USER.ML_POWERUSER, 404);
+    });
+  });
+};

x-pack/test/api_integration/apis/ml/anomaly_detectors/index.ts

@@ -17,5 +17,6 @@ export default function ({ loadTestFile }: FtrProviderContext) {
     loadTestFile(require.resolve('./close_with_spaces'));
     loadTestFile(require.resolve('./delete_with_spaces'));
     loadTestFile(require.resolve('./create_with_spaces'));
+    loadTestFile(require.resolve('./forecast_with_spaces'));
   });
 }