diff --git a/src/database.py b/src/database.py index 0fca9ab4e09a9d1ae753d5334f67fe52f8e14318..58f5fcb393325ee26033f671219ff08bcf7e35db 100644 --- a/src/database.py +++ b/src/database.py @@ -29,6 +29,12 @@ def create_db(my_app): return my_db +def create_db_test(my_app): + my_db = SQLAlchemy(my_app) + my_db.create_all() + return my_db + + def create_celery(my_app): celery = Celery( my_app.import_name, diff --git a/tests/conftest.py b/tests/conftest.py index c9fb3684d8434fedca2409099c428835b420ec1b..da2cabc147d01c14fef08aa95bba113814d7920c 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -1,7 +1,7 @@ import os import tempfile import pytest -from database import create_app, create_db, create_admin +from database import create_app, create_db_test, create_admin @pytest.fixture @@ -18,8 +18,8 @@ def client(): with app.test_client() as client: with app.app_context(): - db = create_db(app) - admin = create_admin(app) + create_db_test(app) + create_admin(app) yield client os.close(db_fd)