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 }