python - 如何在 Pytest 中模拟 httpx.AsyncClient()

Question

我需要为用于从 API 获取数据的函数编写测试用例。在那里我使用 httpx.AsyncClient() 作为上下文管理器。但我不明白如何为该功能编写测试用例。

async def make_dropbox_request(url, payload, dropbox_token):
async with httpx.AsyncClient(timeout=None, follow_redirects=True) as client:
    headers = {
        'Content-Type': 'application/json',
        'authorization': 'Bearer '+ dropbox_token
    }
    # make the api call
    response = await client.post(url, headers=headers, json=payload)
    
    if response.status_code not in [200]:
        print('Dropbox Status Code: ' + str(response.status_code))

    if response.status_code in [200, 202, 303]:
        return json.loads(response.text)

    elif response.status_code == 401:
        raise DropboxAuthenticationError()

    elif response.status_code == 429:
        sleep_time = int(response.headers['Retry-After'])
        if sleep_time < 1*60:
            await asyncio.sleep(sleep_time)
            raise DropboxMaxRateLimitError()
        raise DropboxMaxDailyRateLimitError()

    raise DropboxHTTPError()

我需要在不调用 API 的情况下编写测试用例。所以我相信在这种情况下我需要模拟 client.post() 但我不知道该怎么做。如果有人可以帮助我解决这个问题，那对我来说真的很有帮助。

This image also include my code block

Answer 1

TL;DR:使用 return_value.__aenter__.return_value 模拟异步上下文。

假设您使用的是 Pytest和 pytest-mock ，您可以使用 mocker fixture 来模拟 httpx.AsyncClient。

由于 post 函数是异步的，因此您需要使用 AsyncMock。最后，由于您使用异步上下文，因此您还需要使用 return_value.__aenter__.return_value 来正确模拟返回的上下文。请注意，对于同步上下文，只需使用 __enter__ 而不是 __aenter__。

@pytest.fixture
def mock_AsyncClient(mocker: MockerFixture) -> Mock:
      mocked_AsyncClient = mocker.patch(f"{TESTED_MODULE}.AsyncClient")

      mocked_async_client = Mock()
      response = Response(status_code=200)
      mocked_async_client.post = AsyncMock(return_value=response)
      mocked_AsyncClient.return_value.__aenter__.return_value = mocked_async_client

      return mocked_async_client