4

我有一个带有 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()

4

1 回答 1

1

test_client并排使用flushandrollback是正确的方法。你对 jsonify 有什么顾虑？我没有看到与您的主要问题的联系。

也许您的拆卸回滚没有按照您的意愿调用。包括内联回滚允许您在测试中验证状态是否清晰。

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

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

于 2016-04-01T20:50:19.087 回答