def test_virtual_machine_find_snapshot(wait_for_vcenter_task): fake_snapshots = [mock.MagicMock(), mock.MagicMock(), mock.MagicMock()] for fake_snapshot in fake_snapshots[:-1]: fake_snapshot.name = 'snapshot' fake_snapshot.childSnapshotList = [] fake_snapshots[-1].name = 'other' fake_snapshots[-1].childSnapshotList = [] vm = VirtualMachine() assert vm.find_snapshot('snapshot') is None vm_object_mock = mock.MagicMock() vm.__setattr__('_vm_object', vm_object_mock) vm_object_mock.snapshot.rootSnapshotList = [] with pytest.raises(NoObjectFound): vm.find_snapshot('snapshot') vm_object_mock.snapshot.rootSnapshotList = fake_snapshots[:-2] vm.find_snapshot('snapshot') vm_object_mock.snapshot.rootSnapshotList = fake_snapshots with pytest.raises(TooManyObjectsFound): vm.find_snapshot('snapshot') vm_object_mock.snapshot = None with pytest.raises(NoObjectFound): vm.find_snapshot('snapshot')
def test_virtual_machine_remove_snapshot(wait_for_vcenter_task): vm = VirtualMachine() assert vm.remove_snapshot('snapshot') is None vm.find_snapshot = mock.MagicMock() vm.__setattr__('_vm_object', mock.MagicMock()) vm.remove_snapshot('snapshot')