Question

我有一个使用SQLAlchemy的应用程序，已初始化：

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

和我正在处理的视图只返回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（）什么都不做，但是当我调用视图时数据库回滚。有没有使用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

您应该能够在具有bash setup_and_test.sh工具和正确的python依赖项的环境中运行psql。

Flask视图测试：如何防止数据保存到数据库？

1 个答案: