def test_covr_blank_format(self):
data = Atom.render(
b"data", b"\x00\x00\x00\x00" + b"\x00" * 4 + b"whee")
covr = Atom.render(b"covr", data)
tags = self.wrap_ilst(covr)
self.failUnlessEqual(
MP4Cover.FORMAT_JPEG, tags["covr"][0].imageformat)
def test_write_back_bad_atoms(self):
# write a broken atom and try to load it
data = Atom.render(b"datA", b"\x00\x00\x00\x01\x00\x00\x00\x00wheeee")
data = Atom.render(b"aART", data)
tags = self.wrap_ilst(data)
self.assertFalse(tags)
# save it into an existing mp4
original = os.path.join(DATA_DIR, "has-tags.m4a")
filename = get_temp_copy(original)
try:
delete(filename)
# it should still end up in the file
tags.save(filename)
with open(filename, "rb") as h:
self.assertTrue(b"wheeee" in h.read())
# if we define our own aART throw away the broken one
tags["aART"] = ["new"]
tags.save(filename)
with open(filename, "rb") as h:
self.assertFalse(b"wheeee" in h.read())
# add the broken one back and delete all tags including
# the broken one
del tags["aART"]
tags.save(filename)
with open(filename, "rb") as h:
self.assertTrue(b"wheeee" in h.read())
delete(filename)
with open(filename, "rb") as h:
self.assertFalse(b"wheeee" in h.read())
finally:
os.unlink(filename)
def test_write_back_bad_atoms(self):
# write a broken atom and try to load it
data = Atom.render(b"datA", b"\x00\x00\x00\x01\x00\x00\x00\x00wheeee")
data = Atom.render(b"aART", data)
tags = self.wrap_ilst(data)
self.assertFalse(tags)
# save it into an existing mp4
original = os.path.join("tests", "data", "has-tags.m4a")
fd, filename = mkstemp(suffix=".mp4")
os.close(fd)
shutil.copy(original, filename)
delete(filename)
# it should still end up in the file
tags.save(filename)
with open(filename, "rb") as h:
self.assertTrue(b"wheeee" in h.read())
# if we define our own aART throw away the broken one
tags["aART"] = ["new"]
tags.save(filename)
with open(filename, "rb") as h:
self.assertFalse(b"wheeee" in h.read())
# add the broken one back and delete all tags including the broken one
del tags["aART"]
tags.save(filename)
with open(filename, "rb") as h:
self.assertTrue(b"wheeee" in h.read())
delete(filename)
with open(filename, "rb") as h:
self.assertFalse(b"wheeee" in h.read())
def test_parse_multiple_atoms(self):
# while we don't write multiple values as multiple atoms
# still read them
# https://github.com/quodlibet/mutagen/issues/165
data = Atom.render(b"data", b"\x00\x00\x00\x01" + b"\x00" * 4 + b"foo")
grp1 = Atom.render(b"\xa9grp", data)
data = Atom.render(b"data", b"\x00\x00\x00\x01" + b"\x00" * 4 + b"bar")
grp2 = Atom.render(b"\xa9grp", data)
tags = self.wrap_ilst(grp1 + grp2)
self.assertEqual(tags["\xa9grp"], [u"foo", u"bar"])
def test_mdhd_version_1(self, soun="soun"):
mdhd = Atom.render("mdhd", ("\x01\x00\x00\x00" + "\x00" * 16 +
"\x00\x00\x00\x02" + # 2 Hz
"\x00\x00\x00\x00\x00\x00\x00\x10"))
hdlr = Atom.render("hdlr", "\x00" * 8 + soun)
mdia = Atom.render("mdia", mdhd + hdlr)
trak = Atom.render("trak", mdia)
moov = Atom.render("moov", trak)
fileobj = StringIO(moov)
atoms = Atoms(fileobj)
info = MP4Info(atoms, fileobj)
self.failUnlessEqual(info.length, 8)
def test_mdhd_version_1(self, soun=b"soun"):
mdhd = Atom.render(
b"mdhd",
(b"\x01\x00\x00\x00" + b"\x00" * 16 + b"\x00\x00\x00\x02" + b"\x00\x00\x00\x00\x00\x00\x00\x10"), # 2 Hz
)
hdlr = Atom.render(b"hdlr", b"\x00" * 8 + soun)
mdia = Atom.render(b"mdia", mdhd + hdlr)
trak = Atom.render(b"trak", mdia)
moov = Atom.render(b"moov", trak)
fileobj = cBytesIO(moov)
atoms = Atoms(fileobj)
info = MP4Info(atoms, fileobj)
self.failUnlessEqual(info.length, 8)
def test_purl(self):
# purl can have 0 or 1 flags (implicit or utf8)
data = Atom.render(b"data", b"\x00\x00\x00\x01" + b"\x00" * 4 + b"foo")
purl = Atom.render(b"purl", data)
tags = self.wrap_ilst(purl)
self.failUnlessEqual(tags["purl"], ["foo"])
data = Atom.render(b"data", b"\x00\x00\x00\x00" + b"\x00" * 4 + b"foo")
purl = Atom.render(b"purl", data)
tags = self.wrap_ilst(purl)
self.failUnlessEqual(tags["purl"], ["foo"])
# invalid flag
data = Atom.render(b"data", b"\x00\x00\x00\x03" + b"\x00" * 4 + b"foo")
purl = Atom.render(b"purl", data)
tags = self.wrap_ilst(purl)
self.assertFalse("purl" in tags)
self.assertTrue("purl" in tags._failed_atoms)
# invalid utf8
data = Atom.render(
b"data", b"\x00\x00\x00\x01" + b"\x00" * 4 + b"\xff")
purl = Atom.render(b"purl", data)
tags = self.wrap_ilst(purl)
self.assertFalse("purl" in tags)
def test_read(self):
payload = 8 * b"\xff"
fileobj = cBytesIO(b"\x00\x00\x00\x10atom" + payload)
atom = Atom(fileobj)
ok, data = atom.read(fileobj)
self.assertTrue(ok)
self.assertEqual(data, payload)
payload = 7 * b"\xff"
fileobj = cBytesIO(b"\x00\x00\x00\x10atom" + payload)
atom = Atom(fileobj)
ok, data = atom.read(fileobj)
self.assertFalse(ok)
self.assertEqual(data, payload)
def test_length_0_container(self):
data = StringIO(struct.pack(">I4s", 0, "moov") +
Atom.render("data", "whee"))
atom = Atom(data)
self.failUnlessEqual(len(atom.children), 1)
self.failUnlessEqual(atom.length, 20)
self.failUnlessEqual(atom.children[-1].length, 12)
def test_render_too_big(self):
class TooBig(str):
def __len__(self):
return 1L << 32
data = TooBig("test")
try: len(data)
except OverflowError:
# Py_ssize_t is still only 32 bits on this system.
self.failUnlessRaises(OverflowError, Atom.render, "data", data)
else:
data = Atom.render("data", data)
self.failUnlessEqual(len(data), 4 + 4 + 8 + 4)
def test_multiple_tracks(self):
hdlr = Atom.render("hdlr", "\x00" * 8 + "whee")
mdia = Atom.render("mdia", hdlr)
trak1 = Atom.render("trak", mdia)
mdhd = Atom.render("mdhd", ("\x01\x00\x00\x00" + "\x00" * 16 +
"\x00\x00\x00\x02" + # 2 Hz
"\x00\x00\x00\x00\x00\x00\x00\x10"))
hdlr = Atom.render("hdlr", "\x00" * 8 + "soun")
mdia = Atom.render("mdia", mdhd + hdlr)
trak2 = Atom.render("trak", mdia)
moov = Atom.render("moov", trak1 + trak2)
fileobj = StringIO(moov)
atoms = Atoms(fileobj)
info = MP4Info(atoms, fileobj)
self.failUnlessEqual(info.length, 8)
def test_multiple_tracks(self):
hdlr = Atom.render(b"hdlr", b"\x00" * 8 + b"whee")
mdia = Atom.render(b"mdia", hdlr)
trak1 = Atom.render(b"trak", mdia)
mdhd = Atom.render(
b"mdhd",
(b"\x01\x00\x00\x00" + b"\x00" * 16 + b"\x00\x00\x00\x02" + b"\x00\x00\x00\x00\x00\x00\x00\x10"), # 2 Hz
)
hdlr = Atom.render(b"hdlr", b"\x00" * 8 + b"soun")
mdia = Atom.render(b"mdia", mdhd + hdlr)
trak2 = Atom.render(b"trak", mdia)
moov = Atom.render(b"moov", trak1 + trak2)
fileobj = cBytesIO(moov)
atoms = Atoms(fileobj)
info = MP4Info(atoms, fileobj)
self.failUnlessEqual(info.length, 8)
def test_multi_freeform(self):
# merge multiple freeform tags with the same key
mean = Atom.render(b"mean", b"\x00" * 4 + b"net.sacredchao.Mutagen")
name = Atom.render(b"name", b"\x00" * 4 + b"foo")
data = Atom.render(b"data", b"\x00\x00\x00\x01" + b"\x00" * 4 + b"bar")
result = Atom.render(b"----", mean + name + data)
data = Atom.render(b"data", b"\x00\x00\x00\x01" + b"\x00" * 4 + b"quux")
result += Atom.render(b"----", mean + name + data)
tags = self.wrap_ilst(result)
values = tags["----:net.sacredchao.Mutagen:foo"]
self.assertEqual(values[0], b"bar")
self.assertEqual(values[1], b"quux")
def test_freeform_data(self):
# http://code.google.com/p/mutagen/issues/detail?id=103
key = "----:com.apple.iTunes:Encoding Params"
value = ("vers\x00\x00\x00\x01acbf\x00\x00\x00\x01brat\x00\x01\xf4"
"\x00cdcv\x00\x01\x05\x04")
data = ("\x00\x00\x00\x1cmean\x00\x00\x00\x00com.apple.iTunes\x00\x00"
"\x00\x1bname\x00\x00\x00\x00Encoding Params\x00\x00\x000data"
"\x00\x00\x00\x00\x00\x00\x00\x00vers\x00\x00\x00\x01acbf\x00"
"\x00\x00\x01brat\x00\x01\xf4\x00cdcv\x00\x01\x05\x04")
tags = self.wrap_ilst(Atom.render("----", data))
v = tags[key][0]
self.failUnlessEqual(v, value)
self.failUnlessEqual(v.dataformat, MP4FreeForm.FORMAT_DATA)
data = MP4Tags()._MP4Tags__render_freeform(key, v)
v = self.wrap_ilst(data)[key][0]
self.failUnlessEqual(v.dataformat, MP4FreeForm.FORMAT_DATA)
data = MP4Tags()._MP4Tags__render_freeform(key, value)
v = self.wrap_ilst(data)[key][0]
self.failUnlessEqual(v.dataformat, MP4FreeForm.FORMAT_TEXT)
def test_freeform_data(self):
# https://github.com/quodlibet/mutagen/issues/103
key = "----:com.apple.iTunes:Encoding Params"
value = (b"vers\x00\x00\x00\x01acbf\x00\x00\x00\x01brat\x00\x01\xf4"
b"\x00cdcv\x00\x01\x05\x04")
data = (b"\x00\x00\x00\x1cmean\x00\x00\x00\x00com.apple.iTunes\x00\x00"
b"\x00\x1bname\x00\x00\x00\x00Encoding Params\x00\x00\x000data"
b"\x00\x00\x00\x00\x00\x00\x00\x00vers\x00\x00\x00\x01acbf\x00"
b"\x00\x00\x01brat\x00\x01\xf4\x00cdcv\x00\x01\x05\x04")
tags = self.wrap_ilst(Atom.render(b"----", data))
v = tags[key][0]
self.failUnlessEqual(v, value)
self.failUnlessEqual(v.dataformat, AtomDataType.IMPLICIT)
data = MP4Tags()._MP4Tags__render_freeform(key, v)
v = self.wrap_ilst(data)[key][0]
self.failUnlessEqual(v.dataformat, AtomDataType.IMPLICIT)
data = MP4Tags()._MP4Tags__render_freeform(key, value)
v = self.wrap_ilst(data)[key][0]
self.failUnlessEqual(v.dataformat, AtomDataType.UTF8)
def test_genre(self):
data = Atom.render("data", "\x00" * 8 + "\x00\x01")
genre = Atom.render("gnre", data)
tags = self.wrap_ilst(genre)
self.failIf("gnre" in tags)
self.failUnlessEqual(tags["\xa9gen"], ["Blues"])
def test_extra_trailing_data(self):
data = StringIO(Atom.render("data", "whee") + "\x00\x00")
self.failUnless(Atoms(data))
def test_strips_unknown_types(self):
data = Atom.render("data", "\x00" * 8 + "whee")
foob = Atom.render("foob", data)
tags = self.wrap_ilst(foob)
self.failIf(tags)
def test_empty_cpil(self):
cpil = Atom.render("cpil", Atom.render("data", "\x00" * 8))
tags = self.wrap_ilst(cpil)
self.failUnless("cpil" in tags)
self.failIf(tags["cpil"])
def wrap_ilst(self, data):
ilst = Atom.render("ilst", data)
meta = Atom.render("meta", "\x00" * 4 + ilst)
data = Atom.render("moov", Atom.render("udta", meta))
fileobj = StringIO(data)
return MP4Tags(Atoms(fileobj), fileobj)
def test_bad_freeform(self):
mean = Atom.render(b"mean", b"net.sacredchao.Mutagen")
name = Atom.render(b"name", b"empty test key")
bad_freeform = Atom.render(b"----", b"\x00" * 4 + mean + name)
tags = self.wrap_ilst(bad_freeform)
self.assertFalse(tags)
def test_bad_freeform(self):
mean = Atom.render("mean", "net.sacredchao.Mutagen")
name = Atom.render("name", "empty test key")
bad_freeform = Atom.render("----", "\x00" * 4 + mean + name)
self.failUnlessRaises(MP4MetadataError, self.wrap_ilst, bad_freeform)
def test_extra_trailing_data(self):
data = cBytesIO(Atom.render(b"data", b"whee") + b"\x00\x00")
self.failUnless(Atoms(data))
def test_empty_cpil(self):
cpil = Atom.render(b"cpil", Atom.render(b"data", b"\x00" * 8))
tags = self.wrap_ilst(cpil)
self.assertFalse("cpil" in tags)
def test_bad_freeform(self):
mean = Atom.render(b"mean", b"net.sacredchao.Mutagen")
name = Atom.render(b"name", b"empty test key")
bad_freeform = Atom.render(b"----", b"\x00" * 4 + mean + name)
tags = self.wrap_ilst(bad_freeform)
self.assertFalse(tags)
def test_bad_cprt(self):
data = Atom.render(b"cprt", b"\x00\x00\x00#data\x00")
tags = self.wrap_ilst(data)
self.assertFalse(tags)
def test_bad_text_data(self):
data = Atom.render(b"datA", b"\x00\x00\x00\x01\x00\x00\x00\x00whee")
data = Atom.render(b"aART", data)
tags = self.wrap_ilst(data)
self.assertFalse(tags)
def test_genre_too_big(self):
data = Atom.render("data", "\x00" * 8 + "\x01\x00")
genre = Atom.render("gnre", data)
tags = self.wrap_ilst(genre)
self.failIf("gnre" in tags)
self.failIf("\xa9gen" in tags)
def test_error(self):
fileobj = cBytesIO(b"\x00" * 20)
atom = Atom(fileobj)
self.assertRaises(ASEntryError, AudioSampleEntry, atom, fileobj)
def test_bad_covr(self):
data = Atom.render("foob", "\x00\x00\x00\x0E" + "\x00" * 4 + "whee")
covr = Atom.render("covr", data)
self.failUnlessRaises(MP4MetadataError, self.wrap_ilst, covr)
def test_no_tracks(self):
moov = Atom.render(b"moov", b"")
fileobj = cBytesIO(moov)
atoms = Atoms(fileobj)
with self.assertRaises(MP4StreamInfoError):
MP4Info(atoms, fileobj)
def test_strips_bad_unknown_types(self):
data = Atom.render(b"datA", b"\x00" * 8 + b"whee")
foob = Atom.render(b"foob", data)
tags = self.wrap_ilst(foob)
self.failIf(tags)
def test_no_children(self):
fileobj = cBytesIO(b"\x00\x00\x00\x08atom")
atom = Atom(fileobj)
self.failUnlessRaises(KeyError, atom.__getitem__, "test")
def test_length_0(self):
fileobj = cBytesIO(b"\x00\x00\x00\x00atom" + 40 * b"\x00")
atom = Atom(fileobj)
self.failUnlessEqual(fileobj.tell(), 48)
self.failUnlessEqual(atom.length, 48)
self.failUnlessEqual(atom.datalength, 40)
def wrap_ilst(self, data):
ilst = Atom.render(b"ilst", data)
meta = Atom.render(b"meta", b"\x00" * 4 + ilst)
data = Atom.render(b"moov", Atom.render(b"udta", meta))
fileobj = cBytesIO(data)
return MP4Tags(Atoms(fileobj), fileobj)
def test_genre(self):
data = Atom.render("data", "\x00" * 8 + "\x00\x01")
genre = Atom.render("gnre", data)
tags = self.wrap_ilst(genre)
self.failIf("gnre" in tags)
self.failUnlessEqual(tags["\xa9gen"], ["Blues"])
def test_length_1(self):
fileobj = cBytesIO(b"\x00\x00\x00\x01atom"
b"\x00\x00\x00\x00\x00\x00\x00\x10" + b"\x00" * 16)
atom = Atom(fileobj)
self.failUnlessEqual(atom.length, 16)
self.failUnlessEqual(atom.datalength, 0)
def test_genre_too_big(self):
data = Atom.render("data", "\x00" * 8 + "\x01\x00")
genre = Atom.render("gnre", data)
tags = self.wrap_ilst(genre)
self.failIf("gnre" in tags)
self.failIf("\xa9gen" in tags)
def test_bad_text_data(self):
data = Atom.render("datA", "\x00\x00\x00\x01\x00\x00\x00\x00whee")
data = Atom.render("aART", data)
self.failUnlessRaises(MP4MetadataError, self.wrap_ilst, data)
def test_bad_covr(self):
data = Atom.render("foob", "\x00\x00\x00\x0E" + "\x00" * 4 + "whee")
covr = Atom.render("covr", data)
self.failUnlessRaises(MP4MetadataError, self.wrap_ilst, covr)
def test_bad_freeform(self):
mean = Atom.render("mean", "net.sacredchao.Mutagen")
name = Atom.render("name", "empty test key")
bad_freeform = Atom.render("----", "\x00" * 4 + mean + name)
self.failUnlessRaises(MP4MetadataError, self.wrap_ilst, bad_freeform)
def wrap_ilst(self, data):
ilst = Atom.render("ilst", data)
meta = Atom.render("meta", "\x00" * 4 + ilst)
data = Atom.render("moov", Atom.render("udta", meta))
fileobj = StringIO(data)
return MP4Tags(Atoms(fileobj), fileobj)
def test_extra_trailing_data(self):
data = StringIO(Atom.render("data", "whee") + "\x00\x00")
self.failUnless(Atoms(data))
def test_empty_cpil(self):
cpil = Atom.render("cpil", Atom.render("data", "\x00" * 8))
tags = self.wrap_ilst(cpil)
self.failUnless("cpil" in tags)
self.failIf(tags["cpil"])
def test_length_0(self):
fileobj = StringIO("\x00\x00\x00\x00atom")
Atom(fileobj)
self.failUnlessEqual(fileobj.tell(), 8)
def test_strips_unknown_types(self):
data = Atom.render("data", "\x00" * 8 + "whee")
foob = Atom.render("foob", data)
tags = self.wrap_ilst(foob)
self.failIf(tags)
def test_bad_text_data(self):
data = Atom.render(b"datA", b"\x00\x00\x00\x01\x00\x00\x00\x00whee")
data = Atom.render(b"aART", data)
tags = self.wrap_ilst(data)
self.assertFalse(tags)
def test_bad_text_data(self):
data = Atom.render("datA", "\x00\x00\x00\x01\x00\x00\x00\x00whee")
data = Atom.render("aART", data)
self.failUnlessRaises(MP4MetadataError, self.wrap_ilst, data)
def test_bad_cprt(self):
data = Atom.render(b"cprt", b"\x00\x00\x00#data\x00")
tags = self.wrap_ilst(data)
self.assertFalse(tags)
def test_strips_bad_unknown_types(self):
data = Atom.render(b"datA", b"\x00" * 8 + b"whee")
foob = Atom.render(b"foob", data)
tags = self.wrap_ilst(foob)
self.failIf(tags)
def test_bad_covr(self):
data = Atom.render(
b"foob", b"\x00\x00\x00\x0E" + b"\x00" * 4 + b"whee")
covr = Atom.render(b"covr", data)
tags = self.wrap_ilst(covr)
self.assertFalse(tags)
def test_bad_covr(self):
data = Atom.render(b"foob",
b"\x00\x00\x00\x0E" + b"\x00" * 4 + b"whee")
covr = Atom.render(b"covr", data)
tags = self.wrap_ilst(covr)
self.assertFalse(tags)
def test_covr_blank_format(self):
data = Atom.render(b"data",
b"\x00\x00\x00\x00" + b"\x00" * 4 + b"whee")
covr = Atom.render(b"covr", data)
tags = self.wrap_ilst(covr)
self.failUnlessEqual(MP4Cover.FORMAT_JPEG, tags["covr"][0].imageformat)
def test_empty_cpil(self):
cpil = Atom.render(b"cpil", Atom.render(b"data", b"\x00" * 8))
tags = self.wrap_ilst(cpil)
self.assertFalse("cpil" in tags)
def test_length_1(self):
fileobj = StringIO("\x00\x00\x00\x01atom"
"\x00\x00\x00\x00\x00\x00\x00\x08" + "\x00" * 8)
self.failUnlessEqual(Atom(fileobj).length, 8)
