Add test_plistlib.py
this is basically the same cpython plistlib test suite, rewritten as pytest-style tests
This commit is contained in:
Cosimo Lupo
parent
c585dac259
commit
61c19acace
385
Lib/ufoLib/test/test_plistlib.py
Normal file
385
Lib/ufoLib/test/test_plistlib.py
Normal file
@ -0,0 +1,385 @@
|
||||
from __future__ import absolute_import, unicode_literals
|
||||
from ufoLib import plistlib
|
||||
import os
|
||||
import datetime
|
||||
import codecs
|
||||
import collections
|
||||
from io import BytesIO
|
||||
from numbers import Integral
|
||||
from lxml import etree
|
||||
import pytest
|
||||
|
||||
|
||||
# The testdata is generated using https://github.com/python/cpython/...
|
||||
# Mac/Tools/plistlib_generate_testdata.py
|
||||
# which uses PyObjC to control the Cocoa classes for generating plists
|
||||
datadir = os.path.join(os.path.dirname(__file__), "testdata")
|
||||
with open(os.path.join(datadir, "test.plist"), "rb") as fp:
|
||||
TESTDATA = fp.read()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def pl():
|
||||
data = dict(
|
||||
aString="Doodah",
|
||||
aList=["A", "B", 12, 32.5, [1, 2, 3]],
|
||||
aFloat=0.5,
|
||||
anInt=728,
|
||||
aBigInt=2 ** 63 - 44,
|
||||
aBigInt2=2 ** 63 + 44,
|
||||
aNegativeInt=-5,
|
||||
aNegativeBigInt=-80000000000,
|
||||
aDict=dict(
|
||||
anotherString="<hello & 'hi' there!>",
|
||||
aUnicodeValue="M\xe4ssig, Ma\xdf",
|
||||
aTrueValue=True,
|
||||
aFalseValue=False,
|
||||
deeperDict=dict(a=17, b=32.5, c=[1, 2, "text"]),
|
||||
),
|
||||
someData=b"<binary gunk>",
|
||||
someMoreData=b"<lots of binary gunk>\0\1\2\3" * 10,
|
||||
nestedData=[b"<lots of binary gunk>\0\1\2\3" * 10],
|
||||
aDate=datetime.datetime(2004, 10, 26, 10, 33, 33),
|
||||
anEmptyDict=dict(),
|
||||
anEmptyList=list(),
|
||||
)
|
||||
data["\xc5benraa"] = "That was a unicode key."
|
||||
return data
|
||||
|
||||
|
||||
def test_io(tmpdir, pl):
|
||||
testpath = tmpdir / "test.plist"
|
||||
with testpath.open("wb") as fp:
|
||||
plistlib.dump(pl, fp)
|
||||
|
||||
with testpath.open("rb") as fp:
|
||||
pl2 = plistlib.load(fp)
|
||||
|
||||
assert pl == pl2
|
||||
|
||||
with pytest.raises(AttributeError):
|
||||
plistlib.dump(pl, "filename")
|
||||
|
||||
with pytest.raises(AttributeError):
|
||||
plistlib.load("filename")
|
||||
|
||||
|
||||
def test_invalid_type():
|
||||
pl = [object()]
|
||||
|
||||
with pytest.raises(TypeError):
|
||||
plistlib.dumps(pl)
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"pl",
|
||||
[
|
||||
0,
|
||||
2 ** 8 - 1,
|
||||
2 ** 8,
|
||||
2 ** 16 - 1,
|
||||
2 ** 16,
|
||||
2 ** 32 - 1,
|
||||
2 ** 32,
|
||||
2 ** 63 - 1,
|
||||
2 ** 64 - 1,
|
||||
1,
|
||||
-2 ** 63,
|
||||
],
|
||||
)
|
||||
def test_int(pl):
|
||||
data = plistlib.dumps(pl)
|
||||
pl2 = plistlib.loads(data)
|
||||
assert isinstance(pl2, Integral)
|
||||
assert pl == pl2
|
||||
data2 = plistlib.dumps(pl2)
|
||||
assert data == data2
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"pl", [2 ** 64 + 1, 2 ** 127 - 1, -2 ** 64, -2 ** 127]
|
||||
)
|
||||
def test_int_overflow(pl):
|
||||
with pytest.raises(OverflowError):
|
||||
plistlib.dumps(pl)
|
||||
|
||||
|
||||
def test_bytearray(pl):
|
||||
pl = b"<binary gunk\0\1\2\3>"
|
||||
data = plistlib.dumps(bytearray(pl))
|
||||
pl2 = plistlib.loads(data)
|
||||
assert isinstance(pl2, bytes)
|
||||
assert pl2 == pl
|
||||
data2 = plistlib.dumps(pl2)
|
||||
assert data == data2
|
||||
|
||||
|
||||
def test_bytes(pl):
|
||||
pl = b"<binary gunk\0\1\2\3>"
|
||||
data = plistlib.dumps(pl)
|
||||
pl2 = plistlib.loads(data)
|
||||
assert isinstance(pl2, bytes)
|
||||
assert pl2 == pl
|
||||
data2 = plistlib.dumps(pl2)
|
||||
assert data == data2
|
||||
|
||||
|
||||
def test_indentation_array():
|
||||
data = [[[[[[[[{"test": b"aaaaaa"}]]]]]]]]
|
||||
assert plistlib.loads(plistlib.dumps(data)) == data
|
||||
|
||||
|
||||
def test_indentation_dict():
|
||||
data = {
|
||||
"1": {"2": {"3": {"4": {"5": {"6": {"7": {"8": {"9": b"aaaaaa"}}}}}}}}
|
||||
}
|
||||
assert plistlib.loads(plistlib.dumps(data)) == data
|
||||
|
||||
|
||||
def test_indentation_dict_mix():
|
||||
data = {"1": {"2": [{"3": [[[[[{"test": b"aaaaaa"}]]]]]}]}}
|
||||
assert plistlib.loads(plistlib.dumps(data)) == data
|
||||
|
||||
|
||||
@pytest.mark.xfail(reason="we use two spaces, Apple uses tabs")
|
||||
def test_apple_formatting():
|
||||
# and we also don't split base64 data in multiple chunks
|
||||
# per line, since we don't know the indentation level.
|
||||
# maybe we could guess it, but don't know if worth it.
|
||||
pl = plistlib.loads(TESTDATA)
|
||||
data = plistlib.dumps(pl)
|
||||
assert data == TESTDATA
|
||||
|
||||
|
||||
def test_apple_formatting_fromliteral(pl):
|
||||
pl2 = plistlib.loads(TESTDATA)
|
||||
assert pl == pl2
|
||||
|
||||
|
||||
def test_apple_roundtrips():
|
||||
pl = plistlib.loads(TESTDATA)
|
||||
data = plistlib.dumps(pl)
|
||||
pl2 = plistlib.loads(data)
|
||||
data2 = plistlib.dumps(pl2)
|
||||
assert data == data2
|
||||
|
||||
|
||||
def test_bytesio(pl):
|
||||
b = BytesIO()
|
||||
plistlib.dump(pl, b)
|
||||
pl2 = plistlib.load(BytesIO(b.getvalue()))
|
||||
assert pl == pl2
|
||||
pl2 = plistlib.load(BytesIO(b.getvalue()))
|
||||
assert pl == pl2
|
||||
|
||||
|
||||
@pytest.mark.parametrize("sort_keys", [False, True])
|
||||
def test_keysort_bytesio(sort_keys):
|
||||
pl = collections.OrderedDict()
|
||||
pl["b"] = 1
|
||||
pl["a"] = 2
|
||||
pl["c"] = 3
|
||||
|
||||
b = BytesIO()
|
||||
|
||||
plistlib.dump(pl, b, sort_keys=sort_keys)
|
||||
pl2 = plistlib.load(
|
||||
BytesIO(b.getvalue()), dict_type=collections.OrderedDict
|
||||
)
|
||||
|
||||
assert dict(pl) == dict(pl2)
|
||||
if sort_keys:
|
||||
assert list(pl2.keys()) == ["a", "b", "c"]
|
||||
else:
|
||||
assert list(pl2.keys()) == ["b", "a", "c"]
|
||||
|
||||
|
||||
@pytest.mark.parametrize("sort_keys", [False, True])
|
||||
def test_keysort(sort_keys):
|
||||
pl = collections.OrderedDict()
|
||||
pl["b"] = 1
|
||||
pl["a"] = 2
|
||||
pl["c"] = 3
|
||||
|
||||
data = plistlib.dumps(pl, sort_keys=sort_keys)
|
||||
pl2 = plistlib.loads(data, dict_type=collections.OrderedDict)
|
||||
|
||||
assert dict(pl) == dict(pl2)
|
||||
if sort_keys:
|
||||
assert list(pl2.keys()) == ["a", "b", "c"]
|
||||
else:
|
||||
assert list(pl2.keys()) == ["b", "a", "c"]
|
||||
|
||||
|
||||
def test_keys_no_string():
|
||||
pl = {42: "aNumber"}
|
||||
|
||||
with pytest.raises(TypeError):
|
||||
plistlib.dumps(pl)
|
||||
|
||||
b = BytesIO()
|
||||
with pytest.raises(TypeError):
|
||||
plistlib.dump(pl, b)
|
||||
|
||||
|
||||
def test_skipkeys():
|
||||
pl = {42: "aNumber", "snake": "aWord"}
|
||||
|
||||
data = plistlib.dumps(pl, skipkeys=True, sort_keys=False)
|
||||
|
||||
pl2 = plistlib.loads(data)
|
||||
assert pl2 == {"snake": "aWord"}
|
||||
|
||||
fp = BytesIO()
|
||||
plistlib.dump(pl, fp, skipkeys=True, sort_keys=False)
|
||||
data = fp.getvalue()
|
||||
pl2 = plistlib.loads(fp.getvalue())
|
||||
assert pl2 == {"snake": "aWord"}
|
||||
|
||||
|
||||
def test_tuple_members():
|
||||
pl = {"first": (1, 2), "second": (1, 2), "third": (3, 4)}
|
||||
|
||||
data = plistlib.dumps(pl)
|
||||
pl2 = plistlib.loads(data)
|
||||
assert pl2 == {"first": [1, 2], "second": [1, 2], "third": [3, 4]}
|
||||
assert pl2["first"] is not pl2["second"]
|
||||
|
||||
|
||||
def test_list_members():
|
||||
pl = {"first": [1, 2], "second": [1, 2], "third": [3, 4]}
|
||||
|
||||
data = plistlib.dumps(pl)
|
||||
pl2 = plistlib.loads(data)
|
||||
assert pl2 == {"first": [1, 2], "second": [1, 2], "third": [3, 4]}
|
||||
assert pl2["first"] is not pl2["second"]
|
||||
|
||||
|
||||
def test_dict_members():
|
||||
pl = {"first": {"a": 1}, "second": {"a": 1}, "third": {"b": 2}}
|
||||
|
||||
data = plistlib.dumps(pl)
|
||||
pl2 = plistlib.loads(data)
|
||||
assert pl2 == {"first": {"a": 1}, "second": {"a": 1}, "third": {"b": 2}}
|
||||
assert pl2["first"] is not pl2["second"]
|
||||
|
||||
|
||||
def test_controlcharacters():
|
||||
for i in range(128):
|
||||
c = chr(i)
|
||||
testString = "string containing %s" % c
|
||||
if i >= 32 or c in "\r\n\t":
|
||||
# \r, \n and \t are the only legal control chars in XML
|
||||
data = plistlib.dumps(testString)
|
||||
# the stdlib's plistlib writer always replaces \r with \n
|
||||
# inside string values; we don't (the ctrl character is
|
||||
# escaped by lxml, so it roundtrips)
|
||||
# if c != "\r":
|
||||
assert plistlib.loads(data) == testString
|
||||
else:
|
||||
with pytest.raises(ValueError):
|
||||
plistlib.dumps(testString)
|
||||
|
||||
|
||||
def test_non_bmp_characters():
|
||||
pl = {"python": "\U0001f40d"}
|
||||
data = plistlib.dumps(pl)
|
||||
assert plistlib.loads(data) == pl
|
||||
|
||||
|
||||
def test_nondictroot():
|
||||
test1 = "abc"
|
||||
test2 = [1, 2, 3, "abc"]
|
||||
result1 = plistlib.loads(plistlib.dumps(test1))
|
||||
result2 = plistlib.loads(plistlib.dumps(test2))
|
||||
assert test1 == result1
|
||||
assert test2 == result2
|
||||
|
||||
|
||||
def test_invalidarray():
|
||||
for i in [
|
||||
"<key>key inside an array</key>",
|
||||
"<key>key inside an array2</key><real>3</real>",
|
||||
"<true/><key>key inside an array3</key>",
|
||||
]:
|
||||
with pytest.raises(ValueError):
|
||||
plistlib.loads(
|
||||
("<plist><array>%s</array></plist>" % i).encode("utf-8")
|
||||
)
|
||||
|
||||
|
||||
def test_invaliddict():
|
||||
for i in [
|
||||
"<key><true/>k</key><string>compound key</string>",
|
||||
"<key>single key</key>",
|
||||
"<string>missing key</string>",
|
||||
"<key>k1</key><string>v1</string><real>5.3</real>"
|
||||
"<key>k1</key><key>k2</key><string>double key</string>",
|
||||
]:
|
||||
with pytest.raises(ValueError):
|
||||
plistlib.loads(("<plist><dict>%s</dict></plist>" % i).encode())
|
||||
with pytest.raises(ValueError):
|
||||
plistlib.loads(
|
||||
("<plist><array><dict>%s</dict></array></plist>" % i).encode()
|
||||
)
|
||||
|
||||
|
||||
def test_invalidinteger():
|
||||
with pytest.raises(ValueError):
|
||||
plistlib.loads(b"<plist><integer>not integer</integer></plist>")
|
||||
|
||||
|
||||
def test_invalidreal():
|
||||
with pytest.raises(ValueError):
|
||||
plistlib.loads(b"<plist><integer>not real</integer></plist>")
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"xml_encoding, encoding, bom",
|
||||
[
|
||||
(b"utf-8", "utf-8", codecs.BOM_UTF8),
|
||||
(b"utf-16", "utf-16-le", codecs.BOM_UTF16_LE),
|
||||
(b"utf-16", "utf-16-be", codecs.BOM_UTF16_BE),
|
||||
(b"utf-32", "utf-32-le", codecs.BOM_UTF32_LE),
|
||||
(b"utf-32", "utf-32-be", codecs.BOM_UTF32_BE),
|
||||
],
|
||||
)
|
||||
def test_xml_encodings(pl, xml_encoding, encoding, bom):
|
||||
data = TESTDATA.replace(b"UTF-8", xml_encoding)
|
||||
data = bom + data.decode("utf-8").encode(encoding)
|
||||
pl2 = plistlib.loads(data)
|
||||
assert pl == pl2
|
||||
|
||||
|
||||
def test_fromtree(pl):
|
||||
tree = etree.fromstring(TESTDATA)
|
||||
pl2 = plistlib.fromtree(tree)
|
||||
assert pl == pl2
|
||||
|
||||
|
||||
def test_totree(pl):
|
||||
tree = etree.fromstring(TESTDATA)[0] # ignore root 'plist' element
|
||||
tree2 = plistlib.totree(pl)
|
||||
assert tree.tag == tree2.tag == "dict"
|
||||
for (_, e1), (_, e2) in zip(etree.iterwalk(tree), etree.iterwalk(tree2)):
|
||||
assert e1.tag == e2.tag
|
||||
assert e1.attrib == e2.attrib
|
||||
assert len(e1) == len(e2)
|
||||
# ignore indentation
|
||||
text1 = (
|
||||
"".join(l.strip() for l in e1.text.splitlines())
|
||||
if e1.text is not None
|
||||
else ""
|
||||
)
|
||||
text2 = (
|
||||
"".join(l.strip() for l in e2.text.splitlines())
|
||||
if e2.text is not None
|
||||
else ""
|
||||
)
|
||||
assert text1 == text2
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
import sys
|
||||
|
||||
sys.exit(pytest.main(sys.argv))
|
||||
87
Lib/ufoLib/test/testdata/test.plist
vendored
Normal file
87
Lib/ufoLib/test/testdata/test.plist
vendored
Normal file
@ -0,0 +1,87 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
|
||||
<plist version="1.0">
|
||||
<dict>
|
||||
<key>aBigInt</key>
|
||||
<integer>9223372036854775764</integer>
|
||||
<key>aBigInt2</key>
|
||||
<integer>9223372036854775852</integer>
|
||||
<key>aDate</key>
|
||||
<date>2004-10-26T10:33:33Z</date>
|
||||
<key>aDict</key>
|
||||
<dict>
|
||||
<key>aFalseValue</key>
|
||||
<false/>
|
||||
<key>aTrueValue</key>
|
||||
<true/>
|
||||
<key>aUnicodeValue</key>
|
||||
<string>Mässig, Maß</string>
|
||||
<key>anotherString</key>
|
||||
<string><hello & 'hi' there!></string>
|
||||
<key>deeperDict</key>
|
||||
<dict>
|
||||
<key>a</key>
|
||||
<integer>17</integer>
|
||||
<key>b</key>
|
||||
<real>32.5</real>
|
||||
<key>c</key>
|
||||
<array>
|
||||
<integer>1</integer>
|
||||
<integer>2</integer>
|
||||
<string>text</string>
|
||||
</array>
|
||||
</dict>
|
||||
</dict>
|
||||
<key>aFloat</key>
|
||||
<real>0.5</real>
|
||||
<key>aList</key>
|
||||
<array>
|
||||
<string>A</string>
|
||||
<string>B</string>
|
||||
<integer>12</integer>
|
||||
<real>32.5</real>
|
||||
<array>
|
||||
<integer>1</integer>
|
||||
<integer>2</integer>
|
||||
<integer>3</integer>
|
||||
</array>
|
||||
</array>
|
||||
<key>aNegativeBigInt</key>
|
||||
<integer>-80000000000</integer>
|
||||
<key>aNegativeInt</key>
|
||||
<integer>-5</integer>
|
||||
<key>aString</key>
|
||||
<string>Doodah</string>
|
||||
<key>anEmptyDict</key>
|
||||
<dict/>
|
||||
<key>anEmptyList</key>
|
||||
<array/>
|
||||
<key>anInt</key>
|
||||
<integer>728</integer>
|
||||
<key>nestedData</key>
|
||||
<array>
|
||||
<data>
|
||||
PGxvdHMgb2YgYmluYXJ5IGd1bms+AAECAzxsb3RzIG9mIGJpbmFyeSBndW5r
|
||||
PgABAgM8bG90cyBvZiBiaW5hcnkgZ3Vuaz4AAQIDPGxvdHMgb2YgYmluYXJ5
|
||||
IGd1bms+AAECAzxsb3RzIG9mIGJpbmFyeSBndW5rPgABAgM8bG90cyBvZiBi
|
||||
aW5hcnkgZ3Vuaz4AAQIDPGxvdHMgb2YgYmluYXJ5IGd1bms+AAECAzxsb3Rz
|
||||
IG9mIGJpbmFyeSBndW5rPgABAgM8bG90cyBvZiBiaW5hcnkgZ3Vuaz4AAQID
|
||||
PGxvdHMgb2YgYmluYXJ5IGd1bms+AAECAw==
|
||||
</data>
|
||||
</array>
|
||||
<key>someData</key>
|
||||
<data>
|
||||
PGJpbmFyeSBndW5rPg==
|
||||
</data>
|
||||
<key>someMoreData</key>
|
||||
<data>
|
||||
PGxvdHMgb2YgYmluYXJ5IGd1bms+AAECAzxsb3RzIG9mIGJpbmFyeSBndW5rPgABAgM8
|
||||
bG90cyBvZiBiaW5hcnkgZ3Vuaz4AAQIDPGxvdHMgb2YgYmluYXJ5IGd1bms+AAECAzxs
|
||||
b3RzIG9mIGJpbmFyeSBndW5rPgABAgM8bG90cyBvZiBiaW5hcnkgZ3Vuaz4AAQIDPGxv
|
||||
dHMgb2YgYmluYXJ5IGd1bms+AAECAzxsb3RzIG9mIGJpbmFyeSBndW5rPgABAgM8bG90
|
||||
cyBvZiBiaW5hcnkgZ3Vuaz4AAQIDPGxvdHMgb2YgYmluYXJ5IGd1bms+AAECAw==
|
||||
</data>
|
||||
<key>Åbenraa</key>
|
||||
<string>That was a unicode key.</string>
|
||||
</dict>
|
||||
</plist>
|
||||
Loading…
x
Reference in New Issue
Block a user