def test_iterate_group_members_with_pagination(self):
with mock_ldap() as ldap:
for dn in ["cn=AwesomeFolk", "cn=*Guys"]:
(it, err) = ldap.iterate_group_members({"group_dn": dn},
page_size=1)
self.assertIsNone(err)
results = list(it)
self.assertEqual(2, len(results))
first = results[0][0]
second = results[1][0]
if first.id == "testy":
testy, someuser = first, second
else:
testy, someuser = second, first
self.assertEqual("testy", testy.id)
self.assertEqual("testy", testy.username)
self.assertEqual("*****@*****.**", testy.email)
self.assertEqual("someuser", someuser.id)
self.assertEqual("someuser", someuser.username)
self.assertEqual("*****@*****.**", someuser.email)
def test_iterate_group_members(self):
with mock_ldap() as ldap:
(it, err) = ldap.iterate_group_members(
{"group_dn": "cn=AwesomeFolk"}, disable_pagination=True
)
self.assertIsNone(err)
results = list(it)
self.assertEquals(2, len(results))
first = results[0][0]
second = results[1][0]
if first.id == "testy":
testy, someuser = first, second
else:
testy, someuser = second, first
self.assertEquals("testy", testy.id)
self.assertEquals("testy", testy.username)
self.assertEquals("*****@*****.**", testy.email)
self.assertEquals("someuser", someuser.id)
self.assertEquals("someuser", someuser.username)
self.assertEquals("*****@*****.**", someuser.email)
def test_iterate_group_members_with_pagination(self):
with mock_ldap() as ldap:
for dn in ['cn=AwesomeFolk', 'cn=*Guys']:
(it, err) = ldap.iterate_group_members({'group_dn': dn},
page_size=1)
self.assertIsNone(err)
results = list(it)
self.assertEquals(2, len(results))
first = results[0][0]
second = results[1][0]
if first.id == 'testy':
testy, someuser = first, second
else:
testy, someuser = second, first
self.assertEquals('testy', testy.id)
self.assertEquals('testy', testy.username)
self.assertEquals('*****@*****.**', testy.email)
self.assertEquals('someuser', someuser.id)
self.assertEquals('someuser', someuser.username)
self.assertEquals('*****@*****.**', someuser.email)
def test_ldap_user_filtering_valid_users(self):
valid_user_filter = "(filterField=somevalue)"
with mock_ldap(user_filter=valid_user_filter) as ldap:
# Verify we can login.
(response, _) = ldap.verify_and_link_user("someuser", "somepass")
self.assertEquals(response.username, "someuser")
(it, err) = ldap.iterate_group_members(
{"group_dn": "cn=AwesomeFolk"}, disable_pagination=True
)
self.assertIsNone(err)
results = list(it)
self.assertEquals(2, len(results))
def test_ldap_user_filtering_no_users(self):
no_user_filter = "(filterField=anothervalue)"
with mock_ldap(user_filter=no_user_filter) as ldap:
# Verify we cannot login.
(response, _) = ldap.verify_and_link_user("someuser", "somepass")
assert response is None
(it, err) = ldap.iterate_group_members(
{"group_dn": "cn=AwesomeFolk"}, disable_pagination=True
)
self.assertIsNone(err)
results = list(it)
self.assertEquals(0, len(results))
