def database(app): with app.app_context(): db.drop_all() db.create_all() yield db
def setup(self): app = create_test_app() self.app = app self.client = app.test_client() with app.app_context(): db.drop_all() db.create_all()
def drop_db(): app = Flask(__name__) app.config.from_object(os.environ['APP_SETTINGS']) db.init_app(app) with app.app_context(): db.drop_all() return None
def init_empty_database(app): with app.app_context(): # Create the database db.create_all() yield # tests run here # Destroy the database db.drop_all()
def setup(self): os.environ['MODE'] = 'TESTING' app = create_app() self.app = app self.client = app.test_client() with app.app_context(): db.drop_all() db.create_all()
def create_db(app): """ A test fixture for creating a database table, with the schema defined in model, once per test class and deleted after exiting the scope of a test class. """ db.app = app with app.app_context(): db.create_all() yield db db.session.close() db.drop_all()
def db(app): _db.app = app connection = _db.engine.connect() config = Config(os.path.abspath('alembic.ini')) config.attributes['connection'] = connection command.upgrade(config, 'head') yield _db connection.close() _db.drop_all()
def app_old(): """ Fixture app returning an instance of Flask app with created models in the database """ app = create_app("testing") with app.app_context(): db.create_all() yield app db.drop_all()
def init_database(app, colors_list): with app.app_context(): # Create the database db.create_all() # Insert color data for color in colors_list: db.session.add(Color(color=color['color'], value=color['value'])) # Commit the changes db.session.commit() yield # tests run here # Destroy the database db.drop_all()
def clear(): """Clear DB""" db.drop_all() db.create_all() # add admin user admin_user = User(username='******', email='*****@*****.**', password='******', admin=True) db.session.add(admin_user) db.session.commit() flash("Database refreshed") return redirect(url_for('main_bp.chat'))
def run(): with app.app_context(): db.drop_all() db.create_all() # Administrator admin = User(email="*****@*****.**") db.session.add(admin) # First user user1 = User(email="*****@*****.**") db.session.add(user1) # Second user user2 = User(email="*****@*****.**") db.session.add(user2) db.session.commit()
def run(): with app.app_context(): print(f"ran users script. DB: {db}") db.drop_all() db.create_all() # Admin user admin = User(email="*****@*****.**") db.session.add(admin) # First base user user1 = User(email="*****@*****.**") db.session.add(user1) # Second base user user2 = User(email="*****@*****.**") db.session.add(user2) print(f"user.first(): {User.query.first()}") db.session.commit()
def db_rebuild(): db.drop_all() db.create_all() return "Database rebuilt!"
def teardown(self): with self.app.app_context(): db.drop_all()
def app(): """Flask application fixture """ _app = build_app(env='test') db.create_all() yield _app db.drop_all()
from application.models import db, Alert db.drop_all() db.create_all() print Alert.query.all()
from application.models import db, UserAdmin db.drop_all() db.create_all() # create admin first user adm_user = UserAdmin(email='*****@*****.**', password='******') db.session.add(adm_user) db.session.commit()
def resetdb(): """drop database tables""" db.drop_all()