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")