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
|
# This file is part of beets.
# Copyright 2021, Edgars Supe.
#
# Permission is hereby granted, free of charge, to any person obtaining
# a copy of this software and associated documentation files (the
# "Software"), to deal in the Software without restriction, including
# without limitation the rights to use, copy, modify, merge, publish,
# distribute, sublicense, and/or sell copies of the Software, and to
# permit persons to whom the Software is furnished to do so, subject to
# the following conditions:
#
# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Software.
"""Tests for the 'albumtypes' plugin."""
from collections.abc import Sequence
from beets.test.helper import PluginTestCase
from beetsplug.albumtypes import AlbumTypesPlugin
from beetsplug.musicbrainz import VARIOUS_ARTISTS_ID
class AlbumTypesPluginTest(PluginTestCase):
"""Tests for albumtypes plugin."""
plugin = "albumtypes"
def test_renames_types(self):
"""Tests if the plugin correctly renames the specified types."""
self._set_config(
types=[("ep", "EP"), ("remix", "Remix")], ignore_va=[], bracket="()"
)
album = self._create_album(album_types=["ep", "remix"])
subject = AlbumTypesPlugin()
result = subject._atypes(album)
assert "(EP)(Remix)" == result
return
def test_returns_only_specified_types(self):
"""Tests if the plugin returns only non-blank types given in config."""
self._set_config(
types=[("ep", "EP"), ("soundtrack", "")], ignore_va=[], bracket="()"
)
album = self._create_album(album_types=["ep", "remix", "soundtrack"])
subject = AlbumTypesPlugin()
result = subject._atypes(album)
assert "(EP)" == result
def test_respects_type_order(self):
"""Tests if the types are returned in the same order as config."""
self._set_config(
types=[("remix", "Remix"), ("ep", "EP")], ignore_va=[], bracket="()"
)
album = self._create_album(album_types=["ep", "remix"])
subject = AlbumTypesPlugin()
result = subject._atypes(album)
assert "(Remix)(EP)" == result
return
def test_ignores_va(self):
"""Tests if the specified type is ignored for VA albums."""
self._set_config(
types=[("ep", "EP"), ("soundtrack", "OST")],
ignore_va=["ep"],
bracket="()",
)
album = self._create_album(
album_types=["ep", "soundtrack"], artist_id=VARIOUS_ARTISTS_ID
)
subject = AlbumTypesPlugin()
result = subject._atypes(album)
assert "(OST)" == result
def test_respects_defaults(self):
"""Tests if the plugin uses the default values if config not given."""
album = self._create_album(
album_types=[
"ep",
"single",
"soundtrack",
"live",
"compilation",
"remix",
],
artist_id=VARIOUS_ARTISTS_ID,
)
subject = AlbumTypesPlugin()
result = subject._atypes(album)
assert "[EP][Single][OST][Live][Remix]" == result
def _set_config(
self,
types: Sequence[tuple[str, str]],
ignore_va: Sequence[str],
bracket: str,
):
self.config["albumtypes"]["types"] = types
self.config["albumtypes"]["ignore_va"] = ignore_va
self.config["albumtypes"]["bracket"] = bracket
def _create_album(self, album_types: Sequence[str], artist_id: str = "0"):
return self.add_album(
albumtypes=album_types, mb_albumartistid=artist_id
)
|
