From 2949e780efe25ae65f9e893a952d3e85abf73972 Mon Sep 17 00:00:00 2001
From: pyldap contributors <python-ldap@python.org>
Date: Fri, 24 Nov 2017 10:47:14 +0100
Subject: [PATCH] Tests: Add a test suite for edits

---
 Tests/__init__.py |  1 +
 Tests/t_edit.py   | 89 +++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 90 insertions(+)
 create mode 100644 Tests/t_edit.py

diff --git a/Tests/__init__.py b/Tests/__init__.py
index 1ca8d20..46b3861 100644
--- a/Tests/__init__.py
+++ b/Tests/__init__.py
@@ -18,4 +18,5 @@
 from . import t_ldapurl
 from . import t_ldif
 from . import t_ldapobject
+from . import t_edit
 from . import t_ldap_schema_subentry
diff --git a/Tests/t_edit.py b/Tests/t_edit.py
new file mode 100644
index 0000000..9aee43e
--- /dev/null
+++ b/Tests/t_edit.py
@@ -0,0 +1,89 @@
+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 EditionTests(unittest.TestCase):
+
+    def setUp(self):
+        global server
+        if server is None:
+            server = SlapdObject()
+            server.start()
+            base = server.suffix
+            suffix_dc = base.split(',')[0][3:]
+
+            # insert some Foo* objects via ldapadd
+            server.ldapadd("\n".join([
+                'dn: '+server.suffix,
+                'objectClass: dcObject',
+                'objectClass: organization',
+                'dc: '+suffix_dc,
+                'o: '+suffix_dc,
+                '',
+                'dn: '+server.root_dn,
+                'objectClass: applicationProcess',
+                'cn: '+server.root_cn,
+                '',
+                "dn: cn=Foo1,"+base,
+                "objectClass: organizationalRole",
+                "cn: Foo1",
+                "",
+                "dn: cn=Foo2,"+base,
+                "objectClass: organizationalRole",
+                "cn: Foo2",
+                "",
+                "dn: cn=Foo3,"+base,
+                "objectClass: organizationalRole",
+                "cn: Foo3",
+                "",
+                "dn: ou=Container,"+base,
+                "objectClass: organizationalUnit",
+                "ou: Container",
+                "",
+                "dn: cn=Foo4,ou=Container,"+base,
+                "objectClass: organizationalRole",
+                "cn: Foo4",
+                "",
+            ])+"\n")
+
+        l = LDAPObject(server.ldap_uri, bytes_mode=False)
+        l.protocol_version = 3
+        l.set_option(ldap.OPT_REFERRALS,0)
+        l.simple_bind_s(server.root_dn,
+                server.root_pw)
+        self.ldap = l
+        self.server = server
+
+    def test_add_object(self):
+        base = self.server.suffix
+        dn = "cn=Added,ou=Container," + base
+        self.ldap.add_ext_s(dn, [
+            ("objectClass", [b'organizationalRole']),
+            ("cn", [b'Added']),
+        ])
+
+        # Lookup the object
+        result = self.ldap.search_s(base, ldap.SCOPE_SUBTREE, '(cn=Added)', ['*'])
+        self.assertEqual(result, [
+            ("cn=Added,ou=Container," + base,
+                {'cn': [b'Added'], 'objectClass': [b'organizationalRole']}),
+        ])
+
+
+if __name__ == '__main__':
+    unittest.main()