Question

我尝试使用 Flask-Testing 测试 Flask-SQLAlchemy 模型。更准确地说，这个模型的静态方法使用first_or_404()，我无法找到让我的测试工作的方法。

这是一个突出问题的自包含示例：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from flask.ext.testing import TestCase

db = SQLAlchemy()

class ModelToTest(db.Model):
    __tablename__ = 'model_to_test'
    identifier = db.Column(db.String(80), unique=True, nullable=False, primary_key=True)

    @staticmethod
    def get_by_identifier(identifier):
        return ModelToTest.query.filter_by(identifier=identifier).first_or_404()

class Config:
    DEBUG = True
    TESTING = True
    SQLALCHEMY_DATABASE_URI = 'sqlite:///'
    SQLALCHEMY_TRACK_MODIFICATIONS = False

class TestGetByIdentifier(TestCase):

    def create_app(self):
        app = Flask('test')
        app.config.from_object(Config())
        db.init_app(app)
        return app

    def setUp(self):
        db.create_all()

    def tearDown(self):
        db.session.remove()
        db.drop_all()

    def test_get_by_identifier(self):
        self.assert404(ModelToTest.get_by_identifier('identifier'))

我收到了错误：

(my_env) PS C:\Dev\Test\Python\test_flask> nosetests-3.4.exe
E
======================================================================
ERROR: test_get_by_identifier (test_flask.TestGetByIdentifier)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "C:\Dev\Test\Python\test_flask\test_flask.py", line 37, in test_get_by_identifier
    self.assert404(ModelToTest.get_by_identifier('identifier'))
  File "C:\Dev\Test\Python\test_flask\test_flask.py", line 13, in get_by_identifier
    return ModelToTest.query.filter_by(identifier=identifier).first_or_404()
  File "c:\\my_env\lib\site-packages\flask_sqlalchemy\__init__.py", line 431, in first_or_404
    abort(404)
  File "c:\\my_env\lib\site-packages\werkzeug\exceptions.py", line 646, in __call__
    raise self.mapping[code](*args, **kwargs)
werkzeug.exceptions.NotFound: 404: Not Found

----------------------------------------------------------------------
Ran 1 test in 0.913s

因此，self.assert404(ModelToTest.get_by_identifier('identifier'))行确实会在first_or_404()调用中生成异常，而此异常是werkzeug.exceptions.NotFound，它似乎不是{{1}的预期}}

运行此测试的要求是：

烧瓶
烧瓶-SQLAlchemy的
烧瓶-测试

值得注意的是，当我在应用程序中使用该函数时，它的行为符合预期。

提前致谢。

Answer 1

我引用了我在GitHub上收到的答案：

https://github.com/jarus/flask-testing/issues/89

我认为这是对驾驶考试方式的误解。   first_or_404函数确实会引发NotFound异常。什么时候   在请求的上下文中，异常将冒出来   处理完毕，变成了404 http响应，这就是什么   烧瓶测试正在寻找。

但是，在这种情况下，您不在请求的上下文中   你直接调用这个方法，它只是导致了一个   例外。您可以这样做以使测试工作
from werkzeug.exceptions import NotFound

def test_get_by_identifier(self):
    with self.assertRaises(NotFound):
        ModelToTest.get_by_identifier('identifier')
或者，您可以将该功能放在路线后面并使用它进行测试 self.client会正确地给你一个404 http响应。但是，测试异常（不使用烧瓶测试）可能考虑到你所测试的等级，在这种情况下更合适。

Flask-Testing和Flask-SQLAlchemy：first_or_404（）

1 个答案: