ordr2/ordr/tests/_functional/__init__.py

''' functional tests for ordr2 '''

import pytest
import transaction
import webtest

from .. import APP_SETTINGS

WEBTEST_SETTINGS = APP_SETTINGS.copy()
# WEBTEST_SETTINGS['pyramid.includes'].append('pyramid_mailer.testing')


class CustomTestApp(webtest.TestApp):
    ''' might add custom functionality to webtest.TestApp '''
    pass


def create_users(dbsession):
    ''' create example users '''
    pass


@pytest.fixture(scope='module')
def testapp():
    ''' fixture for using webtest '''
    from ordr.models.meta import Base
    from ordr.models import get_tm_session
    from ordr import main

    app = main({}, **WEBTEST_SETTINGS)
    testapp = CustomTestApp(app)

    session_factory = app.registry['dbsession_factory']
    engine = session_factory.kw['bind']
    Base.metadata.create_all(engine)

    with transaction.manager:
        # set up test data here
        dbsession = get_tm_session(session_factory, transaction.manager)
        create_users(dbsession)

    yield testapp

    Base.metadata.drop_all(engine)