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 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 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()
def teardown(self):
with self.app.app_context():
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()
