ficture _db now comes with a prepopulated Genre table, and _app has an app_context right away
This commit is contained in:
parent
653021c2ae
commit
67259d689b
@ -1,6 +1,7 @@
|
|||||||
import pytest
|
import pytest
|
||||||
from the_works import create_app
|
from the_works import create_app
|
||||||
from the_works.database import db as _db
|
from the_works.database import db as _db
|
||||||
|
from the_works.models import Genre
|
||||||
|
|
||||||
TEST_DATABASE_URI = "sqlite:///:memory:"
|
TEST_DATABASE_URI = "sqlite:///:memory:"
|
||||||
|
|
||||||
@ -15,15 +16,21 @@ def _app():
|
|||||||
_app = create_app(test_config)
|
_app = create_app(test_config)
|
||||||
|
|
||||||
# other setup can go here
|
# other setup can go here
|
||||||
|
context = _app.app_context()
|
||||||
|
context.push()
|
||||||
|
|
||||||
yield _app
|
yield _app
|
||||||
|
|
||||||
# clean up / reset resources here
|
# clean up / reset resources here
|
||||||
|
context.pop()
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(scope="function")
|
@pytest.fixture(scope="function")
|
||||||
def db(_app):
|
def db(_app):
|
||||||
with _app.app_context():
|
with _app.app_context():
|
||||||
|
_db.session.add(Genre(ID=1, Genre="spam"))
|
||||||
|
_db.session.add(Genre(ID=2, Genre="eggs"))
|
||||||
|
_db.session.commit()
|
||||||
yield _db
|
yield _db
|
||||||
_db.drop_all()
|
_db.drop_all()
|
||||||
_db.create_all()
|
_db.create_all()
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user