feat: implement Slapd.init_tree

azmeuk · azmeuk · commit 8782654d0877 · 2023-08-13T15:50:20.000+02:00
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -3,6 +3,11 @@ All notable changes to this project will be documented in this file.
 The format is based on `Keep a Changelog <https://keepachangelog.com/en/1.0.0/>`_,
 and this project adheres to `Semantic Versioning <https://semver.org/spec/v2.0.0.html>`_.
 
+Added
+*****
+
+- Implement `init_tree` method :pr:`8`
+
 [0.1.3] - 2023-03-12
 ====================
 
diff --git a/README.md b/README.md
@@ -9,6 +9,7 @@ pip install slapd
 >>> import slapd
 >>> process = slapd.Slapd()
 >>> process.start()
+>>> process.init_tree()
 >>> process.ldapwhoami().stdout.decode("utf-8")
 'dn:cn=manager,dc=slapd-test,dc=python-ldap,dc=org\n'
 >>> process.stop()
diff --git a/slapd/__init__.py b/slapd/__init__.py
@@ -621,3 +621,25 @@ def slapcat(self, extra_args=None, expected=0):
             extra_args=extra_args,
             expected=expected,
         )
+
+    def init_tree(self):
+        """
+        Creates the organization and applicationProcess object.
+        """
+        suffix_dc = self.suffix.split(",")[0][3:]
+        return self.ldapadd(
+            "\n".join(
+                [
+                    "dn: " + self.suffix,
+                    "objectClass: dcObject",
+                    "objectClass: organization",
+                    "dc: " + suffix_dc,
+                    "o: " + suffix_dc,
+                    "",
+                    "dn: " + self.root_dn,
+                    "objectClass: applicationProcess",
+                    "cn: " + self.root_cn,
+                ]
+            )
+            + "\n"
+        )
diff --git a/tests/test_slapdobject.py b/tests/test_slapdobject.py
@@ -35,17 +35,9 @@ def test_commands():
     )
     server.ldapsearch("ou=home", "dc=slapd-test,dc=python-ldap,dc=org", expected=32)
 
+    server.init_tree()
+
     ldif = (
-        "dn: dc=slapd-test,dc=python-ldap,dc=org\n"
-        "objectClass: dcObject\n"
-        "objectClass: organization\n"
-        "dc: slapd-test\n"
-        "o: slapd-test\n"
-        "\n"
-        "dn: cn=Manager,dc=slapd-test,dc=python-ldap,dc=org\n"
-        "objectClass: applicationProcess\n"
-        "cn: Manager\n"
-        "\n"
         "dn: ou=home,dc=slapd-test,dc=python-ldap,dc=org\n"
         "objectClass: organizationalUnit\n"
         "ou: home\n"
