File: test_metadata.py

package info (click to toggle)
quodlibet 4.7.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 19,228 kB
  • sloc: python: 89,728; sh: 381; xml: 110; makefile: 91
file content (142 lines) | stat: -rw-r--r-- 4,134 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.

from tests import mkstemp, TestCase, get_data_path

import os

from quodlibet import formats
from quodlibet import config

from shutil import copyfileobj


class TestMetaDataBase(TestCase):
    base = get_data_path("silence-44-s")

    def setUp(self):
        """Copy the base silent file to a temp name/location and load it"""

        config.init()
        fd, self.filename = mkstemp(suffix=self.ext, text=False)
        dst = os.fdopen(fd, "wb")
        src = open(self.base + self.ext, "rb")
        copyfileobj(src, dst)
        dst.close()
        self.song = formats.MusicFile(self.filename)

    def tearDown(self):
        """Delete the temp file"""

        os.remove(self.filename)
        del self.filename
        del self.song
        config.quit()


class _TestMetaDataMixin:
    def test_base_data(self):
        self.assertEqual(self.song["artist"], "piman\njzig")
        self.assertEqual(self.song["album"], "Quod Libet Test Data")
        self.assertEqual(self.song["title"], "Silence")

    def test_mutability(self):
        assert not self.song.can_change("=foo")
        assert not self.song.can_change("foo~bar")
        assert self.song.can_change("artist")
        assert self.song.can_change("title")
        assert self.song.can_change("tracknumber")
        assert self.song.can_change("somebadtag")
        assert self.song.can_change("some%punctuated:tag.")

    def _test_tag(self, tag, values, remove=True):
        assert self.song.can_change(tag)
        for value in values:
            self.song[tag] = value
            self.song.write()
            written = formats.MusicFile(self.filename)
            self.assertEqual(written[tag], value)
            if remove:
                del self.song[tag]
                self.song.write()
                deleted = formats.MusicFile(self.filename)
                assert tag not in deleted

    def test_artist(self):  # a normalish tag
        self._test_tag("artist", ["me", "you\nme", "\u6d5c\u5d0e\u3042\u3086\u307f"])

    def test_date(self):  # unusual special handling for mp3s
        self._test_tag("date", ["2004", "2005", "2005-06-12"], False)

    def test_genre(self):  # unusual special handling for mp3s
        self._test_tag(
            "genre",
            [
                "Pop",
                "Rock\nClassical",
                "Big Bird",
                "\u30a2\u30cb\u30e1\u30b5\u30f3\u30c8\u30e9",
            ],
        )

    def test_odd_performer(self):
        values = ["A Person", "Another"]
        self._test_tag("performer:vocals", values)
        self._test_tag("performer:guitar", values)

    def test_wackjob(self):  # undefined tag
        self._test_tag(
            "wackjob",
            ["Jelly\nDanish", "Muppet", "\u30cf\u30f3\u30d0\u30fc\u30ac\u30fc"],
        )


tags = [
    "album",
    "arranger",
    "artist",
    "author",
    "comment",
    "composer",
    "conductor",
    "copyright",
    "discnumber",
    "encodedby",
    "genre",
    "isrc",
    "language",
    "license",
    "lyricist",
    "organization",
    "performer",
    "title",
    "tracknumber",
    "version",
    "xyzzy_undefined_tag",
    "musicbrainz_trackid",
    "releasecountry",
]

for ext in formats.loaders.keys():
    if os.path.exists(TestMetaDataBase.base + ext):
        extra_tests = {}
        for tag in tags:
            if tag in ["artist", "date", "genre"]:
                continue

            def _test_tag(self, tag=tag):
                self._test_tag(tag, ["a"])

            extra_tests["test_tag_" + tag] = _test_tag

            def _test_tags(self, tag=tag):
                self._test_tag(tag, ["b\nc"])

            extra_tests["test_tags_" + tag] = _test_tags

        name = "MetaData" + ext
        testcase = type(name, (TestMetaDataBase, _TestMetaDataMixin), extra_tests)
        testcase.ext = ext
        globals()[name] = testcase