28 lines
687 B
Python
28 lines
687 B
Python
import unittest
|
|
import flaskbase
|
|
import contextlib
|
|
|
|
from flaskbase.extensions import db
|
|
|
|
|
|
class TestCase(unittest.TestCase):
|
|
"""
|
|
Base testcase class
|
|
"""
|
|
|
|
def setUp(self):
|
|
self.flask_app = flaskbase.create_app(settings={
|
|
'SQLALCHEMY_DATABASE_URI': 'sqlite://',
|
|
})
|
|
|
|
self.app = self.flask_app.test_client()
|
|
flaskbase.init_db(self.flask_app)
|
|
|
|
def tearDown(self):
|
|
db.session.remove()
|
|
with contextlib.closing(db.engine.connect()) as con:
|
|
trans = con.begin()
|
|
for table in reversed(db.Model.metadata.sorted_tables):
|
|
con.execute(table.delete())
|
|
trans.commit()
|