summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_asyncio/test_streams.py
diff options
context:
space:
mode:
Diffstat (limited to 'Lib/test/test_asyncio/test_streams.py')
-rw-r--r--Lib/test/test_asyncio/test_streams.py18
1 files changed, 18 insertions, 0 deletions
diff --git a/Lib/test/test_asyncio/test_streams.py b/Lib/test/test_asyncio/test_streams.py
index b47433a..6d16d20 100644
--- a/Lib/test/test_asyncio/test_streams.py
+++ b/Lib/test/test_asyncio/test_streams.py
@@ -3,6 +3,7 @@
import gc
import os
import queue
+import pickle
import socket
import sys
import threading
@@ -845,6 +846,23 @@ os.close(fd)
stream._transport.__repr__.return_value = "<Transport>"
self.assertEqual("<StreamReader t=<Transport>>", repr(stream))
+ def test_IncompleteReadError_pickleable(self):
+ e = asyncio.IncompleteReadError(b'abc', 10)
+ for proto in range(pickle.HIGHEST_PROTOCOL + 1):
+ with self.subTest(pickle_protocol=proto):
+ e2 = pickle.loads(pickle.dumps(e, protocol=proto))
+ self.assertEqual(str(e), str(e2))
+ self.assertEqual(e.partial, e2.partial)
+ self.assertEqual(e.expected, e2.expected)
+
+ def test_LimitOverrunError_pickleable(self):
+ e = asyncio.LimitOverrunError('message', 10)
+ for proto in range(pickle.HIGHEST_PROTOCOL + 1):
+ with self.subTest(pickle_protocol=proto):
+ e2 = pickle.loads(pickle.dumps(e, protocol=proto))
+ self.assertEqual(str(e), str(e2))
+ self.assertEqual(e.consumed, e2.consumed)
+
if __name__ == '__main__':
unittest.main()