test: fix websocket test

ApeWorX · Nov 1, 2023 · 8620fdf · 8620fdf
1 parent 63b1a8f
commit 8620fdf
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/test_provider.py b/tests/test_provider.py
@@ -2,7 +2,7 @@
 import websocket  # type: ignore
 from ape.utils import ZERO_ADDRESS
 
-from ape_infura.provider import Infura
+from ape_infura.provider import Infura, _WEBSOCKET_CAPABLE_ECOSYSTEMS
 
 
 def test_infura_http(provider):
@@ -17,6 +17,11 @@ def test_infura_http(provider):
 
 
 def test_infura_ws(provider):
+    ecosystem = provider.network.ecosystem.name
+    if ecosystem not in _WEBSOCKET_CAPABLE_ECOSYSTEMS:
+        assert provider.ws_uri == None
+        return
+
     assert provider.ws_uri.startswith("wss")
 
     try: