def test_get_volume_uuid(self):
vm_ref = mock.Mock()
uuid = '1234'
opt_val = vmwareapi_fake.OptionValue('volume-%s' % uuid, 'volume-val')
fake_call = self._fake_call_get_dynamic_property(uuid, opt_val)
with mock.patch.object(self._session, "_call_method", fake_call):
val = self._volumeops._get_volume_uuid(vm_ref, uuid)
self.assertEqual('volume-val', val)
def test_get_volume_uuid(self):
vm_ref = vmwareapi_fake.ManagedObjectReference('VirtualMachine',
'vm-134')
uuid = '1234'
opt_val = vmwareapi_fake.OptionValue('volume-%s' % uuid, 'volume-val')
fake_call = self._fake_call_get_object_property(uuid, opt_val)
with mock.patch.object(self._session, "_call_method", fake_call):
val = self._volumeops._get_volume_uuid(vm_ref, uuid)
self.assertEqual('volume-val', val)
def test_get_vm_ref(self):
"""Check retrieval of vm reference using uuid or name"""
respool = fake.ResourcePool()
fake._create_object('ResourcePool', respool)
cluster = fake.ClusterComputeResource()
cluster._add_root_resource_pool(respool.obj)
fake._create_object('ClusterComputeResource', cluster)
vm_folder = fake.Folder(name='fake-folder')
fake_folders = fake.FakeRetrieveResult()
fake_folders.add_object(vm_folder)
fake_pools = fake.FakeRetrieveResult()
fake_pools.add_object(respool)
fake_clusters = fake.FakeRetrieveResult()
fake_clusters.add_object(cluster)
optionVal = fake.OptionValue(key="nvp.vm-uuid", value="test-uuid-1")
virtualMachine1 = fake.VirtualMachine(name="VM-1",
instanceUuid="test-uuid-1",
resourcePool=respool.obj,
extra_config=[optionVal])
fake_vms = fake.FakeRetrieveResult()
fake_vms.add_object(virtualMachine1)
session = fake_session(fake_virtual_machines=fake_vms,
fake_resource_pools=fake_pools,
fake_cluster=fake_clusters,
fake_vm_folder=fake_folders)
driver = fake_driver(session, cluster_ref=cluster.obj)
instance = {}
vm_util.add_vm_to_cache(driver, 'test-uuid-1', {
'vmref': virtualMachine1.obj,
"template": "false"
})
# Verify retrieval based on uuid
instance['uuid'] = 'test-uuid-1'
returned_vm_ref = vm_util.get_vm_ref(driver._session, instance)
returned_vm_name = driver._session._call_method(
None, "get_dynamic_property", returned_vm_ref, "VirtualMachine",
"name")
self.assertEquals(virtualMachine1.obj, returned_vm_ref)
self.assertEquals("VM-1", returned_vm_name)
# Verify exception for wrong uuid
instance['uuid'] = 'non-existent-uuid'
self.assertRaises(exception.InstanceNotFound, vm_util.get_vm_ref,
driver._session, instance)