Fix tests

JVT038 · Jan 29, 2024 · 5678e0a · 5678e0a
1 parent d98ff33
commit 5678e0a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_database.py b/tests/test_database.py
@@ -1,4 +1,3 @@
-from email.policy import default
 from config import Config
 from metatube.init import Default
 from metatube import create_app, db
@@ -39,6 +38,11 @@ def testAutoCreation(self):
         db.drop_all()
         relative_url = 'sqlite:///' + os.path.join('metatube', Config.SQLALCHEMY_DATABASE_URI.replace('sqlite:///', ''))
         default = Default(self.app, relative_url)
+        migrationsPath = os.path.join(Config.BASE_DIR, 'migrations')
+        if os.path.exists(migrationsPath) and os.path.isdir(migrationsPath) and (os.path.exists(os.path.join(migrationsPath, 'env.py')) and os.path.exists(os.path.join(migrationsPath, 'versions'))):
+            default.init_db()
+        else:
+            default.init_db(False)
         self.assertTrue(default.init_db())
 
     def testConfigTable(self):