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

test_wormhole: improve tests of derive_key()

Browse Source
This commit is contained in:
Brian Warner 2017-04-18 22:50:59 -04:00
parent 654bba4d3e
commit 77bb7f5be7
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/wormhole/test/test_wormhole.py
View File

@ -143,6 +143,9 @@ class Wormholes(ServerBase, unittest.TestCase):
def test_basic(self): def test_basic(self):
w1 = wormhole.create(APPID, self.relayurl, reactor) w1 = wormhole.create(APPID, self.relayurl, reactor)
#w1.debug_set_trace("W1") #w1.debug_set_trace("W1")
with self.assertRaises(NoKeyError):
w1.derive_key("purpose", 12)
w2 = wormhole.create(APPID, self.relayurl, reactor) w2 = wormhole.create(APPID, self.relayurl, reactor)
#w2.debug_set_trace(" W2") #w2.debug_set_trace(" W2")
w1.allocate_code() w1.allocate_code()
@ -152,6 +155,9 @@ class Wormholes(ServerBase, unittest.TestCase):
yield w1.when_key() yield w1.when_key()
yield w2.when_key() yield w2.when_key()
key1 = w1.derive_key("purpose", 16)
self.assertEqual(len(key1), 16)
self.assertEqual(type(key1), type(b""))
with self.assertRaises(TypeError): with self.assertRaises(TypeError):
w1.derive_key(b"not unicode", 16) w1.derive_key(b"not unicode", 16)
with self.assertRaises(TypeError): with self.assertRaises(TypeError):