def test_virtual_machine_destroy_vm_on(wait_for_vcenter_task, connection):
vm = VirtualMachine()
vm_object_mock = mock.MagicMock()
vm.__setattr__('_vm_object', vm_object_mock)
vm.destroy()
vm.destroy()
assert vm.__getattribute__('_vm_object') is None
assert wait_for_vcenter_task.call_count == 2
def test_virtual_machine_destroy_vm_off(wait_for_vcenter_task, connection):
vm = VirtualMachine()
vm_object_mock = mock.MagicMock()
vm.__setattr__('_vm_object', vm_object_mock)
wait_for_vcenter_task.side_effect = [vim.fault.InvalidPowerState, None]
vm.destroy()
vm.destroy()
assert vm.__getattribute__('_vm_object') is None
assert wait_for_vcenter_task.call_count == 2
def test_virtual_machine_create_not_enough_disk_space(
wait_for_vcenter_task, relocate_spec, clone_spec,
get_vcenter_object_by_name, connection):
os.environ['vcdriver_resource_pool'] = 'something'
os.environ['vcdriver_data_store'] = 'something'
os.environ['vcdriver_data_store_threshold'] = '120'
os.environ['vcdriver_folder'] = 'something'
load()
vm = VirtualMachine()
with pytest.raises(NotEnoughDiskSpace):
vm.create()
assert vm.__getattribute__('_vm_object') is None
assert wait_for_vcenter_task.call_count == 0
def test_virtual_machine_create(wait_for_vcenter_task, relocate_spec,
clone_spec, get_vcenter_object_by_name,
connection):
os.environ['vcdriver_resource_pool'] = 'something'
os.environ['vcdriver_data_store'] = 'something'
os.environ['vcdriver_data_store_threshold'] = '20'
os.environ['vcdriver_folder'] = 'something'
load()
vm = VirtualMachine()
vm.create()
vm.create()
assert vm.__getattribute__('_vm_object') is not None
assert wait_for_vcenter_task.call_count == 1
def test_virtual_machine_refresh(close, get_vcenter_object_by_name,
connection):
vm = VirtualMachine()
assert vm.__getattribute__('_vm_object') is None
# Test that refresh does nothing if no _vm_object
vm.refresh()
assert vm.__getattribute__('_vm_object') is None
# Test that refresh assigns a new _vm_object
vm.__setattr__('_vm_object', mock.Mock())
initial__vm_object = vm.__getattribute__('_vm_object')
assert initial__vm_object is not None
vm.refresh()
refreshed__vm_object = vm.__getattribute__('_vm_object')
assert refreshed__vm_object is not None
assert initial__vm_object != refreshed__vm_object
close.assert_called_once()
get_vcenter_object_by_name.assert_called_once()
def test_virtual_machine_find(get_vcenter_object_by_name, connection):
vm = VirtualMachine()
vm.find()
vm.find()
assert vm.__getattribute__('_vm_object') is not None
assert get_vcenter_object_by_name.call_count == 1
