test_transit: fix test to tolerate proper virtualized reactor

2020-02-06 22:48:25 -08:00 · 2020-02-06 22:48:25 -08:00 · 0c43499850
commit 0c43499850
parent 4a2f3baf28
1 changed files with 2 additions and 2 deletions
--- 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