-
Notifications
You must be signed in to change notification settings - Fork 136
Expand file tree
/
Copy patht_bind.py
More file actions
53 lines (40 loc) · 1.66 KB
/
t_bind.py
File metadata and controls
53 lines (40 loc) · 1.66 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
import os
import unittest
# Switch off processing .ldaprc or ldap.conf before importing _ldap
os.environ['LDAPNOINIT'] = '1'
import ldap
from ldap.ldapobject import LDAPObject
from slapdtest import SlapdTestCase
class TestBinds(SlapdTestCase):
unicode_val = "abc\U0001f498def"
unicode_val_bytes = unicode_val.encode('utf-8')
dn_unicode = "CN=" + unicode_val
dn_bytes = 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)
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()