Fix test hooks also for Azure Batch (apache#27607)

The apache#27601 also needs fix in Hook test cases.
Adityamalik123 · Nov 12, 2022 · 0a4aa0b · 0a4aa0b
1 parent 445be0b
commit 0a4aa0b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/providers/microsoft/azure/hooks/test_azure_batch.py b/tests/providers/microsoft/azure/hooks/test_azure_batch.py
@@ -73,6 +73,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",
@@ -85,6 +86,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",
@@ -103,6 +105,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",
@@ -117,6 +120,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",
@@ -132,6 +136,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",