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

test to ensure we make a websocket service when passing --websocket

Browse Source
This commit is contained in:
meejah 2021-04-14 17:07:50 -06:00
parent 425f040168
commit 30688f638c
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/wormhole_transit_relay/test/test_service.py
View File

@ -5,6 +5,7 @@ try:
except ImportError: except ImportError:
import mock import mock
from twisted.application.service import MultiService from twisted.application.service import MultiService
from autobahn.twisted.websocket import WebSocketServerFactory
from .. import server_tap from .. import server_tap
class Service(unittest.TestCase): class Service(unittest.TestCase):
@ -40,3 +41,13 @@ class Service(unittest.TestCase):
[mock.call(blur_usage=None, [mock.call(blur_usage=None,
log_file=fd, usage_db=None)]) log_file=fd, usage_db=None)])
def test_websocket(self):
o = server_tap.Options()
o.parseOptions(["--websocket=tcp:4004"])
services = server_tap.makeService(o)
self.assertTrue(
any(
isinstance(s.factory, WebSocketServerFactory)
for s in services.services
)
)