diff --git a/src/wormhole/test/dilate/test_manager.py b/src/wormhole/test/dilate/test_manager.py index 6acf25b..690e1cf 100644 --- a/src/wormhole/test/dilate/test_manager.py +++ b/src/wormhole/test/dilate/test_manager.py @@ -20,7 +20,8 @@ def make_dilator(): eq = EventualQueue(clock) term = mock.Mock(side_effect=lambda: True) # one write per Eventual tick - def term_factory(): return term + def term_factory(): + return term coop = Cooperator(terminationPredicateFactory=term_factory, scheduler=eq.eventually) send = mock.Mock() diff --git a/src/wormhole/test/dilate/test_outbound.py b/src/wormhole/test/dilate/test_outbound.py index db38502..6ba5264 100644 --- a/src/wormhole/test/dilate/test_outbound.py +++ b/src/wormhole/test/dilate/test_outbound.py @@ -24,7 +24,8 @@ def make_outbound(): eq = EventualQueue(clock) term = mock.Mock(side_effect=lambda: True) # one write per Eventual tick - def term_factory(): return term + def term_factory(): + return term coop = Cooperator(terminationPredicateFactory=term_factory, scheduler=eq.eventually) o = Outbound(m, coop) @@ -567,7 +568,8 @@ def make_pushpull(pauses): eq = EventualQueue(clock) term = mock.Mock(side_effect=lambda: True) # one write per Eventual tick - def term_factory(): return term + def term_factory(): + return term coop = Cooperator(terminationPredicateFactory=term_factory, scheduler=eq.eventually) pp = PullToPush(p, unregister, coop) diff --git a/src/wormhole/test/test_machines.py b/src/wormhole/test/test_machines.py index 35854ae..fe956c1 100644 --- a/src/wormhole/test/test_machines.py +++ b/src/wormhole/test/test_machines.py @@ -12,8 +12,8 @@ import mock from .. import (__version__, _allocator, _boss, _code, _input, _key, _lister, _mailbox, _nameplate, _order, _receive, _rendezvous, _send, _terminator, errors, timing) -from .._interfaces import (IAllocator, IBoss, ICode, IDilator, IInput, IKey, ILister, - IMailbox, INameplate, IOrder, IReceive, +from .._interfaces import (IAllocator, IBoss, ICode, IDilator, IInput, IKey, + ILister, IMailbox, INameplate, IOrder, IReceive, IRendezvousConnector, ISend, ITerminator, IWordlist) from .._key import derive_key, derive_phase_key, encrypt_data from ..journal import ImmediateJournal