From 33bed79a13c36e2b4fa59c134da65c6b57f5d6c7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C5=A0ar=C5=ABnas=20Nejus?= <snejus@protonmail.com>
Date: Mon, 21 Apr 2025 20:01:31 +0100
Subject: [PATCH] Move scrub test to a separate file

---
 test/plugins/test_scrub.py | 37 ++++++++++++++++++++++++++++++
 test/test_importer.py      | 47 --------------------------------------
 2 files changed, 37 insertions(+), 47 deletions(-)
 create mode 100644 test/plugins/test_scrub.py

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