symuid/tests/tag_interface/test_ogg_opus.py

import os

import mutagen
import pytest
from symuid._tag_interface import Ogg

# pylint: disable=protected-access


@pytest.mark.parametrize('track_name', ['ogg-opus-empty.opus'])
def test_get_track_path(tracks_dir_path, track_name):
    track_path = os.path.join(tracks_dir_path, track_name)
    iface = Ogg(mutagen.File(track_path))
    assert track_path == iface.track_path


@pytest.mark.parametrize(('track_name', 'tag_label', 'expected_text'), [
    ('ogg-opus-empty.opus', 'artist', None),
    ('ogg-opus-typical.opus', 'artist', 'some artist'),
    ('ogg-opus-typical.opus', 'comment', 'some comment'),
    ('ogg-opus-typical.opus', 'com', None),
    ('ogg-opus-typical.opus', 'symuid:uuid',
     '613ea4ac-a4cf-4026-8e99-1904b2bb5cd0'),
])
def test__get_single_text(tracks_dir_path, track_name, tag_label, expected_text):
    iface = Ogg(mutagen.File(os.path.join(tracks_dir_path, track_name)))
    assert expected_text == iface._get_single_text(tag_label)


@pytest.mark.parametrize(('track_name', 'expected_comment'), [
    ('ogg-opus-empty.opus', None),
    ('ogg-opus-typical.opus', 'some comment'),
])
def test_get_comment(tracks_dir_path, track_name, expected_comment):
    iface = Ogg(mutagen.File(os.path.join(tracks_dir_path, track_name)))
    assert expected_comment == iface.get_comment()


def test_set_comment(empty_ogg_opus_path):
    iface = Ogg(mutagen.File(empty_ogg_opus_path))
    assert iface.get_comment() is None
    iface.set_comment('latin')
    assert iface.get_comment() == 'latin'
    iface.set_comment('你好')
    assert iface.get_comment() == '你好'
    iface.save()
    iface_reread = Ogg(mutagen.File(empty_ogg_opus_path))
    assert iface_reread.get_comment() == '你好'
    tags = mutagen.File(iface.track_path).tags
    assert len(tags) == 1
    assert tags.items()[0] == ('comment', ['你好'])


@pytest.mark.parametrize(('track_name', 'expected_uuid'), [
    ('ogg-opus-empty.opus', None),
    ('ogg-opus-typical.opus', b'a>\xa4\xac\xa4\xcf@&\x8e\x99\x19\x04\xb2\xbb\\\xd0'),
])
def test_get_track_uuid(tracks_dir_path, track_name, expected_uuid):
    iface = Ogg(mutagen.File(os.path.join(tracks_dir_path, track_name)))
    assert expected_uuid == iface.get_track_uuid()


def test_set_track_uuid(empty_ogg_opus_path):
    iface = Ogg(mutagen.File(empty_ogg_opus_path))
    assert iface.get_track_uuid() is None
    uuid_a = b'\x9e\xa7\xc4\xf0\xda\xecE\xb7\xab\x9a\xba\x9f\xc6\xaa\xc0S'
    iface.set_track_uuid(uuid_a)
    assert iface.get_track_uuid() == uuid_a
    uuid_b = b'_\xa0\xd7\xc0\xf3\x15F\x14\xbe\xe4idM=\x80\xb3'
    iface.set_track_uuid(uuid_b)
    assert iface.get_track_uuid() == uuid_b
    iface.save()
    iface_reread = Ogg(mutagen.File(empty_ogg_opus_path))
    assert iface_reread.get_track_uuid() == uuid_b
    tags = mutagen.File(iface.track_path).tags
    assert len(tags) == 1
    assert tags.items()[0] == (
        'symuid:uuid', ['5fa0d7c0-f315-4614-bee4-69644d3d80b3'])


@pytest.mark.parametrize(('track_name', 'tag_label', 'expected_int'), [
    ('ogg-opus-empty.opus', 'tracknumber', None),
    ('ogg-opus-typical.opus', 'tracknumber', 21),
    ('ogg-opus-typical.opus', 'tracknumberr', None),
    ('ogg-opus-typical.opus', 'symuid:pcnt:player:library:1572098177', 43),
])
def test_get_free_int(tracks_dir_path, track_name, tag_label, expected_int):
    iface = Ogg(mutagen.File(os.path.join(tracks_dir_path, track_name)))
    assert expected_int == iface.get_free_int(tag_label)


def test_get_free_ints(tracks_dir_path):
    iface = Ogg(mutagen.File(os.path.join(
        tracks_dir_path, 'ogg-opus-typical.opus')))
    assert not list(iface.get_free_ints('symuid:none'))
    assert list(iface.get_free_ints('tracknumber')) == [('tracknumber', 21)]
    assert not list(iface.get_free_ints('tracknum'))
    pcnt_tags = {('symuid:pcnt:player:library:1572098158', 42),
                 ('symuid:pcnt:player:library:1572098177', 43)}
    assert set(iface.get_free_ints('symuid:pcnt:player:library')) == pcnt_tags
    assert set(iface.get_free_ints('symuid:pcnt:player')) == pcnt_tags
    assert set(iface.get_free_ints('symuid:pcnt')) == pcnt_tags
    assert set(iface.get_free_ints('symuid:pcnt:player:library:1572098158')) \
        == {('symuid:pcnt:player:library:1572098158', 42)}


def test_set_free_int(empty_ogg_opus_path):
    iface = Ogg(mutagen.File(empty_ogg_opus_path))
    assert iface.get_free_int('tracknumber') is None
    iface.set_free_int('tracknumber', 7)
    assert iface.get_free_int('tracknumber') == 7
    iface.set_free_int('tracknumber', 14)
    assert iface.get_free_int('tracknumber') == 14
    iface.save()
    iface_reread = Ogg(mutagen.File(empty_ogg_opus_path))
    assert iface_reread.get_free_int('tracknumber') == 14
    tags = mutagen.File(iface.track_path).tags
    assert len(tags) == 1
    assert tags.items()[0] == ('tracknumber', ['14'])