mirror of
https://github.com/beetbox/beets.git
synced 2025-12-10 02:22:25 +01:00
91 lines
3.3 KiB
Python
91 lines
3.3 KiB
Python
# This file is part of beets.
|
|
# Copyright 2025, Henry Oberholtzer
|
|
#
|
|
# 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 'titlecase' plugin"""
|
|
|
|
import pytest
|
|
|
|
from beets import config
|
|
from beets.test.helper import PluginTestCase
|
|
from beetsplug.titlecase import TitlecasePlugin, EXCLUDED_INFO_FIELDS
|
|
|
|
@pytest.mark.parametrize("given, expected",
|
|
[("a", "A"),
|
|
("PENDULUM", "Pendulum"),
|
|
("Aaron-carl", "Aaron-Carl"),
|
|
("LTJ bukem", "LTJ Bukem"),
|
|
("Freaky chakra Vs. Single Cell orchestra",
|
|
"Freaky Chakra vs. Single Cell Orchestra"),
|
|
("(original mix)", "(Original Mix)"),
|
|
("ALL CAPS TITLE", "All Caps Title")
|
|
])
|
|
def test_basic_titlecase(given, expected):
|
|
""" Assert that general behavior is as expected. """
|
|
assert TitlecasePlugin().titlecase(given) == expected
|
|
|
|
|
|
class TitlecasePluginTest(PluginTestCase):
|
|
plugin = "titlecase"
|
|
preload_plugin = False
|
|
|
|
def test_preserved_case(self):
|
|
""" Test using given strings to preserve case """
|
|
names_to_preserve = ["easyFun", "A.D.O.R.", "D.R.", "ABBA", "LaTeX"]
|
|
with self.configure_plugin({
|
|
"preserve": names_to_preserve}):
|
|
config["titlecase"]["preserve"] = names_to_preserve
|
|
for name in names_to_preserve:
|
|
assert TitlecasePlugin().titlecase(
|
|
name.lower()) == name
|
|
|
|
def test_small_first_last(self):
|
|
with self.configure_plugin({
|
|
"small_first_last": False}):
|
|
assert TitlecasePlugin().titlecase(
|
|
"A Simple Trial") == "a Simple Trial"
|
|
with self.configure_plugin({
|
|
"small_first_last": True}):
|
|
assert TitlecasePlugin().titlecase(
|
|
"A simple Trial") == "A Simple Trial"
|
|
|
|
def test_ui_command(self):
|
|
assert 1 == 3
|
|
|
|
def test_imported(self):
|
|
item = self.add_item(
|
|
artist="A poorly cased artist",
|
|
albumartist="not vEry good tItle caSE",
|
|
mb_artistid="case sensitive field")
|
|
assert item.artist == "A Poorly Cased Artist"
|
|
assert item.albumartist == "Not Very Good Title Case"
|
|
|
|
def test_field_list_default_excluded(self):
|
|
excluded = list(EXCLUDED_INFO_FIELDS)
|
|
config["titlecase"]["include_fields"] = excluded
|
|
t = TitlecasePlugin()
|
|
for field in excluded:
|
|
assert field not in t.fields_to_process
|
|
|
|
def test_field_list_included(self):
|
|
config["titlecase"]["include_fields"] = ["album", "albumartist"]
|
|
t = TitlecasePlugin()
|
|
t.fields_to_process == ["album", "albumartist"]
|
|
|
|
def test_field_list_exclude(self):
|
|
excluded = ["album", "albumartist"]
|
|
config["titlecase"]["exclude_fields"] = excluded
|
|
t = TitlecasePlugin()
|
|
for field in excluded:
|
|
assert field not in t.fields_to_process
|
|
|