2 files changed, 7 insertions, 7 deletions
diff --git a/packages/meshbay-node/tests/test_chat_is_bounded.py b/packages/meshbay-node/tests/test_chat_is_bounded.py
index 5dc1128..3332601 100644
--- a/packages/meshbay-node/tests/test_chat_is_bounded.py
+++ b/packages/meshbay-node/tests/test_chat_is_bounded.py
@@ -36,7 +36,7 @@ from cryptography.hazmat.primitives.asymmetric.ed25519 import Ed25519PrivateKey
 from meshbay_common.chatbox import NONCE_LEN, SIG_LEN
 from meshbay_common.crypto import pk_to_b64
 from meshbay_node.chat import FORMAT_SEALED_V1, ChatStore
-from meshbay_node.transport import webrtc_server as ws
+from meshbay_node.transport.webrtc import chat as ws
 from meshbay_node.transport.webrtc_server import WebRTCPeerSession
 
 # Read with a default rather than imported. Against the source these were
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):