def test_future(self):
signal = AdHocSignal()
with contextlib.ExitStack() as stack:
Future = stack.enter_context(
unittest.mock.patch("asyncio.Future")
)
connect = stack.enter_context(
unittest.mock.patch.object(
signal,
"connect"
)
)
fut = signal.future()
self.assertSequenceEqual(
Future.mock_calls,
[
unittest.mock.call.Future()
]
)
self.assertSequenceEqual(
connect.mock_calls,
[
unittest.mock.call(
Future(),
signal.AUTO_FUTURE,
),
]
)
self.assertEqual(fut, Future())
def test_future(self):
signal = AdHocSignal()
with contextlib.ExitStack() as stack:
Future = stack.enter_context(unittest.mock.patch("asyncio.Future"))
connect = stack.enter_context(
unittest.mock.patch.object(signal, "connect"))
fut = signal.future()
self.assertSequenceEqual(Future.mock_calls,
[unittest.mock.call.Future()])
self.assertSequenceEqual(connect.mock_calls, [
unittest.mock.call(
Future(),
signal.WEAK,
),
])
self.assertEqual(fut, Future())
