symuid/tests/tag_interface/test_ogg_vorbis.py

import os

import mutagen
import pytest

from symuid._tag_interface import OggOpus

# 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 = OggOpus(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 = OggOpus(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 = OggOpus(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 = OggOpus(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 = OggOpus(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 = OggOpus(mutagen.File(os.path.join(tracks_dir_path, track_name)))
    assert expected_uuid == iface.get_track_uuid()