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