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