tests for new establish_key wormhole hook
defered should behave similarly to the `verify` hook
This commit is contained in:
		
							parent
							
								
									cd40fc7a03
								
							
						
					
					
						commit
						33fa6f6ede
					
				| 
						 | 
				
			
			@ -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
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue
	
	Block a user