def resetdb(): from app.utils import timedelta import datetime db.drop_all() db.create_all() Problem.import_from_hustoj() Role.insert_role() admin = User(email = '*****@*****.**', username = '******', password = '******', confirmed = True, role = Role.query.filter_by(name = "Administrator").first()) db.session.add(admin) db.session.commit() judger = User(email = "*****@*****.**", username = "******", password = "******", confirmed = True, role = Role.query.filter_by(name = "Judger").first()) db.session.add(judger) db.session.commit() judger = User(email = "*****@*****.**", username = "******", password = "******", confirmed = True, role = Role.query.filter_by(name = "Judger").first()) db.session.add(judger) db.session.commit() contest = Contest(title = "test", begin_time = datetime.datetime.utcnow() - timedelta(hours = 10), end_time = datetime.datetime.utcnow() + timedelta(hours = 10)) db.session.add(contest) for i in range(8): cp = ContestProblem(contest_id = 1, problem_id = i + 1, problem_relative_id = i) db.session.add(cp) db.session.commit()
def tearDown(self): """ Reset the test cases. """ db.session.remove() db.drop_all() self.app_context.pop()
def tearDown(self): ''' 数据库和程序上下文在 tearDown() 方法中删除。 ''' db.session.remove() db.drop_all() self.app_context.pop()
def tearDown(self): """Remove data from database""" assert self.cls_initialized db.drop_all() self.cls_initialized = False print('TestWeb Class Teardown\n')
def teardown(): app = create_app('test') with app.app_context(): db.session.remove() db.drop_all() db.engine.execute("drop table alembic_version") db.get_engine(app).dispose()
def recreate_tables(): """ dangerous """ if query_yes_no('This will erase all data in the database. Continue?', default='no'): db.drop_all() db.create_all()
def setUp(self): db.session.close() db.drop_all() db.create_all() self.app = app.test_client() if not os.path.exists('users/'): os.makedirs('users/')
def db(app, request): _db.app = app _db.create_all() yield _db _db.drop_all()
def tearDown(self): """database removal""" assert self.cls_initialized db.drop_all() self.cls_initialized = False print('TestAPI Class Teardown\n')
def create_dev_database(): """ Use this when running a development server. It creates tables without using db migration scripts, instead asking SQLAlchemy to create them directly from your models. Good for testing/iterating. When done, write proper db migration scripts and test. """ from app import db from app.models import Role, User db.drop_all() db.create_all() admin_role = Role(name='Admin') mod_role = Role(name='Moderator') user_role = Role(name='User') db.session.add(admin_role) db.session.add(mod_role) db.session.add(user_role) user_john = User(username='******', role=admin_role, password='******', email='*****@*****.**', confirmed=True) user_susan = User(username='******', role=user_role, password='******', email='*****@*****.**', confirmed=False) user_david = User(username='******', role=user_role, password='******', email='*****@*****.**', confirmed=False) db.session.add(user_john) db.session.add(user_susan) db.session.add(user_david) db.session.commit()
def teardown(): _db.session.rollback() _db.session.remove() if _db.engine.dialect.name == 'postgresql': _db.engine.execute('drop schema if exists public cascade') _db.engine.execute('create schema public') _db.drop_all()
def tearDown(self): """ Flask will automatically remove database sessions at the end of the request or when the application shuts down """ db.session.remove() # close the session db.drop_all() self.app_context.pop()
def createdb(): print 'creating db..' db.drop_all() db.create_all() Role.insert_roles() print 'adding feed sources..' with open('sources.yaml', 'r') as f: feed_sources = yaml.load(f) # Insert feedprovider for site in feed_sources: provider_name = unicode(site['name']) provider_domain = site['domain'] provider = FeedProvider(name = provider_name, domain = provider_domain, favicon = site['favicon']) db.session.add(provider) db.session.commit() # Add feedsource for channel_name in site['channels']: source_name = unicode(site['name'] + ' - ' + channel_name) source_href = site['channels'][channel_name] fs = FeedSource(name=source_name, url=source_href, provider=provider) db.session.add(fs) db.session.commit() print 'add feed articles..' from app.mod_crawler.fetch import update_db update_db()
def dropdb(): """Drops database tables""" if prompt_bool("Are you sure you want to lose all your data"): destroy_db() db.drop_all() db_setup() db.create_all()
def recreate_data(): """Generate fake data for development and testing""" print 'dropping existing tables...' db.drop_all() print 'creating tables...' db.create_all() print 'inserting roles...' Role.insert_roles() print 'creating admin user...' u1 = User(username='******', email='*****@*****.**', password='******', name='Ralph Wen', location='Hangzhou, China', about_me='This is the creator of everything', confirmed=True) db.session.add(u1) print 'creating moderate user...' u2 = User(username='******', email='*****@*****.**', password='******', name='Yadong Wen', location='Hangzhou, China', about_me='Yeah', confirmed=True) db.session.add(u2) db.session.commit() print 'generating 50 normal users...' User.generate_fake(50) print 'generating 500 posts...' Post.generate_fake(500) print 'generating follows...' User.generate_following() print 'generating comments...' Comment.generate_comments() print '...done'
def teardown_func(): "tear down test fixtures" db.session.remove() db.drop_all() app.config.from_object('config.DevelopmentConfig') app.config['WTF_CSRF_ENABLED'] = True app.config['CSRF_ENABLED'] = True
def recreateDatabase(): """ Function to recreate database tables, make a user for testing, and create application roles. """ print("Dropping and recreating database tables.") db.drop_all() db.create_all() print("Cleansing user uploads folder.") shutil.rmtree('app/static/uploads/1/') os.mkdir('app/static/uploads/1') os.chmod('app/static/uploads/1', mode=0o777) print("Creating application roles and first user.") user = User(password=bc.generate_password_hash('la73ralu5', rounds=12), user_name='cshaw') from app.roles import active_role, admin_role, verified_role db.session.add(admin_role) db.session.add(active_role) db.session.add(verified_role) db.session.flush() user.roles.append(admin_role) user.roles.append(active_role) user.roles.append(verified_role) db.session.add(user) db.session.commit()
def recreate_schema(test_class): # print 'Re-creating db schema for environmet %s'%flapp.config['REDIS_PREFIX'] db.session.remove() db.drop_all() db.create_all() # setup_func(test_class) return setup_func
def forged(): from forgery_py import basic, lorem_ipsum, name, internet, date from random import randint db.drop_all() db.create_all() Role.seed() guests = Role.query.first() def generate_comment(func_author, func_post): return Comment(body=lorem_ipsum.paragraphs(), created=date.date(past=True), author=func_author(), post=func_post()) def generate_post(func_author): return Post(title=lorem_ipsum.title(), body=lorem_ipsum.paragraphs(), created=date.date(), author=func_author()) def generate_user(): return User(name=internet.user_name(), email=internet.email_address(), password=basic.text(6, at_least=6, spaces=False), role=guests) users = [generate_user() for i in range(0, 5)] db.session.add_all(users) random_user = lambda: users[randint(0, 4)] posts = [generate_post(random_user) for i in range(0, randint(50, 200))] db.session.add_all(posts) random_post = lambda: posts[randint(0, len(posts) - 1)] comments = [generate_comment(random_user, random_post) for i in range(0, randint(2, 100))] db.session.add_all(comments) db.session.commit()
def tearDown(self): # ? db.session.remove() # drop tables db.drop_all() # pop app context self.app_context.pop()
def setUp(self): self.app = create_app('testing') # app context self.app_context = self.app.app_context() self.app_context.push() # drop database in case it exissts, and create new version db.drop_all() db.create_all() # test client self.client = self.app.test_client() # stormpath client self.sp_client = Client( id=environ.get('STORMPATH_API_KEY_ID'), secret=environ.get('STORMPATH_API_KEY_SECRET'), ) # retrieve photog_test application instance href = 'https://api.stormpath.com/v1/applications/Nq2qQQegOh9a9qJRekxeA' self.application = self.sp_client.applications.get(href) self.test_email = '*****@*****.**' self.test_password = '******' self.user = self.create_default_user()
def rebuild(): if config == 'development': with app.app_context(): db.drop_all() db.create_all() Role.insert_roles() admin_me = User(email=app.config['FLASKR_ADMIN'], username=app.config['MY_USERNAME'], password=app.config['MY_PASSWORD'], confirmed=True, name=forgery_py.name.full_name(), location=forgery_py.address.city(), about_me=forgery_py.lorem_ipsum.sentences(10), member_since=forgery_py.date.date(True, min_delta=10)) db.session.add(admin_me) ordinary_me = User(email=forgery_py.internet.email_address(), username='******', password='******', confirmed=True, name=forgery_py.name.full_name(), location=forgery_py.address.city(), about_me=forgery_py.lorem_ipsum.sentences(10), member_since=forgery_py.date.date(True, min_delta=10)) db.session.add(ordinary_me) db.session.commit() User.generate_fake(30) Post.generate_fake(500) Follow.generate_fake(500) Comment.generate_fake(500) else: print('Permission denied.')
def init_data(): from app import db db.drop_all() db.create_all() User.init_data() Category.init_data() return 'ok'
def init(): """ Fixture for test initialisation """ # App configuration overide app.config['TESTING'] = True app.config['WTF_CSRF_ENABLED'] = False app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite://' app.config['SERVER_NAME'] = 'localhost' # DB creation db.create_all() user = User(nickname='utest', email='*****@*****.**', password='******', active=True, confirmed_at=datetime.utcnow(), auth_token='ddg56@dgfdG°dkjvk,') db.session.add(user) db.session.commit() # Context setup context = app.app_context() context.push() yield db.session.remove() db.drop_all() context.pop()
def remove_database(app, db): db.session.remove() db.drop_all() try: os.remove(app.config['TEST_DB_FILEPATH']) except FileNotFoundError: pass
def tearDown(self): try: shutil.rmtree(app.config['UPLOAD_FOLDER']) except OSError: pass db.session.remove() db.drop_all()
def setUp(self): self.app = create_app('testing') self.app_context = self.app.app_context() self.client = self.app.test_client() self.app_context.push() db.drop_all() db.create_all()
def tearDown(self): shutil.rmtree(app.config['UPLOADED_IMAGES_DEST']) shutil.rmtree(app.config['THUMBNAIL_DIR']) shutil.rmtree(app.config['IMAGE_ROOT_DIR']) db.session.remove() db.drop_all() os.unlink(os.path.join(basedir, testDbPath))
def tearDown(self): # Wipes the testing database and remove the test application # context. db.session.remove() db.drop_all() self.app_context.pop()
def setUp(self): _app = create_app("config.TestConfig") ctx = _app.test_request_context() ctx.push() db.session.commit() db.drop_all() db.create_all()
def setUp(self): db.drop_all() db.create_all()
def recreate_db(): # Recreates local db. Do not use in production db.drop_all() db.create_all() db.session.commit()
def tearDown(self): db.session.remove() db.drop_all() self.app_context.pop()
from app import db, Users db.drop_all() db.create_all() testuser = Users(first_name='Grooty', last_name='Toot') db.session.add(testuser) db.session.commit()
def app(): with flask_app.app_context(): # db.reflect() # https://github.com/sqlalchemy/sqlalchemy/wiki/DropEverything db.drop_all() db.create_all() yield flask_app
def tearDown(self): db.session.remove() db.drop_all()
def init_db(): """ Initialize the database.""" db.drop_all() db.create_all() create_users() create_feeds()
def tearDown(self): #pylint: disable=invalid-name """Remove our instance of the app and the associated in-memory database.""" db.session.remove() db.drop_all() self.app_context.pop()
def tearDown(self): '''teardown all initialized variables.''' with self.app.app_context(): # drop all tables db.session.remove() db.drop_all()
def tearDown(self): """Destroy a blank temp database before each test.""" db.drop_all()
def dropdb(): if prompt_bool('Are you sure you eant to lose all your data'): db.drop_all()
def drop_db(): """Drop all rows and tables from SQL database. """ db.engine.execute("DROP EXTENSION IF EXISTS postgis CASCADE;") db.drop_all()
def setUp(self): self.app = create_app(TestConfig) self.app_context = self.app.app_context() self.app_context.push() db.drop_all() db.create_all()
def tearDown(selfs): # 清除数据库会话 db.session.remove() # 删除数据库 db.drop_all()
def drop(): "Drops database tables" if prompt_bool("Are you sure to lose all your data?"): db.drop_all()
def tearDownClass(cls): db.session.remove() db.drop_all() cls.app_context.pop()
def init_db(): db.drop_all() db.create_all()
def tearDown(self): '''在每个测试后运行, 清理测试环境''' db.session.remove() db.drop_all()
def tearDownClass(self): db.drop_all()
def tearDown(self): """ Will be called after every test """ db.session.remove() db.drop_all()
def teardown(): _db.drop_all() os.unlink(test_db_file)
def deleteAll(): """delete database""" db.drop_all()
def tearDown(self): db.drop_all()
def tearDown(self): with self.app.app_context(): db.session.remove() db.reflect() db.drop_all()
def tearDown(self): shutil.rmtree(self.path) db.session.remove() db.drop_all()
def tearDown(): print("Rodando tear down...\n") db.drop_all()
def setUp(self): self.app = create_app('testing') self.app_context = self.app.app_context() self.app_context.push() db.drop_all() db.create_all()
def drop(): """ drop the database """ db.drop_all() db.create_all()