Added change to make tests backwards compatible (#14)

* Added change to make tests backwards compatible with new httpx_mock version

* Update test_simple_agent.py

* Update test_simple_chat_agent.py

* Update test_tools.py

* Added changelog

* Reformatted with proper ruff version

CHANGELOG.md

@@ -11,6 +11,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Fixed
 - Fixed a bug that prevented AsyncSqlite checkpoint to access the DB in streamed endpoints.
+- Fixed a bug that caused some unit tests to fail due to a change in how httpx_mock works in version 0.32
 
 ## [0.1.0] - 19.09.2024
 

tests/agents/test_simple_chat_agent.py

@@ -9,6 +9,7 @@
 from neuroagent.agents import AgentOutput, AgentStep, SimpleChatAgent
 
 
+@pytest.mark.httpx_mock(can_send_already_matched_responses=True)
 @pytest.mark.asyncio
 async def test_arun(fake_llm_with_tools, httpx_mock):
     llm, tools, fake_responses = await anext(fake_llm_with_tools)
@@ -64,6 +65,7 @@ async def test_arun(fake_llm_with_tools, httpx_mock):
         assert len(messages_list) == 10
 
 
+@pytest.mark.httpx_mock(can_send_already_matched_responses=True)
 @pytest.mark.asyncio
 async def test_astream(fake_llm_with_tools, httpx_mock):
     llm, tools, fake_responses = await anext(fake_llm_with_tools)

tests/test_utils.py

@@ -313,6 +313,7 @@ async def test_get_kg_data_errors(httpx_mock):
     )
 
 
+@pytest.mark.httpx_mock(can_send_already_matched_responses=True)
 @pytest.mark.asyncio
 async def test_get_kg_data(httpx_mock):
     url = "http://fake_url"

tests/tools/test_electrophys_tool.py

@@ -15,6 +15,7 @@
 
 
 class TestElectrophysTool:
+    @pytest.mark.httpx_mock(can_send_already_matched_responses=True)
     @pytest.mark.asyncio
     async def test_arun(self, httpx_mock):
         url = "http://fake_url"

tests/tools/test_traces_tool.py

@@ -11,6 +11,7 @@
 
 
 class TestTracesTool:
+    @pytest.mark.httpx_mock(can_send_already_matched_responses=True)
     @pytest.mark.asyncio
     async def test_arun(self, httpx_mock, brain_region_json_path):
         url = "http://fake_url"