def setUp(self, env='test'): self.app = create_app(env) # Allow overriding of database URI from the environment. db_uri_override = os.environ.get('DATABASE_URI') if db_uri_override: self.app.config['SQLALCHEMY_DATABASE_URI'] = db_uri_override self.db = db db.app = self.app db.reflect() db.drop_all() db.create_all() self.create_brand_and_party() self.create_admin()
def setUp(self, config_filename=CONFIG_FILENAME_TEST): self.app = create_app(config_filename) # Allow overriding of database URI from the environment. db_uri_override = os.environ.get('DATABASE_URI') if db_uri_override: self.app.config['SQLALCHEMY_DATABASE_URI'] = db_uri_override self.db = db db.app = self.app db.reflect() db.drop_all() db.create_all() self.create_brand_and_party() self.create_admin()
def execute(): click.echo('Creating database tables ... ', nl=False) db.create_all() click.secho('done.', fg='green')