def setUp(self):
app.config['WTF_CSRF_ENABLED'] = False
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///:memory:'
app.config['TESTING'] = True
#set up memory database
db.app = app
db.create_all()
def reset_database():
log.info('Resetting test database')
db.drop_all()
db.create_all()
# creating testdatabase
py = Category(name='Python')
Post(title='Hello Python!', body='Python is pretty cool', category=py)
p = Post(title='Snakes', body='Ssssssss')
py.posts.append(p)
db.session.add(py)
db.session.commit()
def setUp(self):
self.app = create_app(config='test')
self.app_context = self.app.app_context()
self.app_context.push()
db.create_all()
def setUp(self):
self.app = create_app(TestConfig)
db.create_all()
from celery import Celery
from flask_sqlalchemy import SQLAlchemy
from app.mod_photos.models import Photo
from app.data import db
celery = Celery('tasks')
celery.config_from_object('celeryconfig')
#TODO temporal task test
db.create_all()
def execute_analysis(photo):
data = {'info': 'result!', 'analysed': True}
photo.update(**data)
@celery.task
def analyse_document():
photos = Photo.query.all()
for photo in photos:
if not photo.analysed:
execute_analysis(photo)
def setUp(self):
app.config['TESTING']=True
app.config['WTF_CSRF_ENABLED']=False
db.create_all()
import psycopg2
from migrate.versioning import api
from config import SQLALCHEMY_DATABASE_URI
from config import SQLALCHEMY_MIGRATE_REPO
from app import app
from app.data import db
import os.path
db.init_app(app)
with app.app_context():
db.create_all()
if not os.path.exists(SQLALCHEMY_MIGRATE_REPO):
api.create(SQLALCHEMY_MIGRATE_REPO, 'database repository')
api.version_control(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO)
else:
api.version_control(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO, api.version(SQLALCHEMY_MIGRATE_REPO))
def create_db(num_users=5):
"""Creates data tables and populates them."""
db.create_all()
populate_db(num_users=num_users)
