如何用Flask测试客户端模拟AJAX请求?

14 投票
2 回答
4377 浏览
提问于 2025-04-17 11:57

测试Flask应用程序的方法是:

# main.py
from flask import Flask, request

app = flask.Flask(__name__)

@app.route('/')
def index():
    s = 'Hello world!', 'AJAX Request: {0}'.format(request.is_xhr)
    print s
    return s

if __name__ == '__main__':
    app.run()

接下来,这是我的测试脚本:

# test_script.py
import main
import unittest

class Case(unittest.TestCase):
    def test_index():
        tester = app.test_client()
        rv = tester.get('/')
        assert 'Hello world!' in rv.data

if __name__ == '__main__':
    unittest.main()

在测试输出中,我会看到:

Hello world! AJAX Request: False

问题

我该如何用AJAX请求来测试我的应用?

2 个回答

1
import json

def test_index():
    data = json.dumps({})
    client = app.test_client()
    headers = {
        'Content-Type': 'application/json',
    }
    response = client.post('/', data=data, headers=headers)
    data = json.loads(response.data)
    assert data

`

22

试试这个:

def test_index():
    tester = app.test_client()
    response = tester.get('/', headers=[('X-Requested-With', 'XMLHttpRequest')])
    assert 'Hello world!' in response.data

撰写回答