aboutsummaryrefslogtreecommitdiffstats
path: root/packages/meshbay-node/tests
diff options
context:
space:
mode:
Diffstat (limited to 'packages/meshbay-node/tests')
-rw-r--r--packages/meshbay-node/tests/test_security_regressions.py6
1 files changed, 2 insertions, 4 deletions
diff --git a/packages/meshbay-node/tests/test_security_regressions.py b/packages/meshbay-node/tests/test_security_regressions.py
index 321d22d..b3f373d 100644
--- a/packages/meshbay-node/tests/test_security_regressions.py
+++ b/packages/meshbay-node/tests/test_security_regressions.py
@@ -720,11 +720,9 @@ def test_pre_handshake_message_budget_is_small():
H6: the frame limit was a flat 64 MB applied before authentication, so an
unauthenticated peer could announce a huge frame and dribble bytes into it.
"""
+ from meshbay_node.transport.webrtc.channel import _DataChannelBuffer
+ from meshbay_node.transport.webrtc.core import PRE_HANDSHAKE_MAX_MSG
from meshbay_node.transport.webrtc.limits import MAX_MSG
- from meshbay_node.transport.webrtc_server import (
- PRE_HANDSHAKE_MAX_MSG,
- _DataChannelBuffer,
- )
assert PRE_HANDSHAKE_MAX_MSG <= 1024 * 1024
assert PRE_HANDSHAKE_MAX_MSG < MAX_MSG