-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
pyldap contributors
authored and
Petr Viktorin
committed
Nov 24, 2017
1 parent
388eafb
commit 4549352
Showing
2 changed files
with
83 additions
and
0 deletions.
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,82 @@ | ||
| from __future__ import unicode_literals | ||
|
|
||
| import sys | ||
|
|
||
| if sys.version_info[0] <= 2: | ||
| PY2 = True | ||
| text_type = unicode | ||
| else: | ||
| PY2 = False | ||
| text_type = str | ||
|
|
||
| import ldap, unittest | ||
| from slapdtest import SlapdObject | ||
| from ldap.ldapobject import LDAPObject | ||
|
|
||
| server = None | ||
|
|
||
|
|
||
| class TestBinds(unittest.TestCase): | ||
|
|
||
| def setUp(self): | ||
| global server | ||
| if server is None: | ||
| server = SlapdObject() | ||
| server.start() | ||
|
|
||
| self.server = server | ||
| self.unicode_val = "abc\U0001f498def" | ||
| self.unicode_val_bytes = self.unicode_val.encode('utf-8') | ||
|
|
||
| self.dn_unicode = "CN=" + self.unicode_val | ||
| self.dn_bytes = self.dn_unicode.encode('utf-8') | ||
|
|
||
| def _get_ldapobject(self, bytes_mode=None): | ||
| l = LDAPObject(self.server.ldap_uri, bytes_mode=bytes_mode) | ||
| l.protocol_version = 3 | ||
| l.set_option(ldap.OPT_REFERRALS,0) | ||
| return l | ||
|
|
||
| def test_simple_bind(self): | ||
| l = self._get_ldapobject(False) | ||
| with self.assertRaises(ldap.INVALID_CREDENTIALS): | ||
| l.simple_bind_s(self.dn_unicode, self.unicode_val) | ||
|
|
||
| def test_unicode_bind(self): | ||
| l = self._get_ldapobject(False) | ||
| l.simple_bind(self.dn_unicode, "ascii") | ||
|
|
||
| l = self._get_ldapobject(False) | ||
| l.simple_bind("CN=user", self.unicode_val) | ||
|
|
||
| @unittest.skipUnless(PY2, "no bytes_mode under Py3") | ||
| def test_unicode_bind_bytesmode(self): | ||
| l = self._get_ldapobject(True) | ||
| with self.assertRaises(TypeError): | ||
| l.simple_bind_s(self.dn_unicode, self.unicode_val_bytes) | ||
|
|
||
| with self.assertRaises(TypeError): | ||
| l.simple_bind_s(self.dn_bytes, self.unicode_val) | ||
|
|
||
| # Works when encoded to UTF-8 | ||
| with self.assertRaises(ldap.INVALID_CREDENTIALS): | ||
| l.simple_bind_s(self.dn_bytes, self.unicode_val_bytes) | ||
|
|
||
| def test_unicode_bind_no_bytesmode(self): | ||
| l = self._get_ldapobject(False) | ||
| with self.assertRaises(TypeError): | ||
| l.simple_bind_s(self.dn_bytes, self.unicode_val) | ||
|
|
||
| # Works fine in Python 3 because 'cred' (the password) is read in | ||
| # using the "s#" format which, unlike "s", accepts either a str | ||
| # (unicode) *or* bytes. | ||
| # | ||
| # with self.assertRaises(TypeError): | ||
| # l.simple_bind_s(self.dn_unicode, self.unicode_val_bytes) | ||
|
|
||
| with self.assertRaises(ldap.INVALID_CREDENTIALS): | ||
| l.simple_bind_s(self.dn_unicode, self.unicode_val) | ||
|
|
||
|
|
||
| if __name__ == '__main__': | ||
| unittest.main() |