fix: add update_vm_size to profile preview test mocks

The test_update_agentpool_profile_preview tests mock all update methods
called by update_agentpool_profile_preview(). Adding update_vm_size()
to the call chain requires updating these mocks and assertions.

Signed-off-by: wenhug <50309350+wenhug@users.noreply.github.com>

Author: wenhug

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

@@ -129,6 +129,7 @@ def test_update_agentpool_profile_preview_default_behavior(self):
         decorator.update_os_sku = Mock(return_value=agentpool)
         decorator.update_fips_image = Mock(return_value=agentpool)
         decorator.update_ssh_access = Mock(return_value=agentpool)
+        decorator.update_vm_size = Mock(return_value=agentpool)
         decorator.update_localdns_profile = Mock(return_value=agentpool)
         decorator.update_auto_scaler_properties_vms = Mock(return_value=agentpool)
         decorator.update_upgrade_strategy = Mock(return_value=agentpool)
@@ -154,6 +155,7 @@ def test_update_agentpool_profile_preview_default_behavior(self):
         decorator.update_os_sku.assert_called_once_with(agentpool)
         decorator.update_fips_image.assert_called_once_with(agentpool)
         decorator.update_ssh_access.assert_called_once_with(agentpool)
+        decorator.update_vm_size.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)
         decorator.update_upgrade_strategy.assert_called_once_with(agentpool)
@@ -197,6 +199,7 @@ def test_update_agentpool_profile_preview_with_agentpools_parameter(self):
         decorator.update_os_sku = Mock(return_value=agentpool)
         decorator.update_fips_image = Mock(return_value=agentpool)
         decorator.update_ssh_access = Mock(return_value=agentpool)
+        decorator.update_vm_size = Mock(return_value=agentpool)
         decorator.update_localdns_profile = Mock(return_value=agentpool)
         decorator.update_auto_scaler_properties_vms = Mock(return_value=agentpool)
         decorator.update_upgrade_strategy = Mock(return_value=agentpool)
@@ -359,6 +362,7 @@ def test_update_agentpool_profile_preview_system_mode_regular_flow(self):
         decorator.update_os_sku = Mock(return_value=agentpool)
         decorator.update_fips_image = Mock(return_value=agentpool)
         decorator.update_ssh_access = Mock(return_value=agentpool)
+        decorator.update_vm_size = Mock(return_value=agentpool)
         decorator.update_localdns_profile = Mock(return_value=agentpool)
         decorator.update_auto_scaler_properties_vms = Mock(return_value=agentpool)
         decorator.update_upgrade_strategy = Mock(return_value=agentpool)
@@ -382,6 +386,7 @@ def test_update_agentpool_profile_preview_system_mode_regular_flow(self):
         decorator.update_os_sku.assert_called_once_with(agentpool)
         decorator.update_fips_image.assert_called_once_with(agentpool)
         decorator.update_ssh_access.assert_called_once_with(agentpool)
+        decorator.update_vm_size.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)
         decorator.update_upgrade_strategy.assert_called_once_with(agentpool)
@@ -430,6 +435,7 @@ def mock_method(pool):
         decorator.update_os_sku = create_mock_update_method("update_os_sku")
         decorator.update_fips_image = create_mock_update_method("update_fips_image")
         decorator.update_ssh_access = create_mock_update_method("update_ssh_access")
+        decorator.update_vm_size = create_mock_update_method("update_vm_size")
         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")
         decorator.update_upgrade_strategy = create_mock_update_method("update_upgrade_strategy")
@@ -450,6 +456,7 @@ def mock_method(pool):
             "update_os_sku",
             "update_fips_image",
             "update_ssh_access",
+            "update_vm_size",
             "update_localdns_profile",
             "update_auto_scaler_properties_vms",
             "update_upgrade_strategy",
@@ -500,6 +507,7 @@ def track_and_return(pool):
         decorator.update_os_sku = create_tracking_mock("update_os_sku")
         decorator.update_fips_image = create_tracking_mock("update_fips_image")
         decorator.update_ssh_access = create_tracking_mock("update_ssh_access")
+        decorator.update_vm_size = create_tracking_mock("update_vm_size")
         decorator.update_localdns_profile = create_tracking_mock("update_localdns_profile")
         decorator.update_auto_scaler_properties_vms = create_tracking_mock("update_auto_scaler_properties_vms")
         decorator.update_upgrade_strategy = create_tracking_mock("update_upgrade_strategy")
@@ -567,7 +575,7 @@ def test_update_agentpool_profile_preview_mixed_modes_scenario(self):
                 update_methods = [
                     'update_network_profile', 'update_artifact_streaming', 'update_managed_gpu',
                     'update_secure_boot', 'update_vtpm', 'update_os_sku', 'update_fips_image',
-                    'update_ssh_access', 'update_localdns_profile', 'update_auto_scaler_properties_vms', 
+                    'update_ssh_access', 'update_vm_size', 'update_localdns_profile', 'update_auto_scaler_properties_vms', 
                     'update_upgrade_strategy', 'update_blue_green_upgrade_settings', 'update_gpu_profile',
                     'update_gpu_mig_strategy'
                 ]
@@ -638,6 +646,7 @@ def test_update_agentpool_profile_preview_managed_cluster_mode(self):
         decorator.update_os_sku = Mock(return_value=agentpool)
         decorator.update_fips_image = Mock(return_value=agentpool)
         decorator.update_ssh_access = Mock(return_value=agentpool)
+        decorator.update_vm_size = Mock(return_value=agentpool)
         decorator.update_localdns_profile = Mock(return_value=agentpool)
         decorator.update_auto_scaler_properties_vms = Mock(return_value=agentpool)
         decorator.update_upgrade_strategy = Mock(return_value=agentpool)