diff options
Diffstat (limited to 'packages/meshbay-node/tests/test_link_preview_request.py')
| -rw-r--r-- | packages/meshbay-node/tests/test_link_preview_request.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/packages/meshbay-node/tests/test_link_preview_request.py b/packages/meshbay-node/tests/test_link_preview_request.py index fe7dec6..0e1bd06 100644 --- a/packages/meshbay-node/tests/test_link_preview_request.py +++ b/packages/meshbay-node/tests/test_link_preview_request.py @@ -13,7 +13,7 @@ import pytest from meshbay_common.protocol import MNP from meshbay_node import linkpreview from meshbay_node.media_cache import MediaCache -from meshbay_node.transport import webrtc_server +from meshbay_node.transport.webrtc import chat from meshbay_node.transport.webrtc_server import WebRTCPeerSession pytestmark = pytest.mark.asyncio @@ -29,9 +29,9 @@ async def media_cache(tmp_path): @pytest.fixture(autouse=True) def _clear_cache(): - webrtc_server._link_preview_cache.clear() + chat._link_preview_cache.clear() yield - webrtc_server._link_preview_cache.clear() + chat._link_preview_cache.clear() def _session(media_cache, ctx=None): @@ -83,7 +83,7 @@ async def test_unfurlable_failure_is_ok_false(media_cache, monkeypatch): async def test_rate_limit_per_connection(media_cache, monkeypatch): """A member firing many previews is bounded; over the ceiling the reply is a plain `ok: false` (bare link) and no outbound fetch is made.""" - monkeypatch.setattr(webrtc_server, "_LINK_PREVIEW_RATE_PER_CONN", 3) + monkeypatch.setattr(chat, "_LINK_PREVIEW_RATE_PER_CONN", 3) calls = {"n": 0} async def counting_preview(url, **k): @@ -105,8 +105,8 @@ async def test_rate_limit_per_connection(media_cache, monkeypatch): async def test_rate_limit_is_node_wide(media_cache, monkeypatch): """Two connections share the node-wide ceiling.""" - monkeypatch.setattr(webrtc_server, "_LINK_PREVIEW_RATE_PER_CONN", 100) - monkeypatch.setattr(webrtc_server, "_LINK_PREVIEW_RATE_NODE", 2) + monkeypatch.setattr(chat, "_LINK_PREVIEW_RATE_PER_CONN", 100) + monkeypatch.setattr(chat, "_LINK_PREVIEW_RATE_NODE", 2) calls = {"n": 0} async def counting_preview(url, **k): |