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

improve coverage: Transit._endpoint_from_hint_obj

Browse Source
This commit is contained in:
Brian Warner 2016-12-30 23:34:20 -05:00
parent 7f54d810eb
commit 511a73c491
1 changed files with 21 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/wormhole/test/test_transit.py
View File

@ -3,6 +3,7 @@ import io
import gc import gc
import mock import mock
from binascii import hexlify, unhexlify from binascii import hexlify, unhexlify
from collections import namedtuple
from twisted.trial import unittest from twisted.trial import unittest
from twisted.internet import defer, task, endpoints, protocol, address, error from twisted.internet import defer, task, endpoints, protocol, address, error
from twisted.internet.defer import gatherResults, inlineCallbacks from twisted.internet.defer import gatherResults, inlineCallbacks
@ -134,10 +135,26 @@ class Misc(unittest.TestCase):
class Hints(unittest.TestCase): class Hints(unittest.TestCase):
def test_endpoint_from_hint_obj(self): def test_endpoint_from_hint_obj(self):
c = transit.Common("") c = transit.Common("")
ep = c._endpoint_from_hint_obj(transit.DirectTCPV1Hint("localhost", 1234)) efho = c._endpoint_from_hint_obj
self.assertIsInstance(ep, endpoints.HostnameEndpoint) self.assertIsInstance(efho(transit.DirectTCPV1Hint("host", 1234)),
ep = c._endpoint_from_hint_obj("unknown:stuff:yowza:pivlor") endpoints.HostnameEndpoint)
self.assertEqual(ep, None) self.assertEqual(efho("unknown:stuff:yowza:pivlor"), None)
# c._tor_manager is currently None
self.assertEqual(efho(transit.TorTCPV1Hint("host", "port")), None)
c._tor_manager = mock.Mock()
def tor_ep(hostname, port):
if hostname == "non-public":
return None
return ("tor_ep", hostname, port)
c._tor_manager.get_endpoint_for = mock.Mock(side_effect=tor_ep)
self.assertEqual(efho(transit.DirectTCPV1Hint("host", 1234)),
("tor_ep", "host", 1234))
self.assertEqual(efho(transit.TorTCPV1Hint("host2.onion", 1234)),
("tor_ep", "host2.onion", 1234))
self.assertEqual(efho(transit.DirectTCPV1Hint("non-public", 1234)),
None)
UnknownHint = namedtuple("UnknownHint", ["stuff"])
self.assertEqual(efho(UnknownHint("foo")), None)
def test_comparable(self): def test_comparable(self):
h1 = transit.DirectTCPV1Hint("hostname", "port1") h1 = transit.DirectTCPV1Hint("hostname", "port1")