Use variable when creating/deleting test db

diff --git a/ut/test_db.py b/ut/test_db.py
index b76ef34a2123ef664d3b1fd3e4f297ad5637d68e..a50b88aea65c30a43b49a67a4a8967c4f55edd72 100644 (file)
--- a/ut/test_db.py
+++ b/ut/test_db.py
@@ -7,8 +7,8 @@ from cbdb import Db
 
 TESTDB_FILENAME = ".tmptest.db"
 
-def create_test_db():
-    con = connect(TESTDB_FILENAME)
+def create_test_db(fn):
+    con = connect(fn)
     cur = con.cursor()
     cur.executescript("""
 pragma foreign_keys = ON;
@@ -66,8 +66,8 @@ INSERT INTO coffees VALUES(12,'1111','Club-Mate 0,33 l','2019-03-20 11:22:46');
     """)
     con.close()
 
-def delete_test_db():
-    remove(TESTDB_FILENAME)
+def delete_test_db(fn):
+    remove(fn)
 
 class CoffeesTable(TestCase):
     def test_db_exists(self):
@@ -76,10 +76,10 @@ class CoffeesTable(TestCase):
         self.assertIsNotNone(db.cur)
 
     def test_test_db_exists(self):
-        create_test_db()
+        create_test_db(TESTDB_FILENAME)
         try:
             db = Db(TESTDB_FILENAME)
             self.assertIsNotNone(db.con)
             self.assertIsNotNone(db.cur)
         finally:
-            delete_test_db()
+            delete_test_db(TESTDB_FILENAME)