Skip to content

Add test coverage for AzureBatchHook.wait_for_all_node_state#63765

Open
SameerMesiah97 wants to merge 1 commit intoapache:mainfrom
SameerMesiah97:AzureBatchHook-Missing-Tests
Open

Add test coverage for AzureBatchHook.wait_for_all_node_state#63765
SameerMesiah97 wants to merge 1 commit intoapache:mainfrom
SameerMesiah97:AzureBatchHook-Missing-Tests

Conversation

@SameerMesiah97
Copy link
Contributor

Description

Add unit tests for AzureBatchHook.wait_for_all_node_state. The tests cover scenarios where nodes are immediately ready, where polling continues due to insufficient node count or node state, and where a pool resize error triggers an exception.

Rationale

The test suite contained a #TODO indicating missing coverage for this method. The polling logic in wait_for_all_node_state contains multiple branches controlling loop termination and error handling that were not previously exercised by the test suite. Separate tests are used instead of parametrization to clearly represent the lifecycle and polling transitions of compute nodes.

@SameerMesiah97 SameerMesiah97 requested a review from dabla as a code owner March 16, 2026 23:17
@SameerMesiah97 SameerMesiah97 force-pushed the AzureBatchHook-Missing-Tests branch 2 times, most recently from 2f30081 to 90042bd Compare March 17, 2026 18:19
Add tests verifying behavior when nodes are already in the target state, when node count is insufficient, when node states require retries, and when pool resize errors are present.
@SameerMesiah97 SameerMesiah97 force-pushed the AzureBatchHook-Missing-Tests branch from 90042bd to 76e86b2 Compare March 18, 2026 01:32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant