feature(manager): extend sctool restore task with dc-mapping flag

To run a restore task for multiDC cluster with EaR enabled (otherwise
fails #1), the special flag has been introduced in Manager (#2) to map
backed up DC with DC under restore, for example,

sctool restore ... --dc-mapping dc1=dc1,dc2=dc2

The change introduces this new flag into `create_restore_task` method
and makes sure that if Scylla cluster has more than 1 datacenter - the
restore task will be triggered with this flag applied.

refs:
1. scylladb/scylla-manager#3871
2. scylladb/scylla-manager#4213

Author: mikliapko

mgmt_cli_test.py

@@ -646,6 +646,18 @@ def locations(self) -> list[str]:
         # FIXME: Make it works with multiple locations or file a bug for scylla-manager.
         return [f"{backend}:{location}" for location in buckets[:1]]
 
+    def get_dc_mapping(self) -> str | None:
+        """Get the datacenter mapping string for the restore task if there are > 1 DCs (multiDC) in the cluster.
+        In case of singleDC, return None.
+
+        Example of return string:
+            eu-west-2scylla_node_west=eu-west-2scylla_node_west,us-eastscylla_node_east=us-eastscylla_node_east
+        """
+        if len(dcs := self.get_all_dcs_names()) > 1:
+            return ",".join([f"{dc}={dc}" for dc in dcs])
+        else:
+            return None
+
     # pylint: disable=too-many-arguments
     def verify_backup_success(self, mgr_cluster, backup_task, ks_names: list = None, tables_names: list = None,
                               truncate=True, restore_data_with_task=False, timeout=None):
@@ -726,9 +738,10 @@ def restore_backup_without_manager(self, mgr_cluster, snapshot_tag, ks_tables_li
     def restore_backup_with_task(self, mgr_cluster, snapshot_tag, timeout, restore_schema=False, restore_data=False,
                                  location_list=None, extra_params=None):
         location_list = location_list if location_list else self.locations
+        dc_mapping = self.get_dc_mapping()
         restore_task = mgr_cluster.create_restore_task(restore_schema=restore_schema, restore_data=restore_data,
                                                        location_list=location_list, snapshot_tag=snapshot_tag,
-                                                       extra_params=extra_params)
+                                                       dc_mapping=dc_mapping, extra_params=extra_params)
         restore_task.wait_and_get_final_status(step=30, timeout=timeout)
         assert restore_task.status == TaskStatus.DONE, f"Restoration of {snapshot_tag} has failed!"
         InfoEvent(message=f'The restore task has ended successfully. '

sdcm/mgmt/cli.py

@@ -603,7 +603,7 @@ def set_cluster_id(self, value: str):
         self.id = value
 
     def create_restore_task(self, restore_schema=False, restore_data=False, location_list=None, snapshot_tag=None,
-                            extra_params=None):
+                            dc_mapping=None, extra_params=None):
         cmd = f"restore -c {self.id}"
         if restore_schema:
             cmd += " --restore-schema"
@@ -614,6 +614,8 @@ def create_restore_task(self, restore_schema=False, restore_data=False, location
             cmd += " --location {} ".format(locations_names)
         if snapshot_tag:
             cmd += f" --snapshot-tag {snapshot_tag}"
+        if dc_mapping:
+            cmd += f" --dc-mapping {dc_mapping}"
         if extra_params:
             cmd += f" {extra_params}"