Add simple SqlSession implementation

[cascardo/ipsilon.git] / tests / helpers / common.py

diff --git a/tests/helpers/common.py b/tests/helpers/common.py
index d020a3c..e316718 100755 (executable)
--- a/tests/helpers/common.py
+++ b/tests/helpers/common.py
@@ -20,7 +20,6 @@
 
 import ConfigParser
 import io
-from ipsilon.util.plugin import PluginObject
 import os
 import pwd
 import shutil
@@ -29,10 +28,9 @@ from string import Template
 import subprocess
 
 
-class IpsilonTestBase(PluginObject):
+class IpsilonTestBase(object):
 
     def __init__(self, name, execname):
-        super(IpsilonTestBase, self).__init__()
         self.name = name
         self.execname = execname
         self.rootdir = os.getcwd()
@@ -119,11 +117,31 @@ class IpsilonTestBase(PluginObject):
 
         return http_conf_file
 
+    def setup_pgdb(self, datadir, env):
+        cmd = ['/usr/bin/pg_ctl', 'initdb', '-D', datadir]
+        subprocess.check_call(cmd, env=env)
+        auth = 'host all all 127.0.0.1/24 trust\n'
+        filename = os.path.join(datadir, 'pg_hba.conf')
+        with open(filename, 'a') as f:
+            f.write(auth)
+
     def start_http_server(self, conf, env):
         p = subprocess.Popen(['/usr/sbin/httpd', '-DFOREGROUND', '-f', conf],
                              env=env, preexec_fn=os.setsid)
         self.processes.append(p)
 
+    def start_pgdb_server(self, datadir, rundir, log, addr, port, env):
+        p = subprocess.Popen(['/usr/bin/pg_ctl', 'start', '-D', datadir, '-o',
+                              '-c unix_socket_directories=%s -c port=%s -c \
+                               listen_addresses=%s' % (rundir, port, addr),
+                              '-l', log, '-w'],
+                             env=env, preexec_fn=os.setsid)
+        self.processes.append(p)
+        p.wait()
+        for d in ['adminconfig', 'userprefs', 'transactions', 'sessions']:
+            cmd = ['/usr/bin/createdb', '-h', addr, '-p', port, d]
+            subprocess.check_call(cmd, env=env)
+
     def wait(self):
         for p in self.processes:
             os.killpg(p.pid, signal.SIGTERM)