add more IT

Signed-off-by: Jackie Han <hnyng@amazon.com>

Author: jackiehanyang

src/test/java/org/opensearch/ad/rest/AnomalyDetectorRestApiIT.java

@@ -241,7 +241,7 @@ public void testCreateAnomalyDetector_withFlattenedResultIndex() throws Exceptio
             .format(Locale.ROOT, "/opensearch-ad-plugin-result-test_flattened_%s", id.toLowerCase(Locale.ROOT));
         // wait for the detector starts writing result
         try {
-            Thread.sleep(60 * 1000);
+            Thread.sleep(20 * 1000);
         } catch (InterruptedException e) {
             Thread.currentThread().interrupt();
             throw new RuntimeException("Thread was interrupted while waiting", e);
@@ -256,6 +256,81 @@ public void testCreateAnomalyDetector_withFlattenedResultIndex() throws Exceptio
         assertEquals("Dynamic field is not set to true", "true", dynamicValue.toString());
         Map<String, Object> properties = (Map<String, Object>) mappings.get("properties");
         assertTrue("Flattened field 'feature_data_feature_bytes' does not exist", properties.containsKey("feature_data_feature_bytes"));
+
+    }
+
+    public void testUpdateAnomalyDetector_enableFlattenResultIndex_shouldCreatePipeline() throws Exception {
+        AnomalyDetector detector = createIndexAndGetAnomalyDetector(
+            INDEX_NAME,
+            ImmutableList.of(TestHelpers.randomFeature("feature_bytes", "agg", true)),
+            false,
+            false
+        );
+
+        // test behavior when AD is enabled
+        updateClusterSettings(ADEnabledSetting.AD_ENABLED, true);
+        // create a detector with flatten result index disabled, shouldn't find related ingest pipeline
+        Response response = TestHelpers
+            .makeRequest(client(), "POST", TestHelpers.AD_BASE_DETECTORS_URI, ImmutableMap.of(), TestHelpers.toHttpEntity(detector), null);
+        assertEquals("Create anomaly detector with flattened result index failed", RestStatus.CREATED, TestHelpers.restStatus(response));
+        Map<String, Object> responseMap = entityAsMap(response);
+        String id = (String) responseMap.get("_id");
+        String expectedPipelineId = String.format(Locale.ROOT, "flatten_result_index_ingest_pipeline%s", id.toLowerCase(Locale.ROOT));
+        String getIngestPipelineEndpoint = String.format(Locale.ROOT, "_ingest/pipeline/%s", expectedPipelineId);
+        ResponseException responseException = expectThrows(
+            ResponseException.class,
+            () -> TestHelpers.makeRequest(client(), "GET", getIngestPipelineEndpoint, ImmutableMap.of(), "", null)
+        );
+        int statusCode = responseException.getResponse().getStatusLine().getStatusCode();
+        assertEquals("Expected 404 response but got: " + statusCode, 404, statusCode);
+
+        // update the detector with flatten result index enabled, should be able to find ingest pipeline
+        List<Feature> features = detector.getFeatureAttributes();
+        AnomalyDetector newDetector = new AnomalyDetector(
+            id,
+            detector.getVersion(),
+            detector.getName(),
+            detector.getDescription(),
+            detector.getTimeField(),
+            detector.getIndices(),
+            features,
+            detector.getFilterQuery(),
+            detector.getInterval(),
+            detector.getWindowDelay(),
+            detector.getShingleSize(),
+            detector.getUiMetadata(),
+            detector.getSchemaVersion(),
+            detector.getLastUpdateTime(),
+            null,
+            detector.getUser(),
+            detector.getCustomResultIndexOrAlias(),
+            TestHelpers.randomImputationOption(features),
+            randomIntBetween(1, 10000),
+            randomInt(TimeSeriesSettings.MAX_SHINGLE_SIZE / 2),
+            randomIntBetween(1, 1000),
+            null,
+            null,
+            null,
+            null,
+            true,
+            detector.getLastBreakingUIChangeTime()
+        );
+        Response updateResponse = TestHelpers
+            .makeRequest(
+                client(),
+                "PUT",
+                TestHelpers.AD_BASE_DETECTORS_URI + "/" + id + "?refresh=true",
+                ImmutableMap.of(),
+                TestHelpers.toHttpEntity(newDetector),
+                null
+            );
+        assertEquals("Update anomaly detector failed", RestStatus.OK, TestHelpers.restStatus(updateResponse));
+        Response getPipelineResponse = TestHelpers.makeRequest(client(), "GET", getIngestPipelineEndpoint, ImmutableMap.of(), "", null);
+        assertEquals(
+            "Expected 200 response but got: " + getPipelineResponse.getStatusLine().getStatusCode(),
+            200,
+            getPipelineResponse.getStatusLine().getStatusCode()
+        );
     }
 
     public void testUpdateAnomalyDetector_disableFlattenResultIndex_shouldDeletePipeline() throws Exception {
@@ -273,8 +348,8 @@ public void testUpdateAnomalyDetector_disableFlattenResultIndex_shouldDeletePipe
         assertEquals("Create anomaly detector with flattened result index failed", RestStatus.CREATED, TestHelpers.restStatus(response));
         Map<String, Object> responseMap = entityAsMap(response);
         String id = (String) responseMap.get("_id");
-        String expectedFlattenedIndex = String
-            .format(Locale.ROOT, "opensearch-ad-plugin-result-test_flattened_%s", id.toLowerCase(Locale.ROOT));
+        // String expectedFlattenedIndex = String
+        // .format(Locale.ROOT, "opensearch-ad-plugin-result-test_flattened_%s", id.toLowerCase(Locale.ROOT));
         String expectedPipelineId = String.format(Locale.ROOT, "flatten_result_index_ingest_pipeline%s", id.toLowerCase(Locale.ROOT));
         String getIngestPipelineEndpoint = String.format(Locale.ROOT, "_ingest/pipeline/%s", expectedPipelineId);
         Response getPipelineResponse = TestHelpers.makeRequest(client(), "GET", getIngestPipelineEndpoint, ImmutableMap.of(), "", null);