Skip to content

Commit

Permalink
Tests: Add a test suite for binds
Browse files Browse the repository at this point in the history
  • Loading branch information
pyldap contributors authored and Petr Viktorin committed Nov 24, 2017
1 parent 388eafb commit 4549352
Show file tree
Hide file tree
Showing 2 changed files with 83 additions and 0 deletions.
1 change: 1 addition & 0 deletions Tests/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@

from __future__ import absolute_import

from . import t_bind
from . import t_cext
from . import t_cidict
from . import t_ldap_dn
Expand Down
82 changes: 82 additions & 0 deletions Tests/t_bind.py
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()

0 comments on commit 4549352

Please sign in to comment.