Question

我无法在Flask Rest API测试中模拟API调用函数调用。我正在使用unittest和pytest模块。

代码如下：

rest_api.py


@app.route('/api/v1/recommender', methods=['POST'])
def recommender():
    """Handle POST requests that are sent to /api/v1/recommender REST API endpoint."""
    r = {'recommendation': 'failure', 'external_request_id': None}
    metrics_payload = {
        'pid': os.getpid(),
        'hostname': os.environ.get("HOSTNAME"),
        'endpoint': request.endpoint,
        'request_method': request.method,
        'status_code': 200
    }

    input_json = request.get_json()

        try:
            persist = request.args.get('persist', 'true') == 'true'
            r = RecommendationTask().execute(input_json, persist=persist,
                                             check_license=check_license)
        except Exception as e:
            r = {
                'recommendation': 'unexpected error',
                'external_request_id': input_json.get('external_request_id'),
                'message': '%s' % e
            }
            metrics_payload['status_code'] = 500

    return flask.jsonify(r), 200

test_rest_api.py

import pytest
from src.rest_api import app

@pytest.fixture
def client():
    """Provide the client session used by tests."""
    with app.test_client() as client:
        yield client

@mock.patch('src.recommender.RecommendationTask.execute', side_effect=None, return_result=None)
def test_recommendation_api_endpoint(_mock_object, client):
    """Check the /recommender REST API endpoint."""
    rec_resp = client.post(api_route_for("recommender"),
                           data=json.dumps(payload), content_type='application/json')

我无法模拟src.recommender.RecommendationTask.execute功能。在直接功能测试中，模拟效果很好，但在API调用中效果不佳。

PS：通过直接函数调用，我的意思是通过立即在测试脚本中导入并执行它们来测试功能。请帮忙。

Answer 1

通过创建另一个模拟函数来解决此问题。

class FakedResponse:
    """Fake Response2."""

    status_code = 200

    @staticmethod
    def json():
        """Json Response."""
        return response

然后通过修改实际测试功能

@mock.patch('src.recommender.RecommendationTask.execute', side_effect=None, return_result=None)
def test_recommendation_api_endpoint(_mock_object, client):
    """Check the /recommender REST API endpoint."""
    r.cursor = FakedResponse()

    rec_resp = client.post(api_route_for("recommender"),
                           data=json.dumps(payload), content_type='application/json')

无法在Flask Unitest中模拟api调用内部函数

1 个答案: