def create_all_tables():
db.create_all()
# Create initial data
user = UserModel.find_by(username='******')
if not user:
user = UserModel('Alex', '*****@*****.**', '123456', is_admin=True)
db.session.add(user)
db.session.commit()
def setUp(self):
"""
It runs once for every test method in this class and derived classes
:return: None
"""
# Make sure your database is created
with BaseTest.flaskApp.app_context():
try:
db.create_all()
except OperationalError as ex:
BaseTest.flaskApp.logger.error(
'Database initialization error . Error: {}'.format(
str(ex)))
except DatabaseError as ex:
BaseTest.flaskApp.logger.error(
'Error in database connection: {}\nError: {}'.format(
ex.statement, str(ex)))
# Get a test client and app context for integration and system tests.
self.app_context = BaseTest.flaskApp.app_context
self.client = BaseTest.flaskApp.test_client