python - Flask-WTF - 无法测试表单提交

Question

我想用 crsf token 测试 wtf-form，但我不知道如何发送 token 。

这是我的表格:

class myLoginForm(FlaskForm):
    username = StringField()
    password = StringField()
    mySubmit = SubmitField('Save')

这是我的路线:

@app.route('/login', methods=['GET', 'POST'])
def login():
    loginForm = myLoginForm()

    if loginForm.validate_on_submit():
        result = request.form
        username = result.get("username")
        password = result.get("password")

这是我的测试:

import unittest
from flask_testing import TestCase
from flask import Flask
import json

class TestLogin(TestCase):

    def create_app(self):
        app = Flask(__name__)
        return app

    def test_submission(self):
        headers = {
            'ContentType': 'application/json',
            'dataType': 'json'
        }
        data = {
            'username': 'foo',
            'password': 'bar'
        }

        response = app.test_client().post(
            '/login',
            data=json.dumps(data),
            content_type='application/json',
            follow_redirects=True
        )

        assert self.get_context_variable("loginForm").validate_on_submit() == True

断言失败，因为 validate_on_submit() 返回 False。我认为这是由于 crsf token 。

如何将 crsf token 发送到 POST 请求？

祝你有个愉快的一天

Answer 1

也许这可以帮到你。 Make it easier to access a CSRF token in automated tests