diff options
Diffstat (limited to 'tests/test_streams.py')
| -rw-r--r-- | tests/test_streams.py | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/tests/test_streams.py b/tests/test_streams.py index 242b377..c883ba3 100644 --- a/tests/test_streams.py +++ b/tests/test_streams.py @@ -636,6 +636,23 @@ os.close(fd) protocol = asyncio.StreamReaderProtocol(reader) self.assertIs(protocol._loop, self.loop) + def test_pause_writing_closing(self): + reader = mock.Mock() + transport = asyncio.ReadTransport() + protocol = asyncio.StreamReaderProtocol(reader, loop=self.loop) + protocol.connection_made(transport) + transport._closing = True + self.assertRaises(RuntimeError, protocol.pause_writing) + + def test_resume_writing_closing(self): + reader = mock.Mock() + transport = asyncio.ReadTransport() + protocol = asyncio.StreamReaderProtocol(reader, loop=self.loop) + protocol.connection_made(transport) + protocol.pause_writing() + transport._closing = True + self.assertRaises(RuntimeError, protocol.resume_writing) + if __name__ == '__main__': unittest.main() |