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

cleanup

Browse Source
This commit is contained in:
meejah 2021-04-07 16:44:24 -06:00
parent 941e4fe18a
commit eb3bc6b5a8
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -504,14 +504,11 @@ class UsageWebSockets(Usage):
""" """
def new_protocol(self): def new_protocol(self):
ws_factory = WebSocketServerFactory("ws://localhost:4002") # FIXME: url ws_factory = WebSocketServerFactory("ws://localhost:4002")
ws_factory.protocol = WebSocketTransitConnection ws_factory.protocol = WebSocketTransitConnection
ws_factory.transit = self._transit_server ws_factory.transit = self._transit_server
ws_factory.websocket_protocols = ["binary"]
ws_protocol = ws_factory.buildProtocol(('127.0.0.1', 0)) ws_protocol = ws_factory.buildProtocol(('127.0.0.1', 0))
from autobahn.twisted.websocket import WebSocketClientFactory, WebSocketClientProtocol
class TransitWebSocketClientProtocol(WebSocketClientProtocol): class TransitWebSocketClientProtocol(WebSocketClientProtocol):
def send(self, data): def send(self, data):
self.sendMessage(data, True) self.sendMessage(data, True)