如何使用 pytest 测试 Flask 视图上下文和模板？

Question

我在 Flask 中使用 pytest 并想测试我的视图和模板，但我不清楚如何最好地做到这一点。

我知道我可以测试 HTML 输出的内容，例如：

def test_my_view(test_client):
    # test_client is a fixture representing app.test_client()
    response = test_client.get("/my-url")
    assert b"<h1>My page title</h1>" in response.data

但有些事情我不知道怎么做最好：

1. 如何测试视图正在使用哪个模板？

2. 如何测试视图发送到模板的上下文？ （例如检查login_form是LoginForm一个实例）

3. 如果我想测试是否存在更复杂的 HTML 标记，例如具有正确action属性的<form>标记，则是检查整个标记是否存在的唯一方法（例如<form method="get" class="form-lg" action="/other-url"> ) 即使我不关心其他属性？ 假设页面上也有其他表单，我怎么能只检查action呢？

Answer 1

我已经意识到 1 和 2 可以通过类似这个问题中的解决方案来解决，稍微改变以与 pytest 一起使用。

假设我们有这个 Flask 视图：

from flask import render_template
from app import app

@app.route("/my/view")
def my_view():
    return render_template("my/template.html", greeting="Hello!")

我们想要测试调用该 URL 使用正确的模板，并且传递给它的上下文数据正确。

首先，创建一个可重用的夹具：

from flask import template_rendered
import pytest

@pytest.fixture
def captured_templates(app):
    recorded = []

    def record(sender, template, context, **extra):
        recorded.append((template, context))

    template_rendered.connect(record, app)
    try:
        yield recorded
    finally:
        template_rendered.disconnect(record, app)

我还有一个test_client夹具用于在测试中发出请求（类似于Flask Cookiecutter 中的testapp夹具或本教程中的test_client夹具）。

然后写你的测试：

def test_my_view(test_client, captured_templates):
    response = test_client.get("/my/view")

    assert len(captured_templates) == 1

    template, context = captured_templates[0]

    assert template.name = "my/template.html"

    assert "greeting" in context
    assert context["greeting"] == "Hello!"

请注意，你可能有一个以上的元素captured_templates ，这取决于你的观点做什么。