def inject_vars(): # 函数名可以随意修改
"""模板上下文处理函数"""
from watchlist.models import User
user = User.query.first() # 用户对象
if not user:
user = User()
user.name = 'BL00D'
return locals() # 需要返回字典
def setUp(self):
"""更新配置"""
app.config.update(TESTING=True,
SQLALCHEMY_DATABASE_URI='sqlite:///:memory:')
db.create_all()
user = User()
user.name = "Test"
user.password = "******"
movie = Movie()
movie.title = "Test Movie"
movie.year = '1900'
db.session.add_all([user, movie])
db.session.commit()
self.client = app.test_client() # 创建测试客户端
self.ruuner = app.test_cli_runner() # 创建测试命令运行器
def admin(username, password):
"""create_user"""
db.create_all()
user = User.query.first()
if user is not None:
click.echo("updating user")
user.name = username
user.password = password
else:
click.echo("create user")
user = User()
user.name = username
user.password = password
db.session.add(user)
db.session.commit()
click.echo("Done.")
def setUp(self):
# 更新配置
app.config.update(TESTING=True,
SQLALCHEMY_DATABASE_URI='sqlite:///:memory:')
# 创建数据库和表
db.create_all()
# 创建测试数据，一个用户，一个测试条目
user = User()
user.username = '******'
user.name = 'Test'
user.set_password('test')
movie = Movie(title='Test Movie Title', year='2019')
# 使用add_all()方法一次添加多个模型类实例，传入列表
db.session.add_all([user, movie])
db.session.commit()
self.client = app.test_client() # 创建测试客户端
self.runner = app.test_cli_runner() # 创建测试命令运行器