def test_paged_search_generator_with_schema_base_object(self): self.delete_at_teardown.append(add_user(self.connection, testcase_id, 'mat-1')) r = Reader(self.connection, 'inetorgperson', self.delete_at_teardown[0][0]) entries = r.search_paged(2, True, generator=True, attributes=['cn']) cont = 0 for _ in entries: cont += 1 self.assertEqual(cont, 1)
def test_paged_search_accumulator_with_schema(self): self.delete_at_teardown.append(add_user(self.connection, testcase_id, 'mat-1')) self.delete_at_teardown.append(add_user(self.connection, testcase_id, 'mat-2')) self.delete_at_teardown.append(add_user(self.connection, testcase_id, 'mat-3')) self.delete_at_teardown.append(add_user(self.connection, testcase_id, 'mat-4')) self.delete_at_teardown.append(add_user(self.connection, testcase_id, 'mat-5')) r = Reader(self.connection, 'inetorgperson', test_base, 'cn:=' + testcase_id + 'mat-*') entries = r.search_paged(2, True, generator=False, attributes=['cn']) self.assertEqual(len(entries), 5)
def load_groups(settings): groups = [] attrs = [settings.attr.name, settings.attr.var] try: attrs.append(settings.attr.host) group_type = "attributal" new_cls = AttributalGroup except MissingConfigValue: host_attr = None group_type = "structural" new_cls = StructuralGroup _log.info("Loading %s groups from %s" % (group_type, settings.base)) obj_def = ObjectDef(schema=_ldap, object_class=settings.objectclass) reader = Reader(connection=_ldap, base=settings.base, object_def=obj_def, sub_tree=sub(settings)) entries = reader.search_paged(paged_size=_page_size, attributes=attrs) for entry in entries: group = new_cls(entry, settings) group.dn = entry.entry_dn if group.dn == Host.base: # if host base itself is a group, read its vars _log.debug("Found root group at " + group.dn) cls.__ungrouped = group try: group._want_dn = settings.attr.host_is_dn except MissingConfigValue: pass groups.append(group) _log.info("Loaded %i groups" % len(groups)) return groups
def test_paged_search_accumulator_with_schema_base_object(self): self.delete_at_teardown.append(add_user(self.connection, testcase_id, 'mat-1')) r = Reader(self.connection, 'inetorgperson', self.delete_at_teardown[0][0]) entries = r.search_paged(2, True, generator=False, attributes=['cn']) self.assertEqual(len(entries), 1)