From 0378be374ca58a70d1d9bc1d99d2e0a9e933b4ad Mon Sep 17 00:00:00 2001 From: Juliya Smith Date: Thu, 5 Dec 2024 11:41:14 -0600 Subject: [PATCH] test: ws --- tests/test_integration.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test_integration.py b/tests/test_integration.py index 2d55ceb..8c0fd49 100644 --- a/tests/test_integration.py +++ b/tests/test_integration.py @@ -5,7 +5,7 @@ from ape.utils import ZERO_ADDRESS from ape_alchemy._utils import NETWORKS -from ape_alchemy.provider import NETWORKS_SUPPORTING_WEBSOCKETS, Alchemy +from ape_alchemy.provider import Alchemy @pytest.fixture(params=[(name, net) for name, values in NETWORKS.items() for net in values]) @@ -27,7 +27,8 @@ def test_http(provider): def test_ws(provider): - if provider.network.ecosystem.name not in NETWORKS_SUPPORTING_WEBSOCKETS: + ws_uri = provider.ws_uri + if ws_uri is None: # Test will fail. Network does not support ws clients. return