fix

Author: Miskler

pyaterochka_api/manager.py

@@ -115,7 +115,7 @@ async def _warmup(self) -> None:
                 await self.page.wait_for_load_state("networkidle", timeout=self.timeout_ms)
                 ok = True
             except PWTimeoutError:
-                await self.page.reload()
+                await self.page.reload(wait_until="commit")
         if not ok:
             raise RuntimeError(self.page.content)
 

tests/api_tests.py

@@ -2,6 +2,8 @@
 from PIL import Image
 from pytest_jsonschema_snapshot import SchemaShot
 
+import aiohttp
+from human_requests.abstraction import Proxy
 from pyaterochka_api import PyaterochkaAPI
 
 
@@ -158,3 +160,15 @@ async def test_download_image(api: PyaterochkaAPI, first_category: dict):
     with Image.open(resp) as img:
         fmt = img.format.lower()
     assert fmt in ("png", "jpeg", "webp")
+
+async def test_proxy_ip(api: PyaterochkaAPI):
+    if not api.proxy:
+        pytest.skip("Proxy not configured")
+
+    prx = Proxy(api.proxy)
+
+    async with aiohttp.ClientSession() as session:
+        async with session.get("http://httpbin.org/ip", proxy=prx.as_str()) as resp:
+            ip = resp.json()["origin"]
+
+    assert ip == prx._server.split(":")[0]