Working on test updates

This commit is contained in:
Trenton H 2024-04-09 11:26:00 -07:00
parent c77f01b750
commit 5f18e4c9bc

View File

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