25 lines
891 B
Python
25 lines
891 B
Python
#from __future__ import unicode_literals
|
|
from twisted.internet import reactor, endpoints
|
|
from twisted.internet.defer import inlineCallbacks
|
|
from ..transit_server import Transit
|
|
|
|
class ServerBase:
|
|
@inlineCallbacks
|
|
def setUp(self):
|
|
self._lp = None
|
|
yield self._setup_relay()
|
|
|
|
@inlineCallbacks
|
|
def _setup_relay(self, blur_usage=None, log_file=None, usage_db=None):
|
|
ep = endpoints.TCP4ServerEndpoint(reactor, 0, interface="127.0.0.1")
|
|
self._transit_server = Transit(blur_usage=blur_usage,
|
|
log_file=log_file, usage_db=usage_db)
|
|
self._lp = yield ep.listen(self._transit_server)
|
|
addr = self._lp.getHost()
|
|
# ws://127.0.0.1:%d/wormhole-relay/ws
|
|
self.transit = u"tcp:127.0.0.1:%d" % addr.port
|
|
|
|
def tearDown(self):
|
|
if self._lp:
|
|
return self._lp.stopListening()
|