X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=tests%2Fdbutils.py;h=bb5a362a18e0b576065b76bfd80c3a247945b64e;hb=HEAD;hp=c315fff5e4914ecc2ece30d59b81d3de1301c2d2;hpb=6a5ab6e9ddddda1e97b5fb64c2342a76cc807a3f;p=m_librarian.git

diff --git a/tests/dbutils.py b/tests/dbutils.py
index c315fff..06a9e9c 100644
--- a/tests/dbutils.py
+++ b/tests/dbutils.py
@@ -1,5 +1,8 @@
 
 import os
+
+from sqlobject.tests.dbtest import getConnection
+
 from m_librarian.db import open_db, init_db
 from m_librarian.inp import import_inpx
 
@@ -7,19 +10,23 @@ __all__ = ['setup_module', 'teardown_module', 'load_inpx']
 
 
 def setup_module():
-    try:
-        os.remove('/tmp/m_librarian-test.sqlite')
-    except OSError:
-        pass
-    open_db('sqlite:///tmp/m_librarian-test.sqlite')
+    connection = getConnection()
+    if connection.dbName == 'sqlite':
+        try:
+            connection.dropDatabase()
+        except OSError:
+            pass
+    open_db(connection.uri())
     init_db()
 
 
 def teardown_module():
-    try:
-        os.remove('/tmp/m_librarian-test.sqlite')
-    except OSError:
-        pass
+    connection = getConnection()
+    if connection.dbName == 'sqlite':
+        try:
+            connection.dropDatabase()
+        except OSError:
+            pass
 
 
 def load_inpx(inpx):