Add extended attribute to get-cluster-status

This changes from the default boolean that was being used internally
to what mysql expects. The values between 0 and 3 are valid.

Author: arif-ali

actions.yaml

@@ -3,6 +3,11 @@
 
 get-cluster-status:
   description: Get cluster status information
+  params:
+    extended:
+      type: integer
+      description: The extended attriute for cluster-status, the default value 0.
+        possible values - 0, 1, 2, 3
 
 get-password:
   description: Fetch the system user's password, which is used by charm.

lib/charms/mysql/v0/mysql.py

@@ -494,7 +494,9 @@ def _on_set_password(self, event: ActionEvent) -> None:
 
     def _get_cluster_status(self, event: ActionEvent) -> None:
         """Action used  to retrieve the cluster status."""
-        if status := self._mysql.get_cluster_status():
+        extended = event.params.get("extended") or 0
+
+        if status := self._mysql.get_cluster_status(extended):
             event.set_results(
                 {
                     "success": True,
@@ -1422,7 +1424,7 @@ def is_instance_in_cluster(self, unit_label: str) -> bool:
         stop=stop_after_attempt(3),
         retry=retry_if_exception_type(TimeoutError),
     )
-    def get_cluster_status(self, extended: Optional[bool] = False) -> Optional[dict]:
+    def get_cluster_status(self, extended: Optional[int] = 0) -> Optional[dict]:
         """Get the cluster status.
 
         Executes script to retrieve cluster status.

src/upgrade.py

@@ -110,7 +110,7 @@ def _online_instances(status_dict: dict) -> int:
                 if not item.get("instanceerrors", [])
             ].count("online")
 
-        if cluster_status := self.charm._mysql.get_cluster_status(extended=True):
+        if cluster_status := self.charm._mysql.get_cluster_status(extended=1):
             if _online_instances(cluster_status) < self.charm.app.planned_units():
                 # case any not fully online unit is found
                 raise ClusterNotReadyError(

tests/unit/test_mysql.py

@@ -731,7 +731,7 @@ def test_get_cluster_status(self, _run_mysqlsh_script):
             (
                 "shell.connect('clusteradmin:[email protected]')",
                 "cluster = dba.get_cluster('test_cluster')",
-                "print(cluster.status({'extended': False}))",
+                "print(cluster.status({'extended': 0}))",
             )
         )
         _run_mysqlsh_script.assert_called_once_with(expected_commands, timeout=30)