255 lines
8.6 KiB
Python
255 lines
8.6 KiB
Python
import datetime as dt
|
|
import plistlib
|
|
|
|
import six
|
|
|
|
from benedict.dicts.io import IODict
|
|
|
|
from .test_io_dict import io_dict_test_case
|
|
|
|
|
|
class io_dict_plist_test_case(io_dict_test_case):
|
|
"""
|
|
This class describes an IODict / plist test case.
|
|
"""
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(io_dict_plist_test_case, self).__init__(*args, **kwargs)
|
|
if six.PY2:
|
|
# fmt: off
|
|
self._dict = dict(
|
|
aString=u"Doodah",
|
|
aList=[
|
|
u"A",
|
|
u"B",
|
|
12,
|
|
32.1,
|
|
[1, 2, 3],
|
|
],
|
|
aFloat=0.1,
|
|
anInt=728,
|
|
aDict=dict(
|
|
anotherString=u"<hello & hi there!>",
|
|
aThirdString=u"M\xe4ssig, Ma\xdf",
|
|
aTrueValue=True,
|
|
aFalseValue=False,
|
|
),
|
|
someData=plistlib.Data("<binary gunk>"),
|
|
someMoreData=plistlib.Data("<lots of binary gunk>" * 10),
|
|
aDate=dt.datetime(
|
|
1985, 4, 3, 23, 55
|
|
), # dt.datetime.fromtimestamp(481413300),
|
|
)
|
|
# fmt: on
|
|
else:
|
|
self._dict = dict(
|
|
aString="Doodah",
|
|
aList=[
|
|
"A",
|
|
"B",
|
|
12,
|
|
32.1,
|
|
[1, 2, 3],
|
|
],
|
|
aFloat=0.1,
|
|
anInt=728,
|
|
aDict=dict(
|
|
anotherString="<hello & hi there!>",
|
|
aThirdString="M\xe4ssig, Ma\xdf",
|
|
aTrueValue=True,
|
|
aFalseValue=False,
|
|
),
|
|
someData=bytes("<binary gunk>", encoding="utf-8"),
|
|
someMoreData=bytes("<lots of binary gunk>" * 10, encoding="utf-8"),
|
|
aDate=dt.datetime(
|
|
1985, 4, 3, 23, 55
|
|
), # dt.datetime.fromtimestamp(481413300),
|
|
)
|
|
# self._dict = {
|
|
# 'aString': 'Doodah',
|
|
# 'aList': ['A', 'B', 12, 32.1, [1, 2, 3]],
|
|
# 'aFloat': 0.1,
|
|
# 'anInt': 728,
|
|
# 'aDict': {
|
|
# 'anotherString': '<hello & hi there!>',
|
|
# 'aThirdString': 'M\xe4ssig, Ma\xdf',
|
|
# 'aTrueValue': True,
|
|
# 'aFalseValue': False,
|
|
# },
|
|
# 'someData': b'<binary gunk>',
|
|
# 'someMoreData': b'<lots of binary gunk>' * 10,
|
|
# 'aDate': dt.datetime.fromtimestamp(481406100),
|
|
# }
|
|
self._plist = """
|
|
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
|
|
<plist version="1.0">
|
|
<dict>
|
|
<key>aDate</key>
|
|
<date>1985-04-03T23:55:00Z</date>
|
|
<key>aDict</key>
|
|
<dict>
|
|
<key>aFalseValue</key>
|
|
<false/>
|
|
<key>aThirdString</key>
|
|
<string>Mässig, Maß</string>
|
|
<key>aTrueValue</key>
|
|
<true/>
|
|
<key>anotherString</key>
|
|
<string><hello & hi there!></string>
|
|
</dict>
|
|
<key>aFloat</key>
|
|
<real>0.1</real>
|
|
<key>aList</key>
|
|
<array>
|
|
<string>A</string>
|
|
<string>B</string>
|
|
<integer>12</integer>
|
|
<real>32.1</real>
|
|
<array>
|
|
<integer>1</integer>
|
|
<integer>2</integer>
|
|
<integer>3</integer>
|
|
</array>
|
|
</array>
|
|
<key>aString</key>
|
|
<string>Doodah</string>
|
|
<key>anInt</key>
|
|
<integer>728</integer>
|
|
<key>someData</key>
|
|
<data>
|
|
PGJpbmFyeSBndW5rPg==
|
|
</data>
|
|
<key>someMoreData</key>
|
|
<data>
|
|
PGxvdHMgb2YgYmluYXJ5IGd1bms+PGxvdHMgb2YgYmluYXJ5IGd1bms+PGxvdHMgb2Yg
|
|
YmluYXJ5IGd1bms+PGxvdHMgb2YgYmluYXJ5IGd1bms+PGxvdHMgb2YgYmluYXJ5IGd1
|
|
bms+PGxvdHMgb2YgYmluYXJ5IGd1bms+PGxvdHMgb2YgYmluYXJ5IGd1bms+PGxvdHMg
|
|
b2YgYmluYXJ5IGd1bms+PGxvdHMgb2YgYmluYXJ5IGd1bms+PGxvdHMgb2YgYmluYXJ5
|
|
IGd1bms+
|
|
</data>
|
|
</dict>
|
|
</plist>
|
|
"""
|
|
|
|
def test_from_plist_with_valid_data(self):
|
|
j = self._plist
|
|
# static method
|
|
d = IODict.from_plist(j)
|
|
self.assertTrue(isinstance(d, dict))
|
|
self.assertEqual(d.get("aDate"), self._dict.get("aDate"))
|
|
self.assertEqual(d, self._dict)
|
|
# constructor
|
|
d = IODict(j, format="plist")
|
|
self.assertTrue(isinstance(d, dict))
|
|
self.assertEqual(d, self._dict)
|
|
|
|
def test_from_plist_with_invalid_data(self):
|
|
j = "Lorem ipsum est in ea occaecat nisi officia."
|
|
# static method
|
|
with self.assertRaises(ValueError):
|
|
IODict.from_plist(j)
|
|
# constructor
|
|
with self.assertRaises(ValueError):
|
|
IODict(j, format="plist")
|
|
|
|
def test_from_plist_with_valid_file_valid_content(self):
|
|
filepath = self.input_path("valid-content.plist")
|
|
# static method
|
|
d = IODict.from_plist(filepath)
|
|
self.assertTrue(isinstance(d, dict))
|
|
# constructor
|
|
d = IODict(filepath, format="plist")
|
|
self.assertTrue(isinstance(d, dict))
|
|
# constructor with format autodetection
|
|
d = IODict(filepath)
|
|
self.assertTrue(isinstance(d, dict))
|
|
|
|
def test_from_plist_with_valid_file_valid_content_invalid_format(self):
|
|
filepath = self.input_path("valid-content.base64")
|
|
with self.assertRaises(ValueError):
|
|
IODict.from_plist(filepath)
|
|
filepath = self.input_path("valid-content.csv")
|
|
with self.assertRaises(ValueError):
|
|
IODict.from_plist(filepath)
|
|
filepath = self.input_path("valid-content.json")
|
|
with self.assertRaises(ValueError):
|
|
IODict.from_plist(filepath)
|
|
filepath = self.input_path("valid-content.pickle")
|
|
with self.assertRaises(ValueError):
|
|
IODict.from_plist(filepath)
|
|
filepath = self.input_path("valid-content.qs")
|
|
with self.assertRaises(ValueError):
|
|
IODict.from_plist(filepath)
|
|
filepath = self.input_path("valid-content.toml")
|
|
with self.assertRaises(ValueError):
|
|
IODict.from_plist(filepath)
|
|
filepath = self.input_path("valid-content.xml")
|
|
with self.assertRaises(ValueError):
|
|
IODict.from_plist(filepath)
|
|
filepath = self.input_path("valid-content.yml")
|
|
with self.assertRaises(ValueError):
|
|
IODict.from_plist(filepath)
|
|
|
|
def test_from_plist_with_valid_file_invalid_content(self):
|
|
filepath = self.input_path("invalid-content.plist")
|
|
# static method
|
|
with self.assertRaises(ValueError):
|
|
IODict.from_plist(filepath)
|
|
# constructor
|
|
with self.assertRaises(ValueError):
|
|
IODict(filepath, format="plist")
|
|
|
|
def test_from_plist_with_invalid_file(self):
|
|
filepath = self.input_path("invalid-file.plist")
|
|
# static method
|
|
with self.assertRaises(ValueError):
|
|
IODict.from_plist(filepath)
|
|
# constructor
|
|
with self.assertRaises(ValueError):
|
|
IODict(filepath, format="plist")
|
|
|
|
def test_from_plist_with_valid_url_valid_content(self):
|
|
url = self.input_url("valid-content.plist")
|
|
# static method
|
|
d = IODict.from_plist(url)
|
|
self.assertTrue(isinstance(d, dict))
|
|
# constructor
|
|
d = IODict(url, format="plist")
|
|
self.assertTrue(isinstance(d, dict))
|
|
# constructor with format autodetection
|
|
d = IODict(url)
|
|
self.assertTrue(isinstance(d, dict))
|
|
|
|
def test_from_plist_with_valid_url_invalid_content(self):
|
|
url = "https://github.com/fabiocaccamo/python-benedict"
|
|
# static method
|
|
with self.assertRaises(ValueError):
|
|
IODict.from_plist(url)
|
|
# constructor
|
|
with self.assertRaises(ValueError):
|
|
IODict(url, format="plist")
|
|
|
|
def test_from_plist_with_invalid_url(self):
|
|
url = "https://github.com/fabiocaccamo/python-benedict-invalid"
|
|
# static method
|
|
with self.assertRaises(ValueError):
|
|
IODict.from_plist(url)
|
|
# constructor
|
|
with self.assertRaises(ValueError):
|
|
IODict(url, format="plist")
|
|
|
|
def test_to_plist(self):
|
|
# example data taken from:
|
|
# https://docs.python.org/3/library/plistlib.html#examples
|
|
d = IODict(self._dict)
|
|
s = d.to_plist()
|
|
# print(s)
|
|
self.assertEqual(d, IODict.from_plist(s))
|
|
|
|
def test_to_plist_file(self):
|
|
d = IODict(self._dict)
|
|
filepath = self.output_path("test_to_plist_file.plist")
|
|
d.to_plist(filepath=filepath)
|
|
self.assertFileExists(filepath)
|
|
self.assertEqual(d, IODict.from_plist(filepath))
|