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)
