Teach DB to create schema in existing connection

2020-10-08 20:43:48 +03:00
parent ea3d617e9c
commit 7878cadb30
2 changed files with 10 additions and 1 deletions
--- a/kdoorweb/tests/test_db.py
+++ b/kdoorweb/tests/test_db.py
@@ -4,9 +4,15 @@ from kdoorweb.db import DB


 class TestDB(unittest.TestCase):
+    def setUp(self) -> None:
+        self.db = DB()
+
    def test_create_db_in_memory(self):
        DB.create_db(dbfile=":memory:")

+    def test_create_db_in_connection(self):
+        DB.create_db(dbfile=self.db)
+

 if __name__ == '__main__':
    unittest.main()