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')
