def list_all_users():
app = create_app()
with app.app_context():
users = User.query.all()
for user in users:
print(user.username)
def flask_client(context, *args, **kwargs):
"""Sets up temporary flask context
Generator function.
Yields:
A flask testing client attribute to Behave's context.
"""
app = create_app(TestConfig)
# context.client = app.test_client()
# context.client = create_app( TestConfig )
with app.test_client() as testClient:
# with app.app_context():
context.client = testClient
yield context.client
def splinter_browser(context, type='chrome'):
"""Sets up a splinter browser session
Generator function that provides a web driver browser object for use in
testing. The browser defaults to a Chrome web driver.
Args:
context: Behave testing context
type: Specific webdriver to run
Yields:
A browser object attribute to the Behave testing module's context. The
type of the browser is set to whichever the generator is provided.
"""
if type != 'flask':
context.browser = Browser(driver_name=type)
else:
context.browser = Browser(driver_name=type, app=create_app(TestConfig))
yield context.browser
context.browser.quit()
from blogsite import create_app
application = app = create_app()
if __name__ == '__main__':
app.run()
from blogsite import create_app, db
from blogsite.models import User, Post
app = create_app()
@app.shell_context_processor
def make_shell_context():
return { 'db': db, 'User': User, 'Post': Post }