def reset_db(self):
db.drop_all()
db.create_all()
# flaskr.init_db()
db.session.add(Role(role_type='admin'))
db.session.add(Role(role_type='teacher'))
db.session.add(Role(role_type='student'))
db.session.add(Role(role_type='parent'))
db.session.add(User(username='******', password='******', role_type='admin'))
db.session.add(User(username='******', password='******', role_type='teacher'))
db.session.add(User(username='******', password='******', role_type='student'))
db.session.add(User(username='******', password='******', role_type='parent'))
db.session.add(User(username='******', password='******', role_type='student'))
db.session.add(User(username='******', password='******', role_type='student'))
db.session.add(Day(name='Monday'))
db.session.add(Day(name='Tuesday'))
db.session.add(Day(name='Wednesday'))
db.session.add(Day(name='Thursday'))
db.session.add(Day(name='Friday'))
db.session.add(Day(name='Saturday'))
db.session.add(Day(name='Sunday'))
db.session.commit()
def tearDown(self):
""" Ensures that the database is emptied for next unit test """
app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite://"
db.init_app(app)
with app.app_context():
db.drop_all()
def tearDown(self):
""" Ensures that the database is emptied for next unit test """
with app.app_context():
db.drop_all()
