From e3100fae5a8cd72fb15d38a63ac9643ce0af5424 Mon Sep 17 00:00:00 2001
From: dabarzil <dabarzil@redhat.com>
Date: Mon, 8 Jul 2024 14:46:49 +0300
Subject: [PATCH] add test remove all grastate.dat for OSP18

Change-Id: Ifd042a6652dac52100de8ae4f2dbf81c79bfbe67
---
 .../faults/podified/ha/cloud_disruptions.py   | 54 +++++++++++++++++++
 1 file changed, 54 insertions(+)

diff --git a/tobiko/tests/faults/podified/ha/cloud_disruptions.py b/tobiko/tests/faults/podified/ha/cloud_disruptions.py
index 391aed8fc..24855b1d8 100644
--- a/tobiko/tests/faults/podified/ha/cloud_disruptions.py
+++ b/tobiko/tests/faults/podified/ha/cloud_disruptions.py
@@ -40,6 +40,8 @@ LOG = log.getLogger(__name__)
 
 @podified.skip_if_not_podified
 def kill_all_galera_services():
+    """kill all galera processes,
+    check in pacemaker it is down"""
     galera_pods_num = sum(
         1 for node_name in oc.selector('nodes').qnames()
         for pod_obj in oc.get_pods_by_node(node_name)
@@ -82,3 +84,55 @@ def kill_all_galera_services():
         except oc.OpenShiftPythonException:
             continue
         return False
+
+
+@podified.skip_if_not_podified
+def remove_all_grastate_galera():
+    """shut down galera properly,
+    remove all grastate"""
+    galera_pods_num = sum(
+        1 for node_name in oc.selector('nodes').qnames()
+        for pod_obj in oc.get_pods_by_node(node_name)
+        if 'cell1-galera' in pod_obj.fqname()
+    )
+    for i in range(galera_pods_num):
+        oc.selector('pod/openstack-cell1-galera-{}'.format(i)).object()\
+            .execute(['sh', '-c', 'rm -rf /var/lib/mysql/grastate.dat '],
+                     container_name='galera')
+        LOG.info('delete grastate.dat cell-{}'.format(i))
+    for i in range(galera_pods_num):
+        oc.selector('pod/openstack-cell1-galera-{}'.format(i)).object()\
+            .execute(['sh', '-c', 'kill -9 $(pidof mysqld)'],
+                     container_name='galera')
+        LOG.info('kill galera cell-{}'.format(i))
+    retry = tobiko.retry(timeout=30, interval=5)
+    for _ in retry:
+        try:
+            # checks wsrep cluster size is now unavailable
+            result = oc.selector('pod/openstack-cell1-galera-0').object(
+            ).execute(['sh', '-c', """mysql -u root --password=12345678
+            -e 'SHOW STATUS LIKE "wsrep_cluster_size"'"""])
+            # Capture and filter the error output
+            error_output = result.err()
+            non_error_message = """
+            Defaulted container "galera" out of: galera,
+            mysql-bootstrap (init)\n"""
+            filtered_err_output = error_output.replace(non_error_message, '')
+            if not filtered_err_output.strip():
+                continue
+        except oc.OpenShiftPythonException:
+            LOG.info('all galera cells down')
+            break
+    time.sleep(60)
+    for _ in retry:
+        try:
+            if int(re.search(r'wsrep_cluster_size\s+(\d+)', oc.selector(
+                    'pod/openstack-cell1-galera-0').object().execute(
+                ['sh', '-c', """mysql -u root --password=12345678 -e 'SHOW
+                STATUS LIKE "wsrep_cluster_size"'"""], container_name='galera'
+            ).out()).group(1)) == galera_pods_num:
+                LOG.info('all galera cells are restored')
+                return
+        except oc.OpenShiftPythonException:
+            continue
+        return False