def test_db(test_client):
# Create the database and the database table
db.create_all()
yield db # this is where the testing happens!
# anything after yield is teardown code
db.session.rollback()
db.session.remove()
db.drop_all()
def init_models():
with app.test_request_context():
db.drop_all()
db.create_all()
user_admin = User(username='******', fullname='admin-user')
user_admin._set_password('admin-user129_Ftz')
user_admin_email = UserEmail(email='*****@*****.**',
user=user_admin)
user = User(username='******', fullname='member-user')
user._set_password('member-user341_Wer')
user_email = UserEmail(email='*****@*****.**', user=user)
profile_owner = User(username='******',
fullname='profile-cypress')
profile_owner._set_password('profile-cypress123_St')
profile_owner_email = UserEmail(email='*****@*****.**',
user=profile_owner)
concierge = User(username='******', fullname='concierge-user')
concierge._set_password('concierge-user34_qQE')
concierge_email = UserEmail(email='*****@*****.**',
user=concierge)
usher = User(username='******', fullname='usher-cypress')
usher._set_password('usher-cypress566_YUt')
usher_email = UserEmail(email='*****@*****.**', user=usher)
editor = User(username='******', fullname='editor-cypress')
editor._set_password('editor-cypress9_GH')
editor_email = UserEmail(email='*****@*****.**', user=editor)
user2 = User(username='******', fullname='hg-user')
user2._set_password('hg-user5_HE')
db.session.add_all([
user_admin,
user_admin_email,
user,
user_email,
profile_owner,
profile_owner_email,
concierge,
concierge_email,
usher,
usher_email,
editor,
editor_email,
user2,
])
db.session.commit()
def test_db_structure(test_client):
# Create all database tables
db.create_all()
yield db
# Drop all database tables
db.drop_all()
def tearDown(self):
self.session.rollback()
db.drop_all()
self.ctx.pop()
def drop_models():
with app.test_request_context():
db.drop_all()
def tearDown(self):
self.session.rollback()
db.drop_all()
self.ctx.pop()