def test_query(self):
with mock_ldap() as ldap:
# Lookup cool.
(response, federated_id, error_message) = ldap.query_users("cool")
self.assertIsNone(error_message)
self.assertEquals(1, len(response))
self.assertEquals("ldap", federated_id)
user_info = response[0]
self.assertEquals("cool.user", user_info.username)
self.assertEquals("*****@*****.**", user_info.email)
# Lookup unknown.
(response, federated_id, error_message) = ldap.query_users("unknown")
self.assertIsNone(error_message)
self.assertEquals(0, len(response))
self.assertEquals("ldap", federated_id)
def test_timeout(self):
base_dn = ['dc=quay', 'dc=io']
admin_dn = 'uid=testy,ou=employees,dc=quay,dc=io'
admin_passwd = 'password'
user_rdn = ['ou=employees']
uid_attr = 'uid'
email_attr = 'mail'
secondary_user_rdns = ['ou=otheremployees']
with self.assertRaisesRegexp(Exception, "Can't contact LDAP server"):
ldap = LDAPUsers('ldap://localhost',
base_dn,
admin_dn,
admin_passwd,
user_rdn,
uid_attr,
email_attr,
secondary_user_rdns=secondary_user_rdns,
requires_email=False,
timeout=5)
ldap.query_users('cool')
def test_timeout(self):
base_dn = ["dc=quay", "dc=io"]
admin_dn = "uid=testy,ou=employees,dc=quay,dc=io"
admin_passwd = "password"
user_rdn = ["ou=employees"]
uid_attr = "uid"
email_attr = "mail"
secondary_user_rdns = ["ou=otheremployees"]
with self.assertRaisesRegexp(Exception, "Can't contact LDAP server"):
ldap = LDAPUsers(
"ldap://localhost",
base_dn,
admin_dn,
admin_passwd,
user_rdn,
uid_attr,
email_attr,
secondary_user_rdns=secondary_user_rdns,
requires_email=False,
timeout=5,
)
ldap.query_users("cool")
