diff options
Diffstat (limited to 'Lib')
| -rw-r--r-- | Lib/socket.py | 6 | ||||
| -rw-r--r-- | Lib/test/test_socket.py | 14 |
2 files changed, 17 insertions, 3 deletions
diff --git a/Lib/socket.py b/Lib/socket.py index ed1b10a254..95ce9eb61c 100644 --- a/Lib/socket.py +++ b/Lib/socket.py @@ -209,10 +209,10 @@ class socket(_socket.socket): encoding=None, errors=None, newline=None): """makefile(...) -> an I/O stream connected to the socket - The arguments are as for io.open() after the filename, - except the only mode characters supported are 'r', 'w' and 'b'. - The semantics are similar too. (XXX refactor to share code?) + The arguments are as for io.open() after the filename, except the only + supported mode values are 'r' (default), 'w' and 'b'. """ + # XXX refactor to share code? if not set(mode) <= {"r", "w", "b"}: raise ValueError("invalid mode %r (only r, w, b allowed)" % (mode,)) writing = "w" in mode diff --git a/Lib/test/test_socket.py b/Lib/test/test_socket.py index 1e355ea8fe..3d6d205e02 100644 --- a/Lib/test/test_socket.py +++ b/Lib/test/test_socket.py @@ -1374,6 +1374,20 @@ class GeneralModuleTests(unittest.TestCase): self.assertRaises(ValueError, fp.writable) self.assertRaises(ValueError, fp.seekable) + def test_makefile_mode(self): + for mode in 'r', 'rb', 'rw', 'w', 'wb': + with self.subTest(mode=mode): + with socket.socket() as sock: + with sock.makefile(mode) as fp: + self.assertEqual(fp.mode, mode) + + def test_makefile_invalid_mode(self): + for mode in 'rt', 'x', '+', 'a': + with self.subTest(mode=mode): + with socket.socket() as sock: + with self.assertRaisesRegex(ValueError, 'invalid mode'): + sock.makefile(mode) + def test_pickle(self): sock = socket.socket() with sock: |