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

more coverage: parse_tcp_v1_hint

Browse Source
This commit is contained in:
Brian Warner 2016-12-30 23:40:47 -05:00
parent 511a73c491
commit fe6ff466d8
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -167,6 +167,23 @@ class Hints(unittest.TestCase):
self.assertEqual(r2, r3) self.assertEqual(r2, r3)
self.assertEqual(len(set([r1, r2, r3])), 1) self.assertEqual(len(set([r1, r2, r3])), 1)
def test_parse_tcp_v1_hint(self):
c = transit.Common("")
p = c._parse_tcp_v1_hint
self.assertEqual(p({"type": "unknown"}), None)
h = p({"type": "direct-tcp-v1", "hostname": "foo", "port": 1234})
self.assertEqual(h, transit.DirectTCPV1Hint("foo", 1234))
h = p({"type": "tor-tcp-v1", "hostname": "foo", "port": 1234})
self.assertEqual(h, transit.TorTCPV1Hint("foo", 1234))
self.assertEqual(p({"type": "direct-tcp-v1"}),
None) # missing hostname
self.assertEqual(p({"type": "direct-tcp-v1", "hostname": 12}),
None) # invalid hostname
self.assertEqual(p({"type": "direct-tcp-v1", "hostname": "foo"}),
None) # missing port
self.assertEqual(p({"type": "direct-tcp-v1", "hostname": "foo",
"port": "not a number"}),
None) # invalid port
class Basic(unittest.TestCase): class Basic(unittest.TestCase):
@inlineCallbacks @inlineCallbacks