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

test_blocking: avoid using real hints

Browse Source
This commit is contained in:
Brian Warner 2016-02-27 17:29:04 -08:00
parent 8067acbf82
commit ca06e95bbd
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/wormhole/test/test_blocking.py
View File

@ -449,6 +449,11 @@ class EventSourceClient(unittest.TestCase):
]) ])
class Transit(_DoBothMixin, ServerBase, unittest.TestCase): class Transit(_DoBothMixin, ServerBase, unittest.TestCase):
def test_hints(self):
r = TransitReceiver(self.transit)
hints = r.get_direct_hints()
self.assertTrue(len(hints), hints)
@inlineCallbacks @inlineCallbacks
def test_direct_to_receiver(self): def test_direct_to_receiver(self):
s = TransitSender(self.transit) s = TransitSender(self.transit)
@ -457,7 +462,9 @@ class Transit(_DoBothMixin, ServerBase, unittest.TestCase):
# force the connection to be sender->receiver # force the connection to be sender->receiver
s.set_transit_key(key) s.set_transit_key(key)
s.add_their_direct_hints(r.get_direct_hints()) # only use 127.0.0.1
hint = u"tcp:127.0.0.1:%d" % r.listener.server_address[1]
s.add_their_direct_hints([hint])
s.add_their_relay_hints([]) s.add_their_relay_hints([])
r.set_transit_key(key) r.set_transit_key(key)
r.add_their_direct_hints([]) r.add_their_direct_hints([])
@ -483,7 +490,8 @@ class Transit(_DoBothMixin, ServerBase, unittest.TestCase):
s.add_their_direct_hints([]) s.add_their_direct_hints([])
s.add_their_relay_hints([]) s.add_their_relay_hints([])
r.set_transit_key(key) r.set_transit_key(key)
r.add_their_direct_hints(s.get_direct_hints()) hint = u"tcp:127.0.0.1:%d" % s.listener.server_address[1]
r.add_their_direct_hints([hint])
r.add_their_relay_hints([]) r.add_their_relay_hints([])
(sp, rp) = yield self.doBoth([s.connect], [r.connect]) (sp, rp) = yield self.doBoth([s.connect], [r.connect])