def test_remote_close_during_call(self):
chan, rchan = make_channel_pair()
conn = Connection(chan)
Connection(rchan).close()
with self.assertRaisesRegex(DisconnectError,
'Connection was remotely closed'):
conn.call(nop)
conn.close()
def test_communication_error_during_call(self):
chan = ConnectionChannel()
conn = Connection(chan)
try:
with self.assertRaisesRegex(DisconnectError,
'Connection was aborted due to .*'):
conn.call(Box(chan.cancel)
) # simulate communication error with channel.cancel
with self.assertRaisesRegex(DisconnectError,
'Connection was aborted due to .*'):
conn.call(nop)
finally:
try:
conn.close()
except Exception:
pass