Mirror
/
cpython
mirror of https://github.com/python/cpython
4
1
Fork 0
Code Issues Releases Wiki Activity

gh-98703: Add tests for closing `_ProactorSocketTransport` with proactor event loop (GH-98730)

This commit is contained in:
Kumar Aditya 2022-10-27 08:37:31 +05:30 committed by GitHub
parent 8a755423eb
commit 96ae80f1d0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
Lib/test/test_asyncio/test_proactor_events.py
View File

@ -297,6 +297,27 @@ class ProactorSocketTransportTests(test_utils.TestCase):
# and waiters will never be notified leading to hang.
self.assertTrue(self.protocol.connection_lost.called)
def test_force_close_protocol_connection_lost_once(self):
tr = self.socket_transport()
self.assertFalse(self.protocol.connection_lost.called)
tr._closing = True
# Calling _force_close twice should not call
# protocol.connection_lost twice
tr._force_close(None)
tr._force_close(None)
test_utils.run_briefly(self.loop)
self.assertEqual(1, self.protocol.connection_lost.call_count)
def test_close_protocol_connection_lost_once(self):
tr = self.socket_transport()
self.assertFalse(self.protocol.connection_lost.called)
# Calling close twice should not call
# protocol.connection_lost twice
tr.close()
tr.close()
test_utils.run_briefly(self.loop)
self.assertEqual(1, self.protocol.connection_lost.call_count)
def test_fatal_error_2(self):
tr = self.socket_transport()
tr._buffer = [b'data']