mirror of
https://github.com/beetbox/beets.git
synced 2025-12-16 21:54:07 +01:00
90b388b775
Include import of __future__ features division, absolute_imports and print_function everywhere. Don't add unicode_literals yet for it is harder to convert. Goal is smoothing the transition to python 3.
109 lines
3.1 KiB
Python
109 lines
3.1 KiB
Python
"""Tests for the 'zero' plugin"""
|
|
|
|
from __future__ import division, absolute_import, print_function
|
|
|
|
from test._common import unittest
|
|
from test.helper import TestHelper
|
|
|
|
from beets.library import Item
|
|
from beets import config
|
|
from beetsplug.zero import ZeroPlugin
|
|
from beets.mediafile import MediaFile
|
|
|
|
|
|
class ZeroPluginTest(unittest.TestCase, TestHelper):
|
|
def setUp(self):
|
|
self.setup_beets()
|
|
|
|
def tearDown(self):
|
|
self.teardown_beets()
|
|
self.unload_plugins()
|
|
|
|
def test_no_patterns(self):
|
|
tags = {
|
|
'comments': 'test comment',
|
|
'day': 13,
|
|
'month': 3,
|
|
'year': 2012,
|
|
}
|
|
z = ZeroPlugin()
|
|
z.debug = False
|
|
z.fields = ['comments', 'month', 'day']
|
|
z.patterns = {'comments': ['.'],
|
|
'month': ['.'],
|
|
'day': ['.']}
|
|
z.write_event(None, None, tags)
|
|
self.assertEqual(tags['comments'], None)
|
|
self.assertEqual(tags['day'], None)
|
|
self.assertEqual(tags['month'], None)
|
|
self.assertEqual(tags['year'], 2012)
|
|
|
|
def test_patterns(self):
|
|
z = ZeroPlugin()
|
|
z.debug = False
|
|
z.fields = ['comments', 'year']
|
|
z.patterns = {'comments': 'eac lame'.split(),
|
|
'year': '2098 2099'.split()}
|
|
|
|
tags = {
|
|
'comments': 'from lame collection, ripped by eac',
|
|
'year': 2012,
|
|
}
|
|
z.write_event(None, None, tags)
|
|
self.assertEqual(tags['comments'], None)
|
|
self.assertEqual(tags['year'], 2012)
|
|
|
|
def test_delete_replaygain_tag(self):
|
|
path = self.create_mediafile_fixture()
|
|
item = Item.from_path(path)
|
|
item.rg_track_peak = 0.0
|
|
item.write()
|
|
|
|
mediafile = MediaFile(item.path)
|
|
self.assertIsNotNone(mediafile.rg_track_peak)
|
|
self.assertIsNotNone(mediafile.rg_track_gain)
|
|
|
|
config['zero'] = {
|
|
'fields': ['rg_track_peak', 'rg_track_gain'],
|
|
}
|
|
self.load_plugins('zero')
|
|
|
|
item.write()
|
|
mediafile = MediaFile(item.path)
|
|
self.assertIsNone(mediafile.rg_track_peak)
|
|
self.assertIsNone(mediafile.rg_track_gain)
|
|
|
|
def test_do_not_change_database(self):
|
|
item = self.add_item_fixture(year=2000)
|
|
item.write()
|
|
mediafile = MediaFile(item.path)
|
|
self.assertEqual(2000, mediafile.year)
|
|
|
|
config['zero'] = {'fields': ['year']}
|
|
self.load_plugins('zero')
|
|
|
|
item.write()
|
|
mediafile = MediaFile(item.path)
|
|
self.assertEqual(item['year'], 2000)
|
|
self.assertIsNone(mediafile.year)
|
|
|
|
def test_album_art(self):
|
|
path = self.create_mediafile_fixture(images=['jpg'])
|
|
item = Item.from_path(path)
|
|
|
|
mediafile = MediaFile(item.path)
|
|
self.assertNotEqual(0, len(mediafile.images))
|
|
|
|
config['zero'] = {'fields': ['images']}
|
|
self.load_plugins('zero')
|
|
|
|
item.write()
|
|
mediafile = MediaFile(item.path)
|
|
self.assertEqual(0, len(mediafile.images))
|
|
|
|
|
|
def suite():
|
|
return unittest.TestLoader().loadTestsFromName(__name__)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(defaultTest='suite')
|