def test_initialize_disk_mapping(self):
vsan.add_capacity_to_diskgroup(
self.mock_si, self.mock_vsan_disk_mgmt_system,
self.mock_host_ref, self.mock_diskgroup,
[self.mock_cap_disk1, self.mock_cap_disk2])
self.mock_initialise_disk_mapping.assert_called_once_with(
self.mock_spec)
def test_vsan_spec_hybrid(self):
self.mock_cap_disk1.ssd = False
vsan.add_capacity_to_diskgroup(
self.mock_si, self.mock_vsan_disk_mgmt_system, self.mock_host_ref,
self.mock_diskgroup, [self.mock_cap_disk1, self.mock_cap_disk2])
self.mock_cap_disk1.ssd = False
self.assertEqual(self.mock_spec.creationType, 'hybrid')
def test__wait_for_tasks(self):
mock___wait_for_tasks = MagicMock()
with patch('salt.utils.vsan._wait_for_tasks', mock___wait_for_tasks):
vsan.add_capacity_to_diskgroup(
self.mock_si, self.mock_vsan_disk_mgmt_system,
self.mock_host_ref, self.mock_diskgroup,
[self.mock_cap_disk1, self.mock_cap_disk2])
mock___wait_for_tasks.assert_called_once_with([self.mock_task],
self.mock_si)
def test_vsan_spec_all_flash(self):
self.mock_cap_disk1.ssd = True
vsan.add_capacity_to_diskgroup(
self.mock_si, self.mock_vsan_disk_mgmt_system, self.mock_host_ref,
self.mock_diskgroup, [self.mock_cap_disk1, self.mock_cap_disk2])
self.assertEqual(self.mock_spec.capacityDisks,
[self.mock_cap_disk1, self.mock_cap_disk2])
self.assertEqual(self.mock_spec.cacheDisks, [self.mock_cache_disk])
self.assertEqual(self.mock_spec.creationType, 'allFlash')
self.assertEqual(self.mock_spec.host, self.mock_host_ref)
def test_get_hostname(self):
mock_get_managed_object_name = MagicMock(return_value='fake_hostname')
with patch('salt.utils.vmware.get_managed_object_name',
mock_get_managed_object_name):
vsan.add_capacity_to_diskgroup(
self.mock_si, self.mock_vsan_disk_mgmt_system,
self.mock_host_ref, self.mock_diskgroup,
[self.mock_cap_disk1, self.mock_cap_disk2])
mock_get_managed_object_name.assert_called_once_with(
self.mock_host_ref)
def test_initialize_disk_mapping_raise_runtime_fault(self):
err = vmodl.RuntimeFault()
err.msg = 'runtime_fault'
self.mock_vsan_disk_mgmt_system.InitializeDiskMappings = \
MagicMock(side_effect=err)
with self.assertRaises(VMwareRuntimeError) as excinfo:
vsan.add_capacity_to_diskgroup(
self.mock_si, self.mock_vsan_disk_mgmt_system,
self.mock_host_ref, self.mock_diskgroup,
[self.mock_cap_disk1, self.mock_cap_disk2])
self.assertEqual(excinfo.exception.strerror, 'runtime_fault')
def test_initialize_disk_mapping_raise_method_not_found(self):
err = vmodl.fault.MethodNotFound()
err.method = 'fake_method'
self.mock_vsan_disk_mgmt_system.InitializeDiskMappings = \
MagicMock(side_effect=err)
with self.assertRaises(VMwareRuntimeError) as excinfo:
vsan.add_capacity_to_diskgroup(
self.mock_si, self.mock_vsan_disk_mgmt_system,
self.mock_host_ref, self.mock_diskgroup,
[self.mock_cap_disk1, self.mock_cap_disk2])
self.assertEqual(excinfo.exception.strerror,
'Method \'fake_method\' not found')
def test_initialize_disk_mapping_raise_no_permission(self):
err = vim.fault.NoPermission()
err.privilegeId = 'Fake privilege'
self.mock_vsan_disk_mgmt_system.InitializeDiskMappings = \
MagicMock(side_effect=err)
with self.assertRaises(VMwareApiError) as excinfo:
vsan.add_capacity_to_diskgroup(
self.mock_si, self.mock_vsan_disk_mgmt_system,
self.mock_host_ref, self.mock_diskgroup,
[self.mock_cap_disk1, self.mock_cap_disk2])
self.assertEqual(excinfo.exception.strerror,
'Not enough permissions. Required privilege: '
'Fake privilege')
def test_initialize_disk_mapping_raise_vim_fault(self):
err = vim.fault.VimFault()
err.msg = "vim_fault"
self.mock_vsan_disk_mgmt_system.InitializeDiskMappings = MagicMock(
side_effect=err)
with self.assertRaises(VMwareApiError) as excinfo:
vsan.add_capacity_to_diskgroup(
self.mock_si,
self.mock_vsan_disk_mgmt_system,
self.mock_host_ref,
self.mock_diskgroup,
[self.mock_cap_disk1, self.mock_cap_disk2],
)
self.assertEqual(excinfo.exception.strerror, "vim_fault")
def test_result(self):
res = vsan.add_capacity_to_diskgroup(
self.mock_si,
self.mock_vsan_disk_mgmt_system,
self.mock_host_ref,
self.mock_diskgroup,
[self.mock_cap_disk1, self.mock_cap_disk2],
)
self.assertTrue(res)
