You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.0 KiB
52 lines
1.0 KiB
11 years ago
|
# -*- coding: utf-8 -*-
|
||
|
|
import os
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from {{ cookiecutter.app_name }}.settings import TestConfig
|
||
|
|
from {{cookiecutter.app_name}}.app import create_app
|
||
|
|
from {{cookiecutter.app_name}}.database import db as _db
|
||
|
|
|
||
|
|
from .factories import ALL_FACTORIES
|
||
|
|
|
||
|
|
@pytest.yield_fixture(scope='session')
|
||
|
|
def app():
|
||
|
|
_app = create_app(TestConfig)
|
||
|
|
ctx = _app.app_context()
|
||
|
|
ctx.push()
|
||
|
|
|
||
|
|
yield _app
|
||
|
|
|
||
|
|
ctx.pop()
|
||
|
|
|
||
|
|
@pytest.yield_fixture(scope='session')
|
||
|
|
def db(app):
|
||
|
|
_db.app = app
|
||
|
|
with app.app_context():
|
||
|
|
_db.create_all()
|
||
|
|
|
||
|
|
yield _db
|
||
|
|
|
||
|
|
_db.drop_all()
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.yield_fixture(scope='function')
|
||
|
|
def session(db):
|
||
|
|
conn = db.engine.connect()
|
||
|
|
transaction = conn.begin()
|
||
|
|
|
||
|
|
opts = {'bind': conn, 'binds': {}}
|
||
|
|
_session = db.create_scoped_session(options=opts)
|
||
|
|
|
||
|
|
# Set session for each factory class
|
||
|
|
for FactoryClass in ALL_FACTORIES:
|
||
|
|
FactoryClass.FACTORY_SESSION = _session
|
||
|
|
|
||
|
|
db.session = _session
|
||
|
|
|
||
|
|
yield _session
|
||
|
|
|
||
|
|
transaction.rollback()
|
||
|
|
conn.close()
|
||
|
|
_session.remove()
|