fergalmoran/picard
1
0
Fork 0
mirror of https://github.com/fergalmoran/picard.git synced 2025-12-27 03:37:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Reduce cover art vs formats test code redundancy.

Browse Source
This commit is contained in:
Laurent Monin
2013-06-08 00:54:09 +02:00
parent 5bc82bfa53
commit b246e4b709
1 changed files with 16 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
test/test_formats.py
View File

@@ -532,27 +532,24 @@ class TestCoverArt(unittest.TestCase):
def _test_cover_art(self, filename):
self._set_up(filename)
try:
f = picard.formats.open(self.filename)
metadata = Metadata()
# Use reasonable large data > 64kb.
# This checks a mutagen error with ASF files.
jpegFakeData = "JFIF" + ("a" * 1024 * 128)
metadata.add_image("image/jpeg", jpegFakeData)
f._save(self.filename, metadata, f.config.setting)
dummyload = "a" * 1024 * 128
tests = {
'jpg': {'mime': 'image/jpeg', 'head': 'JFIF'},
'png': {'mime': 'image/png', 'head': 'PNG'},
}
for t in tests:
f = picard.formats.open(self.filename)
metadata = Metadata()
imgdata = tests[t]['head'] + dummyload
metadata.add_image(tests[t]['mime'], imgdata)
f._save(self.filename, metadata, f.config.setting)
f = picard.formats.open(self.filename)
metadata = f._load(self.filename)
self.assertEqual(metadata.images[0]["mime"], "image/jpeg")
self.assertEqual(metadata.images[0]["data"], jpegFakeData)
f = picard.formats.open(self.filename)
metadata = Metadata()
metadata.add_image("image/png", "PNGfoobar")
f._save(self.filename, metadata, f.config.setting)
f = picard.formats.open(self.filename)
metadata = f._load(self.filename)
self.assertEqual(metadata.images[0]["mime"], "image/png")
self.assertEqual(metadata.images[0]["data"], "PNGfoobar")
f = picard.formats.open(self.filename)
loaded_metadata = f._load(self.filename)
image = loaded_metadata.images[0]
self.assertEqual(image["mime"], tests[t]['mime'])
self.assertEqual(image["data"], imgdata)
finally:
self._tear_down()