mirror of
https://github.com/beetbox/beets.git
synced 2025-12-07 09:04:33 +01:00
ae5e55cad6
Related to #1966. Previously, we used a `syspath` call inside MediaFile, which probably wasn't right: the constructor should behave like `open` in that we need to use pass an OS path.
129 lines
3.7 KiB
Python
129 lines
3.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""Tests for the 'zero' plugin"""
|
|
|
|
from __future__ import division, absolute_import, print_function
|
|
|
|
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
|
|
from beets.util import syspath
|
|
|
|
|
|
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': u'test comment',
|
|
'day': 13,
|
|
'month': 3,
|
|
'year': 2012,
|
|
}
|
|
z = ZeroPlugin()
|
|
z.debug = False
|
|
z.fields = ['comments', 'month', 'day']
|
|
z.patterns = {'comments': [u'.'],
|
|
'month': [u'.'],
|
|
'day': [u'.']}
|
|
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': u'eac lame'.split(),
|
|
'year': u'2098 2099'.split()}
|
|
|
|
tags = {
|
|
'comments': u'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(syspath(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(syspath(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(syspath(item.path))
|
|
self.assertEqual(2000, mediafile.year)
|
|
|
|
config['zero'] = {'fields': ['year']}
|
|
self.load_plugins('zero')
|
|
|
|
item.write()
|
|
mediafile = MediaFile(syspath(item.path))
|
|
self.assertEqual(item['year'], 2000)
|
|
self.assertIsNone(mediafile.year)
|
|
|
|
def test_change_database(self):
|
|
item = self.add_item_fixture(year=2000)
|
|
item.write()
|
|
mediafile = MediaFile(syspath(item.path))
|
|
self.assertEqual(2000, mediafile.year)
|
|
|
|
config['zero'] = {
|
|
'fields': [u'year'],
|
|
'update_database': True,
|
|
}
|
|
self.load_plugins('zero')
|
|
|
|
item.write()
|
|
mediafile = MediaFile(syspath(item.path))
|
|
self.assertEqual(item['year'], 0)
|
|
self.assertIsNone(mediafile.year)
|
|
|
|
def test_album_art(self):
|
|
path = self.create_mediafile_fixture(images=['jpg'])
|
|
item = Item.from_path(path)
|
|
|
|
mediafile = MediaFile(syspath(item.path))
|
|
self.assertNotEqual(0, len(mediafile.images))
|
|
|
|
config['zero'] = {'fields': [u'images']}
|
|
self.load_plugins('zero')
|
|
|
|
item.write()
|
|
mediafile = MediaFile(syspath(item.path))
|
|
self.assertEqual(0, len(mediafile.images))
|
|
|
|
|
|
def suite():
|
|
return unittest.TestLoader().loadTestsFromName(__name__)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(defaultTest='suite')
|