1 files changed, 61 insertions, 0 deletions
diff --git a/src/tests/intg/ds.py b/src/tests/intg/ds.py
new file mode 100644
index 000000000..a87190275
--- /dev/null
+++ b/src/tests/intg/ds.py
@@ -0,0 +1,61 @@
+#
+# Abstract directory server instance class
+#
+# Copyright (c) 2015 Red Hat, Inc.
+# Author: Nikolai Kondrashov <Nikolai.Kondrashov@redhat.com>
+#
+# This is free software; you can redistribute it and/or modify it
+# under the terms of the GNU General Public License as published by
+# the Free Software Foundation; version 2 only
+#
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+
+import ldap
+
+class DS:
+    """Abstract directory server instance."""
+
+    def __init__(self, dir, port, base_dn, admin_rdn, admin_pw):
+        """
+            Initialize the instance.
+
+            Arguments:
+            dir         Path to the root of the filesystem hierarchy to create
+                        the instance under.
+            port        TCP port on localhost to bind the server to.
+            base_dn     Base DN.
+            admin_rdn   Administrator DN, relative to BASE_DN.
+            admin_pw    Administrator password.
+        """
+        self.dir        = dir
+        self.port       = port
+        self.ldap_url   = "ldap://localhost:" + str(self.port)
+        self.base_dn    = base_dn
+        self.admin_rdn  = admin_rdn
+        self.admin_dn   = admin_rdn + "," + base_dn
+        self.admin_pw   = admin_pw
+
+    def setup(self):
+        """Setup the instance"""
+        raise NotImplementedError()
+
+    def teardown(self):
+        """Teardown the instance"""
+        raise NotImplementedError()
+
+    def bind(self):
+        """Connect to the server and bind as admin, return connection."""
+        conn = ldap.initialize(self.ldap_url)
+        conn.simple_bind_s(self.admin_dn, self.admin_pw)
+        return conn
+
+    def __del__(self):
+        """Destroy the instance."""
+        self.teardown()