  def testGetBootDisk(self, mock_get_vm, mock_list_disk, mock_provider):
    """Test that the boot disk from an instance is retrieved."""
    mock_provider.return_value = azure_mocks.MOCK_CAPACITY_PROVIDER
    mock_get_vm.return_value = mock.Mock(
        storage_profile=mock.Mock(os_disk=azure_mocks.MOCK_BOOT_DISK))
    mock_list_disk.return_value = azure_mocks.MOCK_LIST_DISKS
    boot_disk = azure_mocks.FAKE_INSTANCE.GetBootDisk()
    mock_list_disk.assert_called_once()
    mock_list_disk.assert_called_with(
        resource_group_name=azure_mocks.FAKE_INSTANCE.resource_group_name)
    self.assertEqual('fake-boot-disk-name', boot_disk.name)
Test that the boot disk from an instance is retrieved.
tests/providers/azure/internal/test_compute.py