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

Move function to increase DRY

Browse source
This commit is contained in:
David 2014-11-16 21:39:58 -05:00
parent 38f056a30c
commit e1ff258ce5
1 changed files with 13 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
test/test_importer.py
View file

@ -334,6 +334,17 @@ class NonAutotaggedImportTest(_common.TestCase, ImportHelper):
self.assertTrue(os.path.islink(filename))
self.assertEqual(os.readlink(filename), mediafile.path)
def create_archive(session):
(handle, path) = mkstemp(dir=session.temp_dir)
os.close(handle)
archive = ZipFile(path, mode='w')
archive.write(os.path.join(_common.RSRC, 'full.mp3'),
'full.mp3')
archive.close()
return path
class RmTempTest(unittest.TestCase, ImportHelper):
"""Tests that temporarily extracted archives are properly removed
after usage.
@ -349,7 +360,7 @@ class RmTempTest(unittest.TestCase, ImportHelper):
self.teardown_beets()
def test_rm(self):
zip_path = self.create_archive()
zip_path = create_archive(self)
archive_task = importer.ArchiveImportTask(zip_path)
archive_task.extract()
tmp_path = archive_task.toppath
@ -376,7 +387,7 @@ class ImportZipTest(unittest.TestCase, ImportHelper):
self.teardown_beets()
def test_import_zip(self):
zip_path = self.create_archive()
zip_path = create_archive(self)
self.assertEqual(len(self.lib.items()), 0)
self.assertEqual(len(self.lib.albums()), 0)
@ -385,15 +396,6 @@ class ImportZipTest(unittest.TestCase, ImportHelper):
self.assertEqual(len(self.lib.items()), 1)
self.assertEqual(len(self.lib.albums()), 1)
def create_archive(self):
(handle, path) = mkstemp(dir=self.temp_dir)
os.close(handle)
archive = ZipFile(path, mode='w')
archive.write(os.path.join(_common.RSRC, 'full.mp3'),
'full.mp3')
archive.close()
return path
class ImportTarTest(ImportZipTest):