Ложный ответ с 500 кодами состояния в колбе и pytest - PullRequest
Новая
       107

Ложный ответ с 500 кодами состояния в колбе и pytest

1 голос
/ 22 октября 2019

* РЕДАКТИРОВАНИЕ *

Я хотел бы проверить, что произойдет, если внешний API вернет код состояния 500. 

main.py

@app.route("/<a>/<b>", methods=["GET"])
def repo_info(a: str, b: str) -> Union[Response, str]:
    info = some_func(a, b)
    result = create_result_dict(some_func)
    return Response(
        response=json.dumps(result, ensure_ascii=False),
        status=200,
        mimetype="application/json",

@app.errorhandler(500)
def server_error(e):
    logging.exception(f"An error occurred during a request: {e}.")
    return Response(
        response="An internal error occurred. See logs for full stacktrace.",
        status=500,
    )

my_module.py

def some_func(a: str, b: str) -> Dict[str, str]:
    return json.loads(
        (requests.get(f"https://api.github.com/repos/{a}/{b}")).text
    )

Я пыталсяэтот код, но чувствую себя как курица без головы: 

from flask import Response
import pytest
import requests

from unittest.mock import patch
from requests.exceptions import HTTPError

@patch.object(my_module, "some_func")
def test_some_func(mocked):
    mocked.return_value = HTTPError()
    result = my_module.some_func()
    with pytest.raises(HTTPError):
        result == mocked

Также HTTPError не принимает аргументов, как я могу передать информацию, что я хотел бы получить 500 кодов статуса?

1 Ответ

0 голосов
/ 23 октября 2019

Первая проблема заключается в том, что для получения HTTPError исключения вам нужно указать requests, чтобы поднять его с помощью raise_for_status(): 

# my_module.py

import json
import requests
from typing import Dict

def some_func(a: str, b: str) -> Dict[str, str]:
    result = requests.get(f"https://api.github.com/repos/{a}/{b}")
    result.raise_for_status()

    return json.loads(result.text)

Вторая проблема заключается в том, что для насмешки вы хотите толькоимитировать запросы (чтобы установить условия, а также избежать реальных вызовов API), потому что в противном случае вы действительно хотите позвонить some_func(). В конце концов, это идея проверить это. Вы можете исправлять объекты так, как вы пытались, но я бы посоветовал вам установить запросы-макет : 

# test.py

import pytest
import requests_mock
from requests.exceptions import HTTPError
from my_module import some_func

def test_some_func():
    with requests_mock.mock() as m:
        m.get(requests_mock.ANY, status_code=500, text='some error')

        with pytest.raises(HTTPError):
            # You cannot make any assertion on the result
            # because some_func raises an exception and
            # wouldn't return anything
            some_func(a="a", b="b")
...