38 lines
1.0 KiB
Python
38 lines
1.0 KiB
Python
|
from app import app
|
||
|
|
|
||
|
|
# Great class to inherit from, as it sets up and tears down
|
||
|
|
# classes efficiently when we have a database in place.
|
||
|
|
class BaseTest:
|
||
|
|
|
||
|
|
auths = {}
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def setUpClass(cls):
|
||
|
|
app.config.from_object('config.testing')
|
||
|
|
cls.ctx = app.test_request_context()
|
||
|
|
cls.app = app.test_client()
|
||
|
|
# Great place to do a db.create_all()
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def tearDownClass(cls):
|
||
|
|
# Create place to clear everything out ...
|
||
|
|
# db.drop_all()
|
||
|
|
# db.session.remove()
|
||
|
|
# elastic_index.clear()
|
||
|
|
pass
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.ctx.push()
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
# db.session.rollback()
|
||
|
|
# Most efficient thing here is to delete all rows from
|
||
|
|
# the database with a clear db method like this one:
|
||
|
|
# def clean_db(db):
|
||
|
|
# for table in reversed(db.metadata.sorted_tables):
|
||
|
|
# db.session.execute(table.delete())
|
||
|
|
# clean_db(db)
|
||
|
|
self.ctx.pop()
|
||
|
|
self.auths = {}
|
||
|
|
|