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

test_wormhole: use assertRaises as context manager

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

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

@ -514,7 +514,8 @@ class Errors(ServerBase, unittest.TestCase):
def test_derive_key_early(self):
w = wormhole.create(APPID, self.relayurl, reactor)
# definitely too early
self.assertRaises(NoKeyError, w.derive_key, "purpose", 12)
with self.assertRaises(NoKeyError):
w.derive_key("purpose", 12)
yield self.assertFailure(w.close(), LonelyError)
@inlineCallbacks
@ -522,7 +523,8 @@ class Errors(ServerBase, unittest.TestCase):
w = wormhole.create(APPID, self.relayurl, reactor)
w.set_code("123-purple-elephant")
# code can only be set once
self.assertRaises(OnlyOneCodeError, w.set_code, "123-nope")
with self.assertRaises(OnlyOneCodeError):
w.set_code("123-nope")
yield self.assertFailure(w.close(), LonelyError)
@inlineCallbacks
@ -530,7 +532,8 @@ class Errors(ServerBase, unittest.TestCase):
w = wormhole.create(APPID, self.relayurl, reactor)
w.allocate_code()
yield w.when_code()
self.assertRaises(OnlyOneCodeError, w.set_code, "123-nope")
with self.assertRaises(OnlyOneCodeError):
w.set_code("123-nope")
yield self.assertFailure(w.close(), LonelyError)
class Reconnection(ServerBase, unittest.TestCase):