From d8a7828cb570bc6ba0d298fa7978835c1bdeb812 Mon Sep 17 00:00:00 2001
From: Trenton Holmes <holmes.trenton@gmail.com>
Date: Mon, 13 Jun 2022 16:34:46 -0700
Subject: [PATCH] Splits tests for the tasks into their own classes

---
 src/documents/tests/test_tasks.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/documents/tests/test_tasks.py b/src/documents/tests/test_tasks.py
index 998b7d955..4db6e3369 100644
--- a/src/documents/tests/test_tasks.py
+++ b/src/documents/tests/test_tasks.py
@@ -18,7 +18,7 @@ from documents.tests.utils import DirectoriesMixin
 from PIL import Image
 
 
-class TestTasks(DirectoriesMixin, TestCase):
+class TestIndexReindex(DirectoriesMixin, TestCase):
     def test_index_reindex(self):
         Document.objects.create(
             title="test",
@@ -43,6 +43,8 @@ class TestTasks(DirectoriesMixin, TestCase):
 
         tasks.index_optimize()
 
+
+class TestClassifier(DirectoriesMixin, TestCase):
     @mock.patch("documents.tasks.load_classifier")
     def test_train_classifier_no_auto_matching(self, load_classifier):
         tasks.train_classifier()
@@ -93,6 +95,8 @@ class TestTasks(DirectoriesMixin, TestCase):
         mtime3 = os.stat(settings.MODEL_FILE).st_mtime
         self.assertNotEqual(mtime2, mtime3)
 
+
+class TestBarcode(DirectoriesMixin, TestCase):
     def test_barcode_reader(self):
         test_file = os.path.join(
             os.path.dirname(__file__),
@@ -529,6 +533,8 @@ class TestTasks(DirectoriesMixin, TestCase):
 
         self.assertEqual(tasks.consume_file(dst), "File successfully split")
 
+
+class TestSanityCheck(DirectoriesMixin, TestCase):
     @mock.patch("documents.tasks.sanity_checker.check_sanity")
     def test_sanity_check_success(self, m):
         m.return_value = SanityCheckMessages()
@@ -565,6 +571,8 @@ class TestTasks(DirectoriesMixin, TestCase):
         )
         m.assert_called_once()
 
+
+class TestBulkUpdate(DirectoriesMixin, TestCase):
     def test_bulk_update_documents(self):
         doc1 = Document.objects.create(
             title="test",