python-3.x - Pytest Flask应用程序属性错误: module 'src.api' has no attribute 'test_client'

Question

我正在尝试对“hello world”flask 应用程序进行基本的 Pytest 测试请在下面查看我在 src 文件中的内容

api.py:

from flask import Flask, jsonify


api = Flask(__name__)


@api.route('/')
def hello_world():
    return jsonify(message="hello world")


if __name__ == '__main__':
    api.run(debug=True)

这是我为测试写的

test_api.py

import pytest
from src import api


api.testing = True
client = api.test_client()


def test_route(client):
    response = api.get('/')
    assert response.status_code == 200

结构

my_project
    __init__.py
    src
        __init__.py
        api.py
    test
        __init__.py
        test_api.py

我使用 python -m pytest 从根目录运行测试
我收到的错误信息是

test/test_api.py:13: in <module>
    with api.test_client() as client:
E   AttributeError: module 'src.api' has no attribute 'test_client'

我真的不确定如何让它工作。

Answer 1

from src import api 导入src/api.py

的模块

但是，您对该src.api 模块范围内的全局api 对象感兴趣

from src.api import api 将导入 flask 应用程序对象，该对象应该具有您正在调用的 test_client 方法。