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()
def test_roles_and_permissions(self): Role.insert_roles() u = User(email='*****@*****.**', password='******') self.assertTrue(u.can(Permission.WRITE_ARTICLE)) self.assertFalse(u.can(Permission.MODERATE_COMMENT))