code: def test_wait_for_all_nodes(self, mock_batch):
      # TODO: Add test
      pass
  
imports: from __future__ import annotations
  from unittest import mock
  from unittest.mock import PropertyMock
  import pytest
  from azure.batch import BatchServiceClient, models
  from airflow.models import Connection
  from airflow.providers.microsoft.azure.hooks.batch import AzureBatchHook
label_desc: Use image node_id 9 for calling the TestAzureBatchHook obj's underlying member method code with example usage: obj.test_wait_for_all_nodes(mock_batch) without return types