From 5f18e4c9bce531555b407de7d7d4b41e7a14ab10 Mon Sep 17 00:00:00 2001
From: Trenton H <797416+stumpylog@users.noreply.github.com>
Date: Tue, 9 Apr 2024 11:26:00 -0700
Subject: [PATCH] Working on test updates

---
 src/documents/tests/test_consumer.py | 48 ++++++++++++++--------------
 1 file changed, 24 insertions(+), 24 deletions(-)

diff --git a/src/documents/tests/test_consumer.py b/src/documents/tests/test_consumer.py
index e666cb835..891d6d9e2 100644
--- a/src/documents/tests/test_consumer.py
+++ b/src/documents/tests/test_consumer.py
@@ -670,12 +670,12 @@ class TestConsumer(DirectoriesMixin, FileSystemAssertsMixin, TestCase):
     def testNoParsers(self, m):
         m.return_value = []
 
-        self.assertRaisesMessage(
-            ConsumerError,
-            "sample.pdf: Unsupported mime type application/pdf",
-            self.consumer.try_consume_file,
-            self.get_test_file(),
-        )
+        with self.get_consumer(self.get_test_file()) as consumer:
+            with self.assertRaisesMessage(
+                ConsumerError,
+                "sample.pdf: Unsupported mime type application/pdf",
+            ):
+                consumer.run()
 
         self._assert_first_last_send_progress(last_status="FAILED")
 
@@ -692,12 +692,12 @@ class TestConsumer(DirectoriesMixin, FileSystemAssertsMixin, TestCase):
             ),
         ]
 
-        self.assertRaisesMessage(
-            ConsumerError,
-            "sample.pdf: Error occurred while consuming document sample.pdf: Does not compute.",
-            self.consumer.try_consume_file,
-            self.get_test_file(),
-        )
+        with self.get_consumer(self.get_test_file()) as consumer:
+            with self.assertRaisesMessage(
+                ConsumerError,
+                "sample.pdf: Error occurred while consuming document sample.pdf: Does not compute.",
+            ):
+                consumer.run()
 
         self._assert_first_last_send_progress(last_status="FAILED")
 
@@ -714,12 +714,12 @@ class TestConsumer(DirectoriesMixin, FileSystemAssertsMixin, TestCase):
             ),
         ]
 
-        self.assertRaisesMessage(
-            ConsumerError,
-            "sample.pdf: Unexpected error while consuming document sample.pdf: Generic exception.",
-            self.consumer.try_consume_file,
-            self.get_test_file(),
-        )
+        with self.get_consumer(self.get_test_file()) as consumer:
+            with self.assertRaisesMessage(
+                ConsumerError,
+                "sample.pdf: Unexpected error while consuming document sample.pdf: Generic exception.",
+            ):
+                consumer.run()
 
         self._assert_first_last_send_progress(last_status="FAILED")
 
@@ -728,12 +728,12 @@ class TestConsumer(DirectoriesMixin, FileSystemAssertsMixin, TestCase):
         filename = self.get_test_file()
         m.side_effect = OSError("NO.")
 
-        self.assertRaisesMessage(
-            ConsumerError,
-            "sample.pdf: The following error occurred while storing document sample.pdf after parsing: NO.",
-            self.consumer.try_consume_file,
-            filename,
-        )
+        with self.get_consumer(self.get_test_file()) as consumer:
+            with self.assertRaisesMessage(
+                ConsumerError,
+                "sample.pdf: The following error occurred while storing document sample.pdf after parsing: NO.",
+            ):
+                consumer.run()
 
         self._assert_first_last_send_progress(last_status="FAILED")