helius-python 0.5.1__tar.gz → 0.5.2__tar.gz

{helius_python-0.5.1 → helius_python-0.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: helius-python
-Version: 0.5.1
+Version: 0.5.2
 Summary: Typed Python client for the Helius API
 Project-URL: Homepage, https://github.com/markosnarinian/helius-python
 Project-URL: Issues, https://github.com/markosnarinian/helius-python/issues

{helius_python-0.5.1 → helius_python-0.5.2}/pyproject.toml

@@ -10,7 +10,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "helius-python"
-version = "0.5.1"
+version = "0.5.2"
 authors = [
   { name="Markos Narinian", email="manarinian@gmail.com" },
 ]

{helius_python-0.5.1 → helius_python-0.5.2}/src/helius/laserstream/websockets.py

@@ -132,6 +132,10 @@ class WebSocketClient:
         base_url="wss://mainnet.helius-rpc.com",
         api_key: str | None = None,
         proxy: str | None = None,
+        open_timeout: float | None = 10,
+        ping_interval: float | None = 20,
+        ping_timeout: float | None = 20,
+        close_timeout: float | None = 10,
     ):
 
         base_url = base_url
@@ -144,7 +148,14 @@ class WebSocketClient:
         if not api_key:
             raise ValueError("No API key provided.")
         uri = httpx.URL(base_url).copy_with(path="/", params={"api-key": api_key})
-        self._websocket = connect(str(uri), proxy=proxy)
+        self._websocket = connect(
+            str(uri),
+            proxy=proxy,
+            open_timeout=open_timeout,
+            ping_interval=ping_interval,
+            ping_timeout=ping_timeout,
+            close_timeout=close_timeout,
+        )
 
     def close(self):
         self._websocket.close()

{helius_python-0.5.1 → helius_python-0.5.2}/tests/unit/lasterstream/test_websockets.py

@@ -64,9 +64,20 @@ def fake_ws(monkeypatch):
     """Patch `connect` so `WebSocketClient.__init__` returns a FakeWebSocket."""
     fake = FakeWebSocket()
 
-    def fake_connect(uri, proxy=None):
+    def fake_connect(
+        uri,
+        proxy=None,
+        open_timeout=None,
+        ping_interval=None,
+        ping_timeout=None,
+        close_timeout=None,
+    ):
         fake.uri = uri
         fake.proxy = proxy
+        fake.open_timeout = open_timeout
+        fake.ping_interval = ping_interval
+        fake.ping_timeout = ping_timeout
+        fake.close_timeout = close_timeout
         return fake
 
     monkeypatch.setattr(ws_module, "connect", fake_connect)
@@ -283,13 +294,29 @@ def test_init_picks_up_api_key_from_env(monkeypatch):
     monkeypatch.setenv("HELIUS_API_KEY", "from-env")
     monkeypatch.setattr(ws_module, "dotenv_values", lambda: {})
 
-    def fake_connect(uri, proxy=None):
+    def fake_connect(
+        uri,
+        proxy=None,
+        open_timeout=None,
+        ping_interval=None,
+        ping_timeout=None,
+        close_timeout=None,
+    ):
         fake.uri = uri
+        fake.proxy = proxy
+        fake.open_timeout = open_timeout
+        fake.ping_interval = ping_interval
+        fake.ping_timeout = ping_timeout
+        fake.close_timeout = close_timeout
         return fake
 
     monkeypatch.setattr(ws_module, "connect", fake_connect)
     client = WebSocketClient()
     assert "api-key=from-env" in fake.uri
+    assert fake.open_timeout == 10
+    assert fake.ping_interval == 20
+    assert fake.ping_timeout == 20
+    assert fake.close_timeout == 10
     assert client._websocket is fake
 
 
@@ -297,29 +324,67 @@ def test_init_builds_uri_with_api_key_and_passes_proxy(monkeypatch):
     fake = FakeWebSocket()
     captured = {}
 
-    def fake_connect(uri, proxy=None):
+    def fake_connect(
+        uri,
+        proxy=None,
+        open_timeout=None,
+        ping_interval=None,
+        ping_timeout=None,
+        close_timeout=None,
+    ):
         captured["uri"] = uri
         captured["proxy"] = proxy
+        captured["open_timeout"] = open_timeout
+        captured["ping_interval"] = ping_interval
+        captured["ping_timeout"] = ping_timeout
+        captured["close_timeout"] = close_timeout
         return fake
 
     monkeypatch.setattr(ws_module, "connect", fake_connect)
-    WebSocketClient(api_key="test", proxy="http://proxy:8080")
+    WebSocketClient(
+        api_key="test",
+        proxy="http://proxy:8080",
+        open_timeout=1,
+        ping_interval=2,
+        ping_timeout=3,
+        close_timeout=4,
+    )
     assert "api-key=test" in captured["uri"]
     assert captured["uri"].startswith("wss://mainnet.helius-rpc.com")
     assert captured["proxy"] == "http://proxy:8080"
+    assert captured["open_timeout"] == 1
+    assert captured["ping_interval"] == 2
+    assert captured["ping_timeout"] == 3
+    assert captured["close_timeout"] == 4
 
 
 def test_init_uses_custom_base_url(monkeypatch):
     fake = FakeWebSocket()
     captured = {}
 
-    def fake_connect(uri, proxy=None):
+    def fake_connect(
+        uri,
+        proxy=None,
+        open_timeout=None,
+        ping_interval=None,
+        ping_timeout=None,
+        close_timeout=None,
+    ):
         captured["uri"] = uri
+        captured["proxy"] = proxy
+        captured["open_timeout"] = open_timeout
+        captured["ping_interval"] = ping_interval
+        captured["ping_timeout"] = ping_timeout
+        captured["close_timeout"] = close_timeout
         return fake
 
     monkeypatch.setattr(ws_module, "connect", fake_connect)
     WebSocketClient(api_key="test", base_url="wss://devnet.helius-rpc.com")
     assert captured["uri"].startswith("wss://devnet.helius-rpc.com")
+    assert captured["open_timeout"] == 10
+    assert captured["ping_interval"] == 20
+    assert captured["ping_timeout"] == 20
+    assert captured["close_timeout"] == 10
 
 
 def test_close_delegates_to_websocket(client, fake_ws):