Pytest 请求模拟最大重试次数

如何解决Pytest 请求模拟最大重试次数

我有一个项目，需要测试一个特定的功能。 该函数包含一些请求。

class ApiException(Exception):
    def __init__(self,status_code,msg):
        """
        Creates a new api error which contains the error message and the status code
        """
        self.status_code = status_code
        self.message = msg
        super.__init__(self.msg)

@log
def generate_image(url: str,data: str,cred: dict):
    """
    Sends html code to given url and returns a rendered image
    """
    img_req = requests.post(url=url,data=data,auth(cred["username"],cred["password"]))
     img = requests.get(img_req.json()["url"] + ".png",stream=True)

    if img.status_code == 200:
        return BytesIO(img.content)
    else:
        raise ApiException(img.status_code,"Image could not be received!")

我的测试

class HCTIResponse(object):
    def __init__(self):
        self.status_code = 200
        self.content = {
            "url": "https://fakeapi.com/fake"
        }

    def json(self):
        return self.content


class TestApi(TestCase):
    def test_get_covid_data(self) -> None:
        with patch("requests.get") as mock_request:
            url = "https://fakeapi.com/fake"

            mock_request.return_value = SomeResponse()

            get_some_data(url,AGS)

    def test_generate_image(self) -> None:
        with patch("requests.get") as mock_request:
            url = "https://fakeapi.com/fake"

            mock_request.return_value = HCTIResponse()

            generate_image(url,"",{"username": "","password": ""})

当我运行我的测试时，我得到一个 最大重试次数，网址为：/fake

在相同的 fakeapi url 上运行的所有其他测试都可以完美运行。

我现在已经转了一个小时了。

感谢您的帮助！