We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f4e6cdf commit de81ef8Copy full SHA for de81ef8
1 file changed
tests/test_client_lifecycle.py
@@ -25,8 +25,20 @@ def tracked_close() -> None:
25
26
def test_async_client_supports_context_manager():
27
async def run() -> None:
28
- async with AsyncHyperbrowser(api_key="test-key") as client:
29
- assert isinstance(client, AsyncHyperbrowser)
+ client = AsyncHyperbrowser(api_key="test-key")
+ close_calls = {"count": 0}
30
+ original_close = client.transport.close
31
+
32
+ async def tracked_close() -> None:
33
+ close_calls["count"] += 1
34
+ await original_close()
35
36
+ client.transport.close = tracked_close
37
38
+ async with client as entered:
39
+ assert entered is client
40
41
+ assert close_calls["count"] == 1
42
43
asyncio.run(run())
44
0 commit comments