mirror of
https://github.com/python/cpython.git
synced 2025-09-26 18:29:57 +00:00
#15377: Make posixpath.join() more strict when checking for str/bytes mix
Based on a patch by Nick Coghlan.
This commit is contained in:
parent
eb3e62f1c8
commit
c5a4566922
2 changed files with 18 additions and 13 deletions
|
@ -83,11 +83,12 @@ def join(a, *p):
|
||||||
else:
|
else:
|
||||||
path += sep + b
|
path += sep + b
|
||||||
except TypeError:
|
except TypeError:
|
||||||
strs = [isinstance(s, str) for s in (a, ) + p]
|
valid_types = all(isinstance(s, (str, bytes, bytearray))
|
||||||
if any(strs) and not all(strs):
|
for s in (a, ) + p)
|
||||||
|
if valid_types:
|
||||||
|
# Must have a mixture of text and binary data
|
||||||
raise TypeError("Can't mix strings and bytes in path components.")
|
raise TypeError("Can't mix strings and bytes in path components.")
|
||||||
else:
|
raise
|
||||||
raise
|
|
||||||
return path
|
return path
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -1,6 +1,7 @@
|
||||||
import unittest
|
import unittest
|
||||||
from test import support, test_genericpath
|
from test import support, test_genericpath
|
||||||
|
|
||||||
|
import itertools
|
||||||
import posixpath
|
import posixpath
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
|
@ -56,18 +57,21 @@ class PosixPathTest(unittest.TestCase):
|
||||||
self.assertEqual(posixpath.join(b"/foo/", b"bar/", b"baz/"),
|
self.assertEqual(posixpath.join(b"/foo/", b"bar/", b"baz/"),
|
||||||
b"/foo/bar/baz/")
|
b"/foo/bar/baz/")
|
||||||
|
|
||||||
# Check for friendly str/bytes mixing message
|
def check_error_msg(list_of_args, msg):
|
||||||
for args in [[b'bytes', 'str'],
|
"""Check posixpath.join raises friendly TypeErrors."""
|
||||||
[bytearray(b'bytes'), 'str']]:
|
for args in (item for perm in list_of_args
|
||||||
for _ in range(2):
|
for item in itertools.permutations(perm)):
|
||||||
with self.assertRaises(TypeError) as cm:
|
with self.assertRaises(TypeError) as cm:
|
||||||
posixpath.join(*args)
|
posixpath.join(*args)
|
||||||
self.assertEqual(
|
self.assertEqual(msg, cm.exception.args[0])
|
||||||
"Can't mix strings and bytes in path components.",
|
|
||||||
cm.exception.args[0]
|
|
||||||
)
|
|
||||||
args.reverse() # check both orders
|
|
||||||
|
|
||||||
|
check_error_msg([[b'bytes', 'str'], [bytearray(b'bytes'), 'str']],
|
||||||
|
"Can't mix strings and bytes in path components.")
|
||||||
|
# regression, see #15377
|
||||||
|
with self.assertRaises(TypeError) as cm:
|
||||||
|
os.path.join(None, 'str')
|
||||||
|
self.assertNotEqual("Can't mix strings and bytes in path components.",
|
||||||
|
cm.exception.args[0])
|
||||||
|
|
||||||
def test_split(self):
|
def test_split(self):
|
||||||
self.assertEqual(posixpath.split("/foo/bar"), ("/foo", "bar"))
|
self.assertEqual(posixpath.split("/foo/bar"), ("/foo", "bar"))
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue