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.
25 lines
639 B
25 lines
639 B
# -*- coding: utf-8 -*-
|
|
from flask.ext.testing import TestCase
|
|
from {{ cookiecutter.app_name }}.settings import Config
|
|
from {{ cookiecutter.app_name }}.app import create_app
|
|
from {{ cookiecutter.app_name }}.database import db
|
|
|
|
|
|
class TestConfig(Config):
|
|
TESTING = True
|
|
DEBUG = True
|
|
SQLALCHEMY_DATABASE_URI = 'sqlite://'
|
|
|
|
|
|
class DbTestCase(TestCase):
|
|
"""Base TestCase for tests that require a database."""
|
|
|
|
def create_app(self):
|
|
app = create_app(TestConfig)
|
|
with app.app_context():
|
|
db.create_all()
|
|
return app
|
|
|
|
def tearDown(self):
|
|
db.session.remove()
|
|
db.drop_all()
|
|
|