mirror of
https://github.com/beetbox/beets.git
synced 2025-12-06 16:42:42 +01:00
67 lines
2.7 KiB
Python
67 lines
2.7 KiB
Python
"""Tests for the 'the' plugin"""
|
|
|
|
|
|
import unittest
|
|
from test import _common
|
|
from beets import config
|
|
from beetsplug.the import ThePlugin, PATTERN_A, PATTERN_THE, FORMAT
|
|
|
|
|
|
class ThePluginTest(_common.TestCase):
|
|
|
|
def test_unthe_with_default_patterns(self):
|
|
self.assertEqual(ThePlugin().unthe('', PATTERN_THE), '')
|
|
self.assertEqual(ThePlugin().unthe('The Something', PATTERN_THE),
|
|
'Something, The')
|
|
self.assertEqual(ThePlugin().unthe('The The', PATTERN_THE),
|
|
'The, The')
|
|
self.assertEqual(ThePlugin().unthe('The The', PATTERN_THE),
|
|
'The, The')
|
|
self.assertEqual(ThePlugin().unthe('The The X', PATTERN_THE),
|
|
'The X, The')
|
|
self.assertEqual(ThePlugin().unthe('the The', PATTERN_THE),
|
|
'The, the')
|
|
self.assertEqual(ThePlugin().unthe('Protected The', PATTERN_THE),
|
|
'Protected The')
|
|
self.assertEqual(ThePlugin().unthe('A Boy', PATTERN_A),
|
|
'Boy, A')
|
|
self.assertEqual(ThePlugin().unthe('a girl', PATTERN_A),
|
|
'girl, a')
|
|
self.assertEqual(ThePlugin().unthe('An Apple', PATTERN_A),
|
|
'Apple, An')
|
|
self.assertEqual(ThePlugin().unthe('An A Thing', PATTERN_A),
|
|
'A Thing, An')
|
|
self.assertEqual(ThePlugin().unthe('the An Arse', PATTERN_A),
|
|
'the An Arse')
|
|
self.assertEqual(ThePlugin().unthe('TET - Travailleur', PATTERN_THE),
|
|
'TET - Travailleur')
|
|
|
|
def test_unthe_with_strip(self):
|
|
config['the']['strip'] = True
|
|
self.assertEqual(ThePlugin().unthe('The Something', PATTERN_THE),
|
|
'Something')
|
|
self.assertEqual(ThePlugin().unthe('An A', PATTERN_A), 'A')
|
|
|
|
def test_template_function_with_defaults(self):
|
|
ThePlugin().patterns = [PATTERN_THE, PATTERN_A]
|
|
self.assertEqual(ThePlugin().the_template_func('The The'),
|
|
'The, The')
|
|
self.assertEqual(ThePlugin().the_template_func('An A'), 'A, An')
|
|
|
|
def test_custom_pattern(self):
|
|
config['the']['patterns'] = ['^test\\s']
|
|
config['the']['format'] = FORMAT
|
|
self.assertEqual(ThePlugin().the_template_func('test passed'),
|
|
'passed, test')
|
|
|
|
def test_custom_format(self):
|
|
config['the']['patterns'] = [PATTERN_THE, PATTERN_A]
|
|
config['the']['format'] = '{1} ({0})'
|
|
self.assertEqual(ThePlugin().the_template_func('The A'), 'The (A)')
|
|
|
|
|
|
def suite():
|
|
return unittest.TestLoader().loadTestsFromName(__name__)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(defaultTest='suite')
|