diff --git a/test/plugins/test_scrub.py b/test/plugins/test_scrub.py new file mode 100644 index 000000000..129d91a22 --- /dev/null +++ b/test/plugins/test_scrub.py @@ -0,0 +1,37 @@ +import os + +from mediafile import MediaFile + +from beets.test.helper import AsIsImporterMixin, ImportTestCase, PluginMixin + + +class ScrubbedImportTest(AsIsImporterMixin, PluginMixin, ImportTestCase): + db_on_disk = True + plugin = "scrub" + + def test_tags_not_scrubbed(self): + with self.configure_plugin({"auto": False}): + self.run_asis_importer(write=True) + + for item in self.lib.items(): + imported_file = MediaFile(os.path.join(item.path)) + assert imported_file.artist == "Tag Artist" + assert imported_file.album == "Tag Album" + + def test_tags_restored(self): + with self.configure_plugin({"auto": True}): + self.run_asis_importer(write=True) + + for item in self.lib.items(): + imported_file = MediaFile(os.path.join(item.path)) + assert imported_file.artist == "Tag Artist" + assert imported_file.album == "Tag Album" + + def test_tags_not_restored(self): + with self.configure_plugin({"auto": True}): + self.run_asis_importer(write=False) + + for item in self.lib.items(): + imported_file = MediaFile(os.path.join(item.path)) + assert imported_file.artist is None + assert imported_file.album is None diff --git a/test/test_importer.py b/test/test_importer.py index dad30749b..2b353653f 100644 --- a/test/test_importer.py +++ b/test/test_importer.py @@ -50,53 +50,6 @@ from beets.test.helper import ( from beets.util import bytestring_path, displayable_path, syspath -class ScrubbedImportTest(AsIsImporterMixin, PluginMixin, ImportTestCase): - db_on_disk = True - plugin = "scrub" - - def test_tags_not_scrubbed(self): - config["plugins"] = ["scrub"] - config["scrub"]["auto"] = False - config["import"]["write"] = True - for mediafile in self.import_media: - assert mediafile.artist == "Tag Artist" - assert mediafile.album == "Tag Album" - self.run_asis_importer() - for item in self.lib.items(): - imported_file = os.path.join(item.path) - imported_file = MediaFile(imported_file) - assert imported_file.artist == "Tag Artist" - assert imported_file.album == "Tag Album" - - def test_tags_restored(self): - config["plugins"] = ["scrub"] - config["scrub"]["auto"] = True - config["import"]["write"] = True - for mediafile in self.import_media: - assert mediafile.artist == "Tag Artist" - assert mediafile.album == "Tag Album" - self.run_asis_importer() - for item in self.lib.items(): - imported_file = os.path.join(item.path) - imported_file = MediaFile(imported_file) - assert imported_file.artist == "Tag Artist" - assert imported_file.album == "Tag Album" - - def test_tags_not_restored(self): - config["plugins"] = ["scrub"] - config["scrub"]["auto"] = True - config["import"]["write"] = False - for mediafile in self.import_media: - assert mediafile.artist == "Tag Artist" - assert mediafile.album == "Tag Album" - self.run_asis_importer() - for item in self.lib.items(): - imported_file = os.path.join(item.path) - imported_file = MediaFile(imported_file) - assert imported_file.artist is None - assert imported_file.album is None - - @_common.slow_test() class NonAutotaggedImportTest(AsIsImporterMixin, ImportTestCase): db_on_disk = True