def test_get_vm_names(self):
"""Test the get vm names method."""
children = []
for ident in range(1, 3):
name = 'vm' + str(ident)
config = Mock()
config.name = name
summary = Mock()
summary.config = config
child = Mock()
child.summary = summary
children.append(child)
vm_container_view = Mock(view=children)
vm_names = get_vm_names(vm_container_view)
self.assertTrue(isinstance(vm_names, list))
self.assertEqual(vm_names, ['vm1', 'vm2'])
def test_get_vm_names(self):
"""Test the get vm names method."""
objects = [
vim.ObjectContent(
obj=vim.VirtualMachine('vm-1'),
propSet=[vim.DynamicProperty(name='name', val='vm1')]),
vim.ObjectContent(
obj=vim.VirtualMachine('vm-2'),
propSet=[vim.DynamicProperty(name='name', val='vm2')]),
]
content = Mock()
content.rootFolder = vim.Folder('group-d1')
content.propertyCollector.RetrievePropertiesEx(ANY).token = None
content.propertyCollector.RetrievePropertiesEx(ANY).objects = objects
vm_names = get_vm_names(content)
self.assertTrue(isinstance(vm_names, list))
self.assertEqual(vm_names, ['vm1', 'vm2'])