RETIRED, further work has moved to Debian project infrastructure
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
deb-python-pyldap/Tests/t_ldif.py

131 lines
3.7 KiB

# -*- coding: utf-8 -*-
import unittest
import textwrap
import ldif
try:
from StringIO import StringIO
except ImportError:
from io import StringIO
class TestParse(unittest.TestCase):
maxDiff = None
def _parse_entry_records(self, ldif_string):
return ldif.ParseLDIF(StringIO(ldif_string))
def _unparse_entry_records(self, records):
f = StringIO()
ldif_writer = ldif.LDIFWriter(f)
for dn, attrs in records:
ldif_writer.unparse(dn, attrs)
return f.getvalue()
def check_roundtrip(self, ldif_source, entry_records):
ldif_source = textwrap.dedent(ldif_source).lstrip() + '\n'
parsed_entry_records = self._parse_entry_records(ldif_source)
parsed_entry_records2 = self._parse_entry_records(
self._unparse_entry_records(entry_records)
)
self.assertEqual(parsed_entry_records, entry_records)
self.assertEqual(parsed_entry_records2, entry_records)
def test_simple(self):
self.check_roundtrip("""
dn: cn=x,cn=y,cn=z
attrib: value
attrib: value2
""", [
('cn=x,cn=y,cn=z', {'attrib': [b'value', b'value2']}),
])
def test_simple2(self):
self.check_roundtrip("""
dn:cn=x,cn=y,cn=z
attrib:value
attrib:value2
""", [
('cn=x,cn=y,cn=z', {'attrib': [b'value', b'value2']}),
])
def test_multiple(self):
self.check_roundtrip("""
dn: cn=x,cn=y,cn=z
a: v
attrib: value
attrib: value2
dn: cn=a,cn=b,cn=c
attrib: value2
attrib: value3
b: v
""", [
('cn=x,cn=y,cn=z', {'attrib': [b'value', b'value2'], 'a': [b'v']}),
('cn=a,cn=b,cn=c', {'attrib': [b'value2', b'value3'], 'b': [b'v']}),
])
def test_folded(self):
self.check_roundtrip("""
dn: cn=x,cn=y,cn=z
attrib: very
long
value
attrib2: %s
""" % ('asdf.' * 20), [
('cn=x,cn=y,cn=z', {'attrib': [b'verylong value'],
'attrib2': [b'asdf.' * 20]}),
])
def test_empty(self):
self.check_roundtrip("""
dn: cn=x,cn=y,cn=z
attrib:
attrib: foo
""", [
('cn=x,cn=y,cn=z', {'attrib': [b'', b'foo']}),
])
def test_binary(self):
self.check_roundtrip("""
dn: cn=x,cn=y,cn=z
attrib:: CQAKOiVA
""", [
('cn=x,cn=y,cn=z', {'attrib': [b'\t\0\n:%@']}),
])
def test_binary2(self):
self.check_roundtrip("""
dn: cn=x,cn=y,cn=z
attrib::CQAKOiVA
""", [
('cn=x,cn=y,cn=z', {'attrib': [b'\t\0\n:%@']}),
])
def test_unicode(self):
self.check_roundtrip("""
dn: cn=Michael Stroeder,dc=stroeder,dc=com
lastname: Ströder
""", [
('cn=Michael Stroeder,dc=stroeder,dc=com',
{'lastname': [b'Str\303\266der']}),
])
def test_sorted(self):
self.check_roundtrip("""
dn: cn=x,cn=y,cn=z
b: value_b
c: value_c
a: value_a
""", [
('cn=x,cn=y,cn=z', {'a': [b'value_a'],
'b': [b'value_b'],
'c': [b'value_c']}),
])
if __name__ == '__main__':
unittest.main()