Fix test hooks also for Azure Batch (#27607)

The #27601 also needs fix in Hook test cases.

(cherry picked from commit 3e3ff7c)

tests/providers/microsoft/azure/hooks/test_azure_batch.py

@@ -72,6 +72,7 @@ def test_configure_pool_with_vm_config(self):
         pool = hook.configure_pool(
             pool_id='mypool',
             vm_size="test_vm_size",
+            vm_node_agent_sku_id=self.test_vm_sku,
             target_dedicated_nodes=1,
             vm_publisher="test.vm.publisher",
             vm_offer="test.vm.offer",
@@ -84,6 +85,7 @@ def test_configure_pool_with_cloud_config(self):
         pool = hook.configure_pool(
             pool_id='mypool',
             vm_size="test_vm_size",
+            vm_node_agent_sku_id=self.test_vm_sku,
             target_dedicated_nodes=1,
             vm_publisher="test.vm.publisher",
             vm_offer="test.vm.offer",
@@ -102,6 +104,7 @@ def test_configure_pool_with_latest_vm(self):
             pool = hook.configure_pool(
                 pool_id='mypool',
                 vm_size="test_vm_size",
+                vm_node_agent_sku_id=self.test_vm_sku,
                 use_latest_image_and_sku=True,
                 vm_publisher="test.vm.publisher",
                 vm_offer="test.vm.offer",
@@ -116,6 +119,7 @@ def test_create_pool_with_vm_config(self, mock_batch):
         pool = hook.configure_pool(
             pool_id='mypool',
             vm_size="test_vm_size",
+            vm_node_agent_sku_id=self.test_vm_sku,
             target_dedicated_nodes=1,
             vm_publisher="test.vm.publisher",
             vm_offer="test.vm.offer",
@@ -131,6 +135,7 @@ def test_create_pool_with_cloud_config(self, mock_batch):
         pool = hook.configure_pool(
             pool_id='mypool',
             vm_size="test_vm_size",
+            vm_node_agent_sku_id=self.test_vm_sku,
             target_dedicated_nodes=1,
             vm_publisher="test.vm.publisher",
             vm_offer="test.vm.offer",