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