132 lines
3.7 KiB
Python
132 lines
3.7 KiB
Python
# -*- 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()
|