[8.0] Add cleanups for filters, calendars, transforms, jobs

Co-authored-by: Seth Michael Larson <[email protected]>

Author: github-actions[bot]

test_elasticsearch/utils.py

@@ -149,6 +149,9 @@ def wipe_cluster(client):
         wipe_tasks(client)
         wipe_node_shutdown_metadata(client)
         wait_for_pending_datafeeds_and_jobs(client)
+        wipe_calendars(client)
+        wipe_filters(client)
+        wipe_transforms(client)
 
     wait_for_cluster_state_updates_to_finish(client)
     if close_after_wipe:
@@ -334,7 +337,7 @@ def wait_for_pending_tasks(client, filter, timeout=30):
             break
 
 
-def wait_for_pending_datafeeds_and_jobs(client, timeout=30):
+def wait_for_pending_datafeeds_and_jobs(client: Elasticsearch, timeout=30):
     end_time = time.time() + timeout
     while time.time() < end_time:
         resp = client.ml.get_datafeeds(datafeed_id="*", allow_no_match=True)
@@ -345,6 +348,7 @@ def wait_for_pending_datafeeds_and_jobs(client, timeout=30):
                 datafeed_id=datafeed["datafeed_id"]
             )
 
+    end_time = time.time() + timeout
     while time.time() < end_time:
         resp = client.ml.get_jobs(job_id="*", allow_no_match=True)
         if resp["count"] == 0:
@@ -353,6 +357,56 @@ def wait_for_pending_datafeeds_and_jobs(client, timeout=30):
             client.options(ignore_status=404).ml.close_job(job_id=job["job_id"])
             client.options(ignore_status=404).ml.delete_job(job_id=job["job_id"])
 
+    end_time = time.time() + timeout
+    while time.time() < end_time:
+        resp = client.ml.get_data_frame_analytics(id="*")
+        if resp["count"] == 0:
+            break
+        for job in resp["data_frame_analytics"]:
+            client.options(ignore_status=404).ml.stop_data_frame_analytics(id=job["id"])
+            client.options(ignore_status=404).ml.delete_data_frame_analytics(
+                id=job["id"]
+            )
+
+
+def wipe_filters(client: Elasticsearch, timeout=30):
+    end_time = time.time() + timeout
+    while time.time() < end_time:
+        resp = client.ml.get_filters(filter_id="*")
+        if resp["count"] == 0:
+            break
+        for filter in resp["filters"]:
+            client.options(ignore_status=404).ml.delete_filter(
+                filter_id=filter["filter_id"]
+            )
+
+
+def wipe_calendars(client: Elasticsearch, timeout=30):
+    end_time = time.time() + timeout
+    while time.time() < end_time:
+        resp = client.ml.get_calendars(calendar_id="*")
+        if resp["count"] == 0:
+            break
+        for calendar in resp["calendars"]:
+            client.options(ignore_status=404).ml.delete_calendar(
+                calendar_id=calendar["calendar_id"]
+            )
+
+
+def wipe_transforms(client: Elasticsearch, timeout=30):
+    end_time = time.time() + timeout
+    while time.time() < end_time:
+        resp = client.transform.get_transform(transform_id="*")
+        if resp["count"] == 0:
+            break
+        for trasnform in resp["transforms"]:
+            client.options(ignore_status=404).transform.stop_transform(
+                transform_id=trasnform["id"]
+            )
+            client.options(ignore_status=404).transform.delete_transform(
+                transform_id=trasnform["id"]
+            )
+
 
 def wait_for_cluster_state_updates_to_finish(client, timeout=30):
     end_time = time.time() + timeout