Backup_Repos/beets
1
0
Fork 0
mirror of https://github.com/beetbox/beets.git synced 2025-12-25 01:53:31 +01:00
Code Issues Projects Releases Wiki Activity

lastgenre: use an ordinary Item instead of a Mock

Browse source
This commit is contained in:
Fabrice Laporte 2014-10-09 06:38:40 +02:00
parent b95990ee9c
commit 2b908b48bc
1 changed files with 12 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
test/test_lastgenre.py
View file

@ -16,6 +16,7 @@
from mock import Mock
import _common
from _common import unittest
from beetsplug import lastgenre
from beets import config
@ -179,32 +180,32 @@ class LastGenrePluginTest(unittest.TestCase, TestHelper):
lastgenre.LastGenrePlugin.fetch_artist_genre = mock_fetch_artist_genre
self._setup_config(whitelist=False)
item_mock = Mock(spec=library.Item)
item_mock.genre = MOCK_GENRES['track']
item = _common.item()
item.genre = MOCK_GENRES['track']
config['lastgenre'] = {'force': False}
res = self.plugin._get_genre(item_mock)
self.assertEqual(res, (item_mock.genre, 'keep'))
res = self.plugin._get_genre(item)
self.assertEqual(res, (item.genre, 'keep'))
config['lastgenre'] = {'force': True, 'source': 'track'}
res = self.plugin._get_genre(item_mock)
res = self.plugin._get_genre(item)
self.assertEqual(res, (MOCK_GENRES['track'], 'track'))
config['lastgenre'] = {'source': 'album'}
res = self.plugin._get_genre(item_mock)
res = self.plugin._get_genre(item)
self.assertEqual(res, (MOCK_GENRES['album'], 'album'))
config['lastgenre'] = {'source': 'artist'}
res = self.plugin._get_genre(item_mock)
res = self.plugin._get_genre(item)
self.assertEqual(res, (MOCK_GENRES['artist'], 'artist'))
MOCK_GENRES['artist'] = None
res = self.plugin._get_genre(item_mock)
self.assertEqual(res, (item_mock.genre, 'original'))
res = self.plugin._get_genre(item)
self.assertEqual(res, (item.genre, 'original'))
config['lastgenre'] = {'fallback': 'rap'}
item_mock.genre = None
res = self.plugin._get_genre(item_mock)
item.genre = None
res = self.plugin._get_genre(item)
self.assertEqual(res, (config['lastgenre']['fallback'].get(),
'fallback'))