symuid/tests/tag_interface/test_mp4.py

import mutagen
import pytest

from symuid._tag_interface import MP4

# pylint: disable=protected-access


def test_set_comment(empty_mp4_path):
    iface = MP4(mutagen.File(empty_mp4_path))
    assert iface.get_comment() is None
    iface.set_comment('latin')
    assert iface.get_comment() == 'latin'
    iface.set_comment('mp4 你好')
    assert iface.get_comment() == 'mp4 你好'
    iface.save()
    iface_reread = MP4(mutagen.File(empty_mp4_path))
    assert iface_reread.get_comment() == 'mp4 你好'
    tags = mutagen.File(iface.track_path).tags
    assert len(tags) == 1
    assert tags.items()[0] == ('©cmt', ['mp4 你好'])


def test__get_free_uuid(empty_mp4_path):
    uuid = b'h\x97\x8c_1?B\t\x9d\xa3$\xdf\xd0Y\xa1\xc2'
    mutagen_file = mutagen.File(empty_mp4_path)
    mutagen_file['----:foo:bar'] = mutagen.mp4.MP4FreeForm(
        dataformat=mutagen.mp4.AtomDataType.UUID, data=uuid)
    mutagen_file.save()
    mp4_iface = MP4(mutagen.File(empty_mp4_path))
    assert mp4_iface._get_free_uuid('foo:bar') == uuid


def test_get_track_uuid(empty_mp4_path):
    uuid = b'h\x97\x8c_1?B\t\x9d\xa3$\xdf\xd0Y\xa1\xc2'
    mutagen_file = mutagen.File(empty_mp4_path)
    mutagen_file['----:symuid:uuid'] = mutagen.mp4.MP4FreeForm(
        dataformat=mutagen.mp4.AtomDataType.UUID, data=uuid)
    mutagen_file.save()
    mp4_iface = MP4(mutagen.File(empty_mp4_path))
    assert mp4_iface.get_track_uuid() == uuid


@pytest.mark.parametrize(('nominator', 'denominator'), [
    (21, 42),
    (-21, 42),
    (21, -42),
    (-21, -42),
    (0, 42),
])
def test_get_free_int_ratio(empty_mp4_path, nominator, denominator):
    mutagen_file = mutagen.File(empty_mp4_path)
    mutagen_file.tags['----:foo:bar'] = [
        mutagen.mp4.MP4FreeForm(dataformat=mutagen.mp4.AtomDataType.INTEGER,
                                data=i.to_bytes(4, byteorder='big', signed=True))
        for i in (nominator, denominator)]
    mutagen_file.save()
    mp4_iface = MP4(mutagen.File(empty_mp4_path))
    assert mp4_iface.get_free_int_ratio('foo:bar') == (nominator, denominator)


@pytest.mark.parametrize(('nominator', 'denominator', 'expected_tag_data'), [
    (21, 42, [b'\x15', b'\x2a']),
    (-21, 42, [b'\xeb', b'\x2a']),
    (21, -42, [b'\x15', b'\xd6']),
    (-21, -42, [b'\xeb', b'\xd6']),
    (0, 42, [b'\x00', b'\x2a']),
])
def test_set_free_int_ratio(empty_mp4_path, nominator, denominator, expected_tag_data):
    mp4_iface = MP4(mutagen.File(empty_mp4_path))
    mp4_iface.set_free_int_ratio('test:some-ratio', nominator, denominator)
    mp4_iface.save()
    mutagen_file = mutagen.File(empty_mp4_path)
    assert len(mutagen_file.tags) == 1
    tag = mutagen_file.tags['----:test:some-ratio']
    assert len(tag) == 2
    assert all(f.dataformat == mutagen.mp4.AtomDataType.INTEGER for f in tag)
    assert [bytes(f) for f in tag] == expected_tag_data


@pytest.mark.parametrize(('integer', 'expected_tag_data'), [
    (0, b'\x00'),
    (4, b'\x04'),
    (-1, b'\xff'),
    (-2, b'\xfe'),
    (2**6, b'\x40'),
    (2**7-1, b'\x7f'),
    (2**7, b'\x00\x80'),
    (2**8, b'\x01\x00'),
    (2**24, b'\x01\x00\x00\x00'),
    (2**31-1, b'\x7f\xff\xff\xff'),
])
def test_set_free_int(empty_mp4_path, integer, expected_tag_data):
    mp4_iface = MP4(mutagen.File(empty_mp4_path))
    mp4_iface.set_free_int('foo:bar', integer)
    mp4_iface.save()
    mutagen_file = mutagen.File(empty_mp4_path)
    assert len(mutagen_file.tags) == 1
    tag, = mutagen_file.get('----:foo:bar')
    assert tag.dataformat == mutagen.mp4.AtomDataType.INTEGER
    assert bytes(tag) == expected_tag_data
    assert MP4._freeform_to_int(tag) == integer