python - Flask View 测试 : How do I prevent data from being saved to the database?

Question

我有一个带有 SQLAlchemy 的应用程序，已初始化:

config_name = os.getenv('FLASK_CONFIG') or 'default'
app = Flask(__name__)
db = SQLAlchemy(app)
db.init_app(app)

我正在处理的 View 仅以 json 格式返回所有用户:

@app.route('/users', methods=['GET'])
def users():
    users = db.session.query(User)

    output = {'users': [user.to_json() for user in users]}
    return jsonify(**output)

还有我的测试:

class UserViewTest(BaseTestCase, CreateUserMixin):
    def test_users(self):
        user1 = self._make_user() # defined in the above mixin
        user2 = self._make_user()
        user2.email = 'hello@abc.com'
        user2.username = 'hello@abc.com'
        db.session.add_all([user1, user2])
        db.session.flush()
        response = app.test_client().get('/users')

我使用 FLASK_CONFIG=testing nosetests 运行我的测试(我检查过，app.testing 设置为 true)。

对我的 psql 数据库运行这个，我发现我添加到数据库中的这些用户被保存了!

triller_social_test=# SELECT username FROM users;
   username    
---------------
 foobar
 hello@abc.com
(2 rows)

如何阻止这种情况发生？我尝试覆盖 db.session.commit() 什么也不做，但是当我调用 View 时数据库回滚。是否有使用 test_client() 的替代方法？我无法直接调用该方法，因为 jsonify 不允许我在响应之外的任何地方返回数据。

更新:这是我的临时(？)解决方案:

def json_response_converter(dict_):
    if config_name == 'testing':
        return dict_
    else:
        return jsonify(**dict_)

现在我不需要在我的测试中使用 jsonify()

Answer 1

将 test_client 与 flush 和 rollback 一起使用是正确的方法。你对 jsonify 有什么顾虑？我看不出这与您的主要问题有什么联系。

也许您的拆解回滚没有按照您的意愿被调用。包括在线回滚允许您验证状态是否清晰作为测试的一部分。

这里有一个要点，表明这种方法有效。 https://gist.github.com/bhtucker/58a90e59265a0994bf42a23921d69718

您应该能够在具有 psql 工具和适当的 python 依赖项的环境中简单地运行 bash setup_and_test.sh。