adding mocks for localdns update command

Author: anushkasingh16

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_localdns_profile = Mock(return_value=agentpool)
 
         # Act
         result = decorator.update_agentpool_profile_preview()
@@ -148,6 +149,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_localdns_profile.assert_called_once_with(agentpool)
 
     def test_update_agentpool_profile_preview_with_agentpools_parameter(self):
         """Test update_agentpool_profile_preview with agentpools parameter."""
@@ -185,6 +187,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_localdns_profile = Mock(return_value=agentpool)
 
         # Act
         result = decorator.update_agentpool_profile_preview(agentpools)
@@ -233,6 +236,7 @@ def test_update_agentpool_profile_preview_managed_system_mode(self):
         decorator.update_os_sku = Mock()
         decorator.update_fips_image = Mock()
         decorator.update_ssh_access = Mock()
+        decorator.update_localdns_profile = Mock()
 
         # Act
         result = decorator.update_agentpool_profile_preview()
@@ -258,6 +262,7 @@ def test_update_agentpool_profile_preview_managed_system_mode(self):
         decorator.update_os_sku.assert_not_called()
         decorator.update_fips_image.assert_not_called()
         decorator.update_ssh_access.assert_not_called()
+        decorator.update_localdns_profile.assert_not_called()
 
     def test_update_agentpool_profile_preview_managed_system_mode_with_agentpools(self):
         """Test update_agentpool_profile_preview with ManagedSystem mode and agentpools parameter."""
@@ -333,6 +338,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_localdns_profile = Mock(return_value=agentpool)
 
         # Act
         result = decorator.update_agentpool_profile_preview()
@@ -350,6 +356,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_localdns_profile.assert_called_once_with(agentpool)
 
     def test_update_agentpool_profile_preview_execution_order(self):
         """Test that update methods are called in the correct order."""
@@ -392,6 +399,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_localdns_profile = create_mock_update_method("update_localdns_profile")
 
         # Act
         decorator.update_agentpool_profile_preview()
@@ -406,6 +414,7 @@ def mock_method(pool):
             "update_os_sku",
             "update_fips_image",
             "update_ssh_access",
+            "update_localdns_profile"
         ]
         self.assertEqual(call_order, expected_order)
 
@@ -450,6 +459,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_localdns_profile = create_tracking_mock("update_localdns_profile")
 
         # Act
         result = decorator.update_agentpool_profile_preview()
@@ -511,7 +521,7 @@ def test_update_agentpool_profile_preview_mixed_modes_scenario(self):
                 update_methods = [
                     'update_custom_ca_trust', 'update_network_profile', 'update_artifact_streaming',
                     'update_secure_boot', 'update_vtpm', 'update_os_sku', 'update_fips_image',
-                    'update_ssh_access'
+                    'update_ssh_access', 'update_localdns_profile'
                 ]
 
                 for method_name in update_methods:
@@ -580,6 +590,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_localdns_profile = Mock(return_value=agentpool)
 
         # Act
         result = decorator.update_agentpool_profile_preview(agentpools)