Does not work with asynchronously defined functions #8
Description
Hiya,
I'm looking to use capturemock to emulate an LLM in some testing that will be part of a CI/CD pipeline. The functions that need testing are defined asynchronously and this seems to be causing problems with capturemock. If the function that needs testing is defined in a non-asynchronous way, everything works fine. Unfortunately, when it is defined asynchronously (which is needed), capturemock doesn't seem to recognise the library it's supposed to be mocking. It doesn't create an error but it will not create a .mock file if in record mode, nor will it use one (that was created using a non-asynchronous definition) if it is in replay mode.
I have created a MWE that doesn't involve large LLM files but still demonstrates the problem.
python version:
Python 3.7.17
requirements.txt:
CaptureMock==2.3.0
certifi==2023.7.22
charset-normalizer==3.2.0
exceptiongroup==1.1.3
idna==3.4
importlib-metadata==6.7.0
iniconfig==2.0.0
packaging==23.1
pluggy==1.2.0
pytest==7.4.0
pytest-asyncio==0.21.1
requests==2.31.0
tomli==2.0.1
typing_extensions==4.7.1
urllib3==2.0.4
zipp==3.15.0
test_requests.py:
import pytest
from capturemock import RECORD, REPLAY, capturemock
# mode = REPLAY
mode = RECORD
@capturemock("requests", mode=mode)
@pytest.mark.asyncio
async def test_requests():
# def test_requests():
import requests
async def requests_get(url):
# def requests_get(url):
return requests.get(url)
resp = await requests_get("http://www.google.com/")
# resp = requests_get("http://www.google.com/")
assert isinstance(resp, requests.Response)
assert repr(resp) == "<Response [200]>"
assert resp.url == "http://www.google.com/"
assert resp.status_code == 200
Then I run pytest but it won't create a .mock file. If you want to run it in a non-aynchronous way, then you can comment out the 3 relevant lines (11,15,19) and comment in the lines directly below. This will allow you to confirm that it's only the async aspect that's causing the problem and create a .mock file which you can try use if you want to try it when defined asynchronously in replay mode. It might be that this capturemock has to be used in a different way with asynchronous functions but I couldn't see anything in the documentation about how to do this. Any help you could provide on this would be appreciated.