fix

Author: FumingZhang

src/aks-preview/azext_aks_preview/_roleassignments.py

@@ -3,7 +3,10 @@
 # Licensed under the MIT License. See License.txt in the project root for license information.
 # --------------------------------------------------------------------------------------------
 
-from azure.cli.command_modules.acs._roleassignments import add_role_assignment
+from azure.cli.command_modules.acs._roleassignments import (
+    add_role_assignment,
+    add_role_assignment_executor,
+)
 
 
 def add_role_assignment(
@@ -19,8 +22,28 @@ def add_role_assignment(
         cmd,
         role,
         service_principal_msi_id,
-        is_service_principal,
-        delay,
-        scope,
-        assignee_principal_type,
+        is_service_principal=is_service_principal,
+        delay=delay,
+        scope=scope,
+        assignee_principal_type=assignee_principal_type,
+    )
+
+
+def add_role_assignment_executor(
+    cmd,
+    role,
+    assignee,
+    resource_group_name=None,
+    scope=None,
+    resolve_assignee=True,
+    assignee_principal_type=None,
+):
+    return add_role_assignment_executor(
+        cmd,
+        role,
+        assignee,
+        resource_group_name=resource_group_name,
+        scope=scope,
+        resolve_assignee=resolve_assignee,
+        assignee_principal_type=assignee_principal_type,
     )

src/aks-preview/azext_aks_preview/managed_cluster_decorator.py

@@ -73,7 +73,7 @@
 )
 from azext_aks_preview._roleassignments import (
     add_role_assignment,
-    _add_role_assignment_executor_new
+    add_role_assignment_executor
 )
 from azext_aks_preview.agentpool_decorator import (
     AKSPreviewAgentPoolAddDecorator,
@@ -212,7 +212,7 @@ def external_functions(self) -> SimpleNamespace:
             ] = ensure_azure_monitor_profile_prerequisites
             # temp workaround for the breaking change caused by default API version bump of the auth SDK
             external_functions["add_role_assignment"] = add_role_assignment
-            external_functions["_add_role_assignment_executor_new"] = _add_role_assignment_executor_new
+            external_functions["add_role_assignment_executor"] = add_role_assignment_executor
             # azure container storage functions
             external_functions[
                 "perform_enable_azure_container_storage_v1"
@@ -4227,7 +4227,7 @@ def postprocessing_after_mc_created(self, cluster: ManagedCluster) -> None:
             except Exception as e:  # pylint: disable=broad-except
                 logger.warning("Could not get signed in user: %s", str(e))
             else:
-                self.context.external_functions._add_role_assignment_executor_new(  # type: ignore # pylint: disable=protected-access
+                self.context.external_functions.add_role_assignment_executor(  # type: ignore # pylint: disable=protected-access
                     self.cmd,
                     "Azure Kubernetes Service RBAC Cluster Admin",
                     user["id"],