diff --git a/src/wormhole/test/test_transit.py b/src/wormhole/test/test_transit.py index 8676ccc..0af8d3b 100644 --- a/src/wormhole/test/test_transit.py +++ b/src/wormhole/test/test_transit.py @@ -1312,8 +1312,8 @@ class Transit(unittest.TestCase): @inlineCallbacks def test_success_direct(self): - clock = task.Clock() - s = transit.TransitSender("", reactor=clock) + reactor = mock.Mock() + s = transit.TransitSender("", reactor=reactor) s.set_transit_key(b"key") hints = yield s.get_connection_hints() # start the listener del hints