Skip to content

Commit

Permalink
Fix test hooks also for Azure Batch (#27607)
Browse files Browse the repository at this point in the history
The #27601 also needs fix in Hook test cases.

(cherry picked from commit 3e3ff7c)
  • Loading branch information
potiuk authored and ephraimbuddy committed Nov 11, 2022
1 parent 7b5628a commit c53a3e1
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions tests/providers/microsoft/azure/hooks/test_azure_batch.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand All @@ -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",
Expand All @@ -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",
Expand All @@ -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",
Expand All @@ -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",
Expand Down

0 comments on commit c53a3e1

Please sign in to comment.