def add_default_users(db):
try:
UserRole.add_default_roles()
User.add_default_users()
except sqlalchemy.exc.IntegrityError as e:
db.session.rollback()
print(e)
def admin_login(app, client):
Role.add_default_roles()
User.add_default_users()
return client.post(url_for('account.login'),
data=dict(email=app.config['ADMIN_EMAIL'],
password=app.config['ADMIN_PASSWORD']),
follow_redirects=True)
def db_create():
""" Creates a local database
"""
with app.app_context():
db.create_all()
Role.add_default_roles()
User.add_default_users()
ControlLists.add_default_lists()
db.session.commit()
click.echo("Created the database")
def setUp(self):
"""Setup the test driver and create test users"""
db.session.commit()
db.drop_all()
db.create_all()
db.session.commit()
# add default roles & admin user
Role.add_default_roles()
User.add_default_users()
self.create_driver()
self.driver.get(self.get_server_url())
def db_recreate():
""" Recreates a local database. You probably should not use this on
production.
"""
with app.app_context():
db.drop_all()
db.create_all()
Role.add_default_roles()
User.add_default_users()
ControlLists.add_default_lists()
db.session.commit()
click.echo("Dropped then recreated the database")
def add_default_users(db):
UserRole.add_default_roles()
db.session.flush()
User.add_default_users()
