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

Refactor ImportDuplicateAlbumTest

Browse source
This commit is contained in:
Thomas Scholtes 2014-04-19 15:49:39 +02:00
parent 52e3413c5b
commit efee81cda7
1 changed files with 18 additions and 24 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
test/test_importer.py
View file

@ -963,9 +963,13 @@ class ImportDuplicateAlbumTest(unittest.TestCase, TestHelper):
def setUp(self):
self.setup_beets()
# Original album
self.add_album_fixture(albumartist=u'artist', album=u'album')
# Create duplicate through autotagger
self.match_album_patcher = patch('beets.autotag.mb.match_album')
self.match_album = self.match_album_patcher.start()
track_info = TrackInfo(
title=u'new title',
track_id=u'trackid',
@ -979,18 +983,20 @@ class ImportDuplicateAlbumTest(unittest.TestCase, TestHelper):
)
self.match_album.return_value = iter([album_info])
# Create import session
self.importer = self.create_importer()
config['import']['autotag'] = True
def tearDown(self):
self.match_album_patcher.stop()
self.teardown_beets()
def test_remove_duplicate_album(self):
self.add_album_fixture()
item = self.lib.items().get()
self.assertNotEqual(item.title, u'new title')
self.assertEqual(item.title, u't\xeftle 0')
importer = self.create_importer()
importer.default_resolution = importer.Resolution.REMOVE
importer.run()
self.importer.default_resolution = self.importer.Resolution.REMOVE
self.importer.run()
self.assertEqual(len(self.lib.albums()), 1)
self.assertEqual(len(self.lib.items()), 1)
@ -998,23 +1004,18 @@ class ImportDuplicateAlbumTest(unittest.TestCase, TestHelper):
self.assertEqual(item.title, u'new title')
def test_keep_duplicate_album(self):
self.add_album_fixture()
importer = self.create_importer()
importer.default_resolution = importer.Resolution.KEEPBOTH
importer.run()
self.importer.default_resolution = self.importer.Resolution.KEEPBOTH
self.importer.run()
self.assertEqual(len(self.lib.albums()), 2)
self.assertEqual(len(self.lib.items()), 2)
def test_skip_duplicate_album(self):
self.add_album_fixture()
item = self.lib.items().get()
self.assertEqual(item.title, u't\xeftle 0')
importer = self.create_importer()
importer.default_resolution = importer.Resolution.SKIP
importer.run()
self.importer.default_resolution = self.importer.Resolution.SKIP
self.importer.run()
self.assertEqual(len(self.lib.albums()), 1)
self.assertEqual(len(self.lib.items()), 1)
@ -1024,20 +1025,13 @@ class ImportDuplicateAlbumTest(unittest.TestCase, TestHelper):
def test_twice_in_import_dir(self):
self.skipTest('write me')
def add_album_fixture(self):
def add_album_fixture(self, **kwargs):
# TODO move this into upstream
album = super(ImportDuplicateAlbumTest, self).add_album_fixture()
album.artist = 'artist'
album.albumartist = 'artist'
album.album = 'album'
album.update(kwargs)
album.store()
return album
def create_importer(self):
importer = super(ImportDuplicateAlbumTest, self).create_importer()
config['import']['autotag'] = True
return importer
class ImportDuplicateSingletonTest(unittest.TestCase, TestHelper):