Update mock

Author: wenxuan0923

src/aks-preview/azext_aks_preview/agentpool_decorator.py

@@ -957,7 +957,7 @@ def get_update_enable_disable_cluster_autoscaler_and_min_max_count_vmsize_vms(
         if not update_cluster_autoscaler and vm_size is not None:
             raise MutuallyExclusiveArgumentError(
                 "--node-vm-size is only applicable when updating cluster autoscaler settings "
-                "with --updata-cluster-autoscaler"
+                "with --update-cluster-autoscaler"
             )
 
         self._AKSAgentPoolContext__validate_counts_in_autoscaler(

src/aks-preview/azext_aks_preview/tests/latest/test_update_agentpool_profile_preview.py

@@ -130,6 +130,7 @@ def test_update_agentpool_profile_preview_default_behavior(self):
         decorator.update_fips_image = Mock(return_value=agentpool)
         decorator.update_ssh_access = Mock(return_value=agentpool)
         decorator.update_localdns_profile = Mock(return_value=agentpool)
+        decorator.update_auto_scaler_properties_vms = Mock(return_value=agentpool)
 
         # Act
         result = decorator.update_agentpool_profile_preview()
@@ -150,6 +151,7 @@ def test_update_agentpool_profile_preview_default_behavior(self):
         decorator.update_fips_image.assert_called_once_with(agentpool)
         decorator.update_ssh_access.assert_called_once_with(agentpool)
         decorator.update_localdns_profile.assert_called_once_with(agentpool)
+        decorator.update_auto_scaler_properties_vms.assert_called_once_with(agentpool)
 
     def test_update_agentpool_profile_preview_with_agentpools_parameter(self):
         """Test update_agentpool_profile_preview with agentpools parameter."""
@@ -188,6 +190,7 @@ def test_update_agentpool_profile_preview_with_agentpools_parameter(self):
         decorator.update_fips_image = Mock(return_value=agentpool)
         decorator.update_ssh_access = Mock(return_value=agentpool)
         decorator.update_localdns_profile = Mock(return_value=agentpool)
+        decorator.update_auto_scaler_properties_vms = Mock(return_value=agentpool)
 
         # Act
         result = decorator.update_agentpool_profile_preview(agentpools)
@@ -237,6 +240,7 @@ def test_update_agentpool_profile_preview_managed_system_mode(self):
         decorator.update_fips_image = Mock()
         decorator.update_ssh_access = Mock()
         decorator.update_localdns_profile = Mock()
+        decorator.update_auto_scaler_properties_vms = Mock()
 
         # Act
         result = decorator.update_agentpool_profile_preview()
@@ -339,6 +343,7 @@ def test_update_agentpool_profile_preview_system_mode_regular_flow(self):
         decorator.update_fips_image = Mock(return_value=agentpool)
         decorator.update_ssh_access = Mock(return_value=agentpool)
         decorator.update_localdns_profile = Mock(return_value=agentpool)
+        decorator.update_auto_scaler_properties_vms = Mock(return_value=agentpool)
 
         # Act
         result = decorator.update_agentpool_profile_preview()
@@ -357,6 +362,7 @@ def test_update_agentpool_profile_preview_system_mode_regular_flow(self):
         decorator.update_fips_image.assert_called_once_with(agentpool)
         decorator.update_ssh_access.assert_called_once_with(agentpool)
         decorator.update_localdns_profile.assert_called_once_with(agentpool)
+        decorator.update_auto_scaler_properties_vms.assert_called_once_with(agentpool)
 
     def test_update_agentpool_profile_preview_execution_order(self):
         """Test that update methods are called in the correct order."""
@@ -400,6 +406,7 @@ def mock_method(pool):
         decorator.update_fips_image = create_mock_update_method("update_fips_image")
         decorator.update_ssh_access = create_mock_update_method("update_ssh_access")
         decorator.update_localdns_profile = create_mock_update_method("update_localdns_profile")
+        decorator.update_auto_scaler_properties_vms = create_mock_update_method("update_auto_scaler_properties_vms")
 
         # Act
         decorator.update_agentpool_profile_preview()
@@ -414,7 +421,8 @@ def mock_method(pool):
             "update_os_sku",
             "update_fips_image",
             "update_ssh_access",
-            "update_localdns_profile"
+            "update_localdns_profile",
+            "update_auto_scaler_properties_vms"
         ]
         self.assertEqual(call_order, expected_order)
 
@@ -460,6 +468,7 @@ def track_and_return(pool):
         decorator.update_fips_image = create_tracking_mock("update_fips_image")
         decorator.update_ssh_access = create_tracking_mock("update_ssh_access")
         decorator.update_localdns_profile = create_tracking_mock("update_localdns_profile")
+        decorator.update_auto_scaler_properties_vms = create_tracking_mock("update_auto_scaler_properties_vms")
 
         # Act
         result = decorator.update_agentpool_profile_preview()
@@ -591,6 +600,7 @@ def test_update_agentpool_profile_preview_managed_cluster_mode(self):
         decorator.update_fips_image = Mock(return_value=agentpool)
         decorator.update_ssh_access = Mock(return_value=agentpool)
         decorator.update_localdns_profile = Mock(return_value=agentpool)
+        decorator.update_auto_scaler_properties_vms = Mock(return_value=agentpool)
 
         # Act
         result = decorator.update_agentpool_profile_preview(agentpools)