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

Add setup_singleton_importer for singleton importer

Browse source
This commit is contained in:
Šarūnas Nejus 2024-07-14 12:39:40 +01:00
parent 92f79dbfb9
commit 139b54ebd1
No known key found for this signature in database
GPG key ID: DD28F6704DBE3435
1 changed files with 15 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
test/test_importer.py
View file

@ -311,7 +311,7 @@ class ImportSingletonTest(ImportTestCase):
def setUp(self): def setUp(self):
super().setUp() super().setUp()
self.prepare_album_for_import(1) self.prepare_album_for_import(1)
self.setup_importer(singletons=True) self.importer = self.setup_singleton_importer()
self.matcher = AutotagStub().install() self.matcher = AutotagStub().install()
def tearDown(self): def tearDown(self):
@ -560,7 +560,7 @@ class ImportTest(ImportTestCase):
def test_empty_directory_singleton_warning(self): def test_empty_directory_singleton_warning(self):
import_dir = os.path.join(self.temp_dir, b"empty") import_dir = os.path.join(self.temp_dir, b"empty")
self.touch(b"non-audio", dir=import_dir) self.touch(b"non-audio", dir=import_dir)
self.setup_importer(import_dir=import_dir, singletons=True) self.setup_singleton_importer(import_dir=import_dir)
with capture_log() as logs: with capture_log() as logs:
self.importer.run() self.importer.run()
@ -1251,8 +1251,8 @@ class ImportDuplicateSingletonTest(ImportTestCase):
# Import session # Import session
self.prepare_album_for_import(1) self.prepare_album_for_import(1)
self.importer = self.setup_importer( self.importer = self.setup_singleton_importer(
duplicate_keys={"album": "artist title"}, singletons=True duplicate_keys={"album": "artist title"}
) )
def test_remove_duplicate(self): def test_remove_duplicate(self):
@ -1351,8 +1351,8 @@ class ResumeImportTest(ImportTestCase):
@patch("beets.plugins.send") @patch("beets.plugins.send")
def test_resume_singleton(self, plugins_send): def test_resume_singleton(self, plugins_send):
self.prepare_album_for_import(2) self.prepare_album_for_import(2)
self.importer = self.setup_importer( self.importer = self.setup_singleton_importer(
autotag=False, resume=True, singletons=True autotag=False, resume=True
) )
# Aborts import after one track. This also ensures that we skip # Aborts import after one track. This also ensures that we skip
@ -1718,10 +1718,8 @@ class ImportPretendTest(ImportTestCase):
os.makedirs(syspath(path)) os.makedirs(syspath(path))
self.empty_path = path self.empty_path = path
def __run(self, import_paths, singletons=True): def __run(self, import_paths):
self.setup_importer(singletons=singletons)
self.importer.paths = import_paths self.importer.paths = import_paths
with capture_log() as logs: with capture_log() as logs:
self.importer.run() self.importer.run()
@ -1733,6 +1731,7 @@ class ImportPretendTest(ImportTestCase):
return logs return logs
def test_import_singletons_pretend(self): def test_import_singletons_pretend(self):
self.importer = self.setup_singleton_importer()
logs = self.__run(self.import_paths) logs = self.__run(self.import_paths)
self.assertEqual( self.assertEqual(
@ -1744,7 +1743,8 @@ class ImportPretendTest(ImportTestCase):
) )
def test_import_album_pretend(self): def test_import_album_pretend(self):
logs = self.__run(self.import_paths, singletons=False) self.importer = self.setup_importer()
logs = self.__run(self.import_paths)
self.assertEqual( self.assertEqual(
logs, logs,
@ -1757,6 +1757,7 @@ class ImportPretendTest(ImportTestCase):
) )
def test_import_pretend_empty(self): def test_import_pretend_empty(self):
self.importer = self.setup_importer()
logs = self.__run([self.empty_path]) logs = self.__run([self.empty_path])
self.assertEqual( self.assertEqual(
@ -1907,9 +1908,8 @@ class ImportMusicBrainzIdTest(ImportTestCase):
self.assertEqual(self.lib.albums().get().album, "VALID_RELEASE_1") self.assertEqual(self.lib.albums().get().album, "VALID_RELEASE_1")
def test_one_mbid_one_singleton(self): def test_one_mbid_one_singleton(self):
self.setup_importer( self.setup_singleton_importer(
search_ids=[self.MB_RECORDING_PREFIX + self.ID_RECORDING_0], search_ids=[self.MB_RECORDING_PREFIX + self.ID_RECORDING_0]
singletons=True,
) )
self.importer.add_choice(importer.action.APPLY) self.importer.add_choice(importer.action.APPLY)
@ -1917,12 +1917,11 @@ class ImportMusicBrainzIdTest(ImportTestCase):
self.assertEqual(self.lib.items().get().title, "VALID_RECORDING_0") self.assertEqual(self.lib.items().get().title, "VALID_RECORDING_0")
def test_several_mbid_one_singleton(self): def test_several_mbid_one_singleton(self):
self.setup_importer( self.setup_singleton_importer(
search_ids=[ search_ids=[
self.MB_RECORDING_PREFIX + self.ID_RECORDING_0, self.MB_RECORDING_PREFIX + self.ID_RECORDING_0,
self.MB_RECORDING_PREFIX + self.ID_RECORDING_1, self.MB_RECORDING_PREFIX + self.ID_RECORDING_1,
], ]
singletons=True,
) )
self.importer.add_choice(2) # Pick the 2nd best match (recording 1). self.importer.add_choice(2) # Pick the 2nd best match (recording 1).