[airflow] branch main updated: Fix test hooks also for Azure Batch (#27607)

[potiuk]

This is an automated email from the ASF dual-hosted git repository.

potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git


The following commit(s) were added to refs/heads/main by this push:
     new 3e3ff7ce52 Fix test hooks also for Azure Batch (#27607)
3e3ff7ce52 is described below

commit 3e3ff7ce5232a2662cd9f35e21ee8f088e568d6d
Author: Jarek Potiuk <ja...@potiuk.com>
AuthorDate: Thu Nov 10 23:16:35 2022 +0100

    Fix test hooks also for Azure Batch (#27607)
    
    The #27601 also needs fix in Hook test cases.
---
 tests/providers/microsoft/azure/hooks/test_azure_batch.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tests/providers/microsoft/azure/hooks/test_azure_batch.py 
b/tests/providers/microsoft/azure/hooks/test_azure_batch.py
index 6673261919..a4e4f40117 100644
--- a/tests/providers/microsoft/azure/hooks/test_azure_batch.py
+++ b/tests/providers/microsoft/azure/hooks/test_azure_batch.py
@@ -73,6 +73,7 @@ class TestAzureBatchHook(unittest.TestCase):
         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 @@ class TestAzureBatchHook(unittest.TestCase):
         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 @@ class TestAzureBatchHook(unittest.TestCase):
             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 @@ class TestAzureBatchHook(unittest.TestCase):
         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 @@ class TestAzureBatchHook(unittest.TestCase):
         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",