def test_get_host_name_for_vm(self): fake_vm = fake.ManagedObject( "VirtualMachine", fake.ManagedObjectReference("vm-123", "VirtualMachine")) fake_vm.propSet.append(fake.Property('name', 'vm-123')) fake_objects = fake.FakeRetrieveResult() fake_objects.add_object(fake_vm) vm_ref = vm_util.get_vm_ref_from_name(fake_session(fake_objects), 'vm-123') self.assertIsNotNone(vm_ref) fake_results = [ fake.ObjectContent(None, [ fake.Property( 'runtime.host', fake.ManagedObjectReference('host-123', 'HostSystem')) ]) ] fake_objects = fake.FakeRetrieveResult() for results in fake_results: fake_objects.add_object(results) host_id = vm_util.get_host_id_from_vm_ref(fake_session(fake_objects), vm_ref) self.assertEqual('host-123', host_id)
def _create_fake_vms(self): fake_vms = fake.FakeRetrieveResult() OptionValue = collections.namedtuple('OptionValue', ['key', 'value']) for i in range(10): vm = fake.ManagedObject() opt_val = OptionValue(key='', value=5900 + i) vm.set(vm_util.VNC_CONFIG_KEY, opt_val) fake_vms.add_object(vm) return fake_vms