revert: remove incorrect LRO handling changes in custom.py and put_mc

Revert changes to aks_enable_addons, aks_disable_addons and put_mc
that added explicit LongRunningOperation calls. The stable CLI uses
sdk_no_wait pattern consistently and the CLI framework handles
LRO poller resolution automatically for supports_no_wait commands.

Author: charleswool

src/aks-preview/azext_aks_preview/custom.py

@@ -3032,10 +3032,7 @@ def aks_disable_addons(cmd, client, resource_group_name, name, addons, no_wait=F
     )
 
     # send the managed cluster representation to update the addon profiles
-    if no_wait:
-        return sdk_no_wait(True, client.begin_create_or_update, resource_group_name, name, instance)
-    return LongRunningOperation(cmd.cli_ctx)(
-        client.begin_create_or_update(resource_group_name, name, instance))
+    return sdk_no_wait(no_wait, client.begin_create_or_update, resource_group_name, name, instance)
 
 
 def aks_enable_addons(
@@ -3192,12 +3189,8 @@ def aks_enable_addons(
                 # we don't need to handle it in client side in this case.
 
     else:
-        if no_wait:
-            result = sdk_no_wait(True, client.begin_create_or_update,
-                                 resource_group_name, name, instance, headers=headers)
-        else:
-            result = LongRunningOperation(cmd.cli_ctx)(
-                client.begin_create_or_update(resource_group_name, name, instance, headers=headers))
+        result = sdk_no_wait(no_wait, client.begin_create_or_update,
+                             resource_group_name, name, instance, headers=headers)
     return result
 
 

src/aks-preview/azext_aks_preview/managed_cluster_decorator.py

@@ -7791,9 +7791,9 @@ def put_mc(self, mc: ManagedCluster) -> ManagedCluster:
             # poll until the result is returned
             cluster = LongRunningOperation(self.cmd.cli_ctx)(poller)
             self.postprocessing_after_mc_created(cluster)
-        elif self.context.get_no_wait():
+        else:
             cluster = sdk_no_wait(
-                True,
+                self.context.get_no_wait(),
                 self.client.begin_create_or_update,
                 resource_group_name=self.context.get_resource_group_name(),
                 resource_name=self.context.get_name(),
@@ -7802,15 +7802,5 @@ def put_mc(self, mc: ManagedCluster) -> ManagedCluster:
                 if_none_match=self.context.get_if_none_match(),
                 headers=self.context.get_aks_custom_headers(),
             )
-        else:
-            poller = self.client.begin_create_or_update(
-                resource_group_name=self.context.get_resource_group_name(),
-                resource_name=self.context.get_name(),
-                parameters=mc,
-                if_match=self.context.get_if_match(),
-                if_none_match=self.context.get_if_none_match(),
-                headers=self.context.get_aks_custom_headers(),
-            )
-            cluster = LongRunningOperation(self.cmd.cli_ctx)(poller)
 
         return cluster