Refactor tests

Author: nielsbauman

x-pack/plugin/ilm/qa/multi-node/src/javaRestTest/java/org/elasticsearch/xpack/TimeSeriesRestDriver.java

@@ -23,6 +23,7 @@
 import org.elasticsearch.common.xcontent.XContentHelper;
 import org.elasticsearch.core.Nullable;
 import org.elasticsearch.core.TimeValue;
+import org.elasticsearch.test.rest.ESRestTestCase;
 import org.elasticsearch.xcontent.ToXContent;
 import org.elasticsearch.xcontent.XContentBuilder;
 import org.elasticsearch.xcontent.XContentType;
@@ -58,7 +59,6 @@
 import static org.elasticsearch.xpack.core.ilm.ShrinkIndexNameSupplier.SHRUNKEN_INDEX_PREFIX;
 import static org.hamcrest.Matchers.anyOf;
 import static org.hamcrest.Matchers.equalTo;
-import static org.hamcrest.Matchers.is;
 import static org.junit.Assert.assertThat;
 
 /**
@@ -403,6 +403,28 @@ public static void updatePolicy(RestClient client, String indexName, String poli
         assertOK(client.performRequest(changePolicyRequest));
     }
 
+    /**
+     * Moves the specified index from the current ILM step to the next step.
+     */
+    public static void moveIndexToStep(RestClient client, String indexName, Step.StepKey currentStep, Step.StepKey nextStep)
+        throws IOException {
+        Request moveToStepRequest = new Request("POST", "_ilm/move/" + indexName);
+        moveToStepRequest.setJsonEntity(Strings.format("""
+            {
+              "current_step": {
+                "phase": "%s",
+                "action": "%s",
+                "name": "%s"
+              },
+              "next_step": {
+                "phase": "%s",
+                "action": "%s",
+                "name": "%s"
+              }
+            }""", currentStep.phase(), currentStep.action(), currentStep.name(), nextStep.phase(), nextStep.action(), nextStep.name()));
+        ESRestTestCase.assertAcknowledged(client.performRequest(moveToStepRequest));
+    }
+
     @SuppressWarnings("unchecked")
     public static String getSnapshotState(RestClient client, String snapshot) throws IOException {
         Response response = client.performRequest(new Request("GET", "/_snapshot/repo/" + snapshot));
@@ -485,24 +507,6 @@ public static String waitAndGetShrinkIndexName(RestClient client, String origina
         return shrunkenIndexName[0];
     }
 
-    @SuppressWarnings("unchecked")
-    public static List<String> getBackingIndices(RestClient client, String dataStreamName) throws IOException {
-        Response getDataStream = client.performRequest(new Request("GET", "_data_stream/" + dataStreamName));
-        Map<String, Object> responseMap;
-        try (InputStream is = getDataStream.getEntity().getContent()) {
-            responseMap = XContentHelper.convertToMap(XContentType.JSON.xContent(), is, true);
-        }
-
-        List<Map<String, Object>> dataStreams = (List<Map<String, Object>>) responseMap.get("data_streams");
-        assertThat(dataStreams.size(), is(1));
-        Map<String, Object> dataStream = dataStreams.get(0);
-        assertThat(dataStream.get("name"), is(dataStreamName));
-        List<String> indices = ((List<Map<String, Object>>) dataStream.get("indices")).stream()
-            .map(indexMap -> (String) indexMap.get("index_name"))
-            .toList();
-        return indices;
-    }
-
     private static void executeDummyClusterStateUpdate(RestClient client) throws IOException {
         createIndexWithSettings(
             client,