def test_enter_after_cancel(self): # There are no writes to chan2, so handshake should never complete chan1, chan2 = make_channel_pair() conn = Connection(chan1) conn.cancel() with self.assertRaises(ValueError): conn.__enter__()
def test_enter_when_connect_fails(self): conn = Connection(UnconnectableChannel()) with self.assertRaises(ConnectError): conn.__enter__() with self.assertRaisesRegex(ValueError, 'Connection is closed'): conn.__enter__()