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.")
