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()