tests for new establish_key wormhole hook

defered should behave similarly to the `verify` hook
This commit is contained in:
laharah 2016-06-04 23:52:32 -07:00 committed by Brian Warner
parent cd40fc7a03
commit 33fa6f6ede

View File

@ -522,6 +522,32 @@ class Basic(unittest.TestCase):
self.assertEqual(len(pieces), 3) # nameplate plus two words
self.assert_(re.search(r'^\d+-\w+-\w+$', code), code)
def _test_establish_key_hook(self, established):
timing = DebugTiming()
w = wormhole._Wormhole(APPID, "relay_url", reactor, None, timing)
if established is True:
w._key = b"key"
elif established is False:
w._key = None
else:
w._key = b"key"
w._error = WelcomeError()
d = w.establish_key()
if w._key is not None and established is True:
self.successResultOf(d)
elif established is False:
self.assertNot(d.called)
else:
self.failureResultOf(d)
def test_establish_key_hook(self):
for established in (True, False, "error"):
self._test_establish_key_hook(established)
# make sure verify() can be called both before and after the verifier is
# computed