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

test_hints: finish coverage of hints.py

Browse Source
This commit is contained in:
Brian Warner 2018-12-21 23:37:07 -05:00
parent 1bb5634d0e
commit d64c94a1dc
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/wormhole/test/test_hints.py
View File

@ -21,11 +21,12 @@ class Hints(unittest.TestCase):
# tor=None # tor=None
self.assertEqual(efho(TorTCPV1Hint("host", "port", 0)), None) self.assertEqual(efho(TorTCPV1Hint("host", "port", 0)), None)
self.assertEqual(efho(UnknownHint("foo")), None)
tor = mock.Mock() tor = mock.Mock()
def tor_ep(hostname, port): def tor_ep(hostname, port):
if hostname == "non-public": if hostname == "non-public":
return None raise ValueError
return ("tor_ep", hostname, port) return ("tor_ep", hostname, port)
tor.stream_via = mock.Mock(side_effect=tor_ep) tor.stream_via = mock.Mock(side_effect=tor_ep)
@ -35,7 +36,7 @@ class Hints(unittest.TestCase):
("tor_ep", "host2.onion", 1234)) ("tor_ep", "host2.onion", 1234))
self.assertEqual( efho(DirectTCPV1Hint("non-public", 1234, 0.0), tor), None) self.assertEqual( efho(DirectTCPV1Hint("non-public", 1234, 0.0), tor), None)
self.assertEqual(efho(UnknownHint("foo")), None) self.assertEqual(efho(UnknownHint("foo"), tor), None)
def test_comparable(self): def test_comparable(self):
h1 = DirectTCPV1Hint("hostname", "port1", 0.0) h1 = DirectTCPV1Hint("hostname", "port1", 0.0)
@ -136,6 +137,12 @@ class Hints(unittest.TestCase):
d = describe_hint_obj d = describe_hint_obj
self.assertEqual(d(DirectTCPV1Hint("host", 1234, 0.0), False, False), self.assertEqual(d(DirectTCPV1Hint("host", 1234, 0.0), False, False),
"->tcp:host:1234") "->tcp:host:1234")
self.assertEqual(d(DirectTCPV1Hint("host", 1234, 0.0), True, False),
"->relay:tcp:host:1234")
self.assertEqual(d(DirectTCPV1Hint("host", 1234, 0.0), False, True),
"tor->tcp:host:1234")
self.assertEqual(d(DirectTCPV1Hint("host", 1234, 0.0), True, True),
"tor->relay:tcp:host:1234")
self.assertEqual(d(TorTCPV1Hint("host", 1234, 0.0), False, False), self.assertEqual(d(TorTCPV1Hint("host", 1234, 0.0), False, False),
"->tor:host:1234") "->tor:host:1234")
self.assertEqual(d(UnknownHint("stuff"), False, False), self.assertEqual(d(UnknownHint("stuff"), False, False),