def setUpClass(cls):
try:
cls.client = webdriver.Chrome('./chromedriver')
except Exception:
pass
if not cls.client:
return
cls.app = create_app('test')
cls.app_context = cls.app.app_context()
cls.app_context.push()
# 禁止日志
import logging
logger = logging.getLogger('werkzeug')
logger.setLevel('ERROR')
db.create_all()
Role.insert_roles()
Category.insert_categories()
FakeUtil.generate_fake_users(10)
FakeUtil.generate_fake_articles(10)
admin_role = Role.query.filter_by(name='Administrator').first()
admin = User(email='*****@*****.**',
username='******',
password='******',
role=admin_role,
confirmed=True)
db.session.add(admin)
db.session.commit()
threading.Thread(target=cls.app.run).start()
def deploy():
"""部署"""
from flask_migrate import upgrade
# 迁移数据库到最新版本
upgrade()
Role.insert_roles()
Category.insert_categories()
def setUp(self):
self.app = create_app('test')
self.app_context = self.app.app_context()
self.app_context.push()
self.client = self.app.test_client(use_cookies=True)
db.create_all()
Role.insert_roles()
Category.insert_categories()
