diff options
| author | Guido van Rossum <guido@python.org> | 2013-12-03 02:36:30 (GMT) |
|---|---|---|
| committer | Guido van Rossum <guido@python.org> | 2013-12-03 02:36:30 (GMT) |
| commit | f10345e1026c087d0e2cba7187a94df769b55e62 (patch) | |
| tree | 129acc43ee33d29c6b22f44fc1fd1271f25887c4 /Lib/test/test_asyncio/test_transports.py | |
| parent | afbf827875dc2fb0cb603b03fe8f79c8ac1e30af (diff) | |
| download | cpython-f10345e1026c087d0e2cba7187a94df769b55e62.zip cpython-f10345e1026c087d0e2cba7187a94df769b55e62.tar.gz cpython-f10345e1026c087d0e2cba7187a94df769b55e62.tar.bz2 | |
asyncio: Improve default writelines().
Diffstat (limited to 'Lib/test/test_asyncio/test_transports.py')
| -rw-r--r-- | Lib/test/test_asyncio/test_transports.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/Lib/test/test_asyncio/test_transports.py b/Lib/test/test_asyncio/test_transports.py index f96445c..29393b5 100644 --- a/Lib/test/test_asyncio/test_transports.py +++ b/Lib/test/test_asyncio/test_transports.py @@ -24,12 +24,18 @@ class TransportTests(unittest.TestCase): transport = transports.Transport() transport.write = unittest.mock.Mock() - transport.writelines(['line1', 'line2', 'line3']) - self.assertEqual(3, transport.write.call_count) + transport.writelines([b'line1', + bytearray(b'line2'), + memoryview(b'line3')]) + self.assertEqual(1, transport.write.call_count) + transport.write.assert_called_with(b'line1line2line3') def test_not_implemented(self): transport = transports.Transport() + self.assertRaises(NotImplementedError, + transport.set_write_buffer_limits) + self.assertRaises(NotImplementedError, transport.get_write_buffer_size) self.assertRaises(NotImplementedError, transport.write, 'data') self.assertRaises(NotImplementedError, transport.write_eof) self.assertRaises(NotImplementedError, transport.can_write_eof) |