open.source/magic-wormhole-transit-relay
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

websocket tests already use the interface

Browse Source
This commit is contained in:
meejah 2021-04-10 18:45:20 -06:00
parent a057da49cf
commit ef96af2a80
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/wormhole_transit_relay/test/test_transit_server.py
View File

@ -12,7 +12,11 @@ from autobahn.twisted.testing import (
MemoryReactorClockResolver, MemoryReactorClockResolver,
) )
from autobahn.exception import Disconnected from autobahn.exception import Disconnected
from .common import ServerBase from zope.interface import implementer
from .common import (
ServerBase,
IRelayTestClient,
)
from ..server_state import ( from ..server_state import (
MemoryUsageRecorder, MemoryUsageRecorder,
blur_size, blur_size,
@ -400,6 +404,7 @@ class TransitWebSockets(_Transit, ServerBase, unittest.TestCase):
ws_factory.transit = self._transit_server ws_factory.transit = self._transit_server
ws_protocol = ws_factory.buildProtocol(('127.0.0.1', 0)) ws_protocol = ws_factory.buildProtocol(('127.0.0.1', 0))
@implementer(IRelayTestClient)
class TransitWebSocketClientProtocol(WebSocketClientProtocol): class TransitWebSocketClientProtocol(WebSocketClientProtocol):
_received = b"" _received = b""
connected = False connected = False