tests:include assert for rmtree calling after exception

reclamador · Feb 22, 2018 · 4405931 · 4405931
1 parent 57bf33e
commit 4405931
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_document_clipper_pdf.py b/tests/test_document_clipper_pdf.py
@@ -319,11 +319,13 @@ def test_pdf_to_text_from_pdf_with_images_no_jpg(self):
         self.document_clipper_pdf_reader.pdf_to_text(self._images_to_text_method_mocked())
         self.assertEqual(1, len(self.document_clipper_pdf_reader._pdf_image_to_text_method.call_args_list))
 
-    def test_pdf_to_text_from_pdf_with_images_exception_raised(self):
+    @patch('shutil.rmtree')
+    def test_pdf_to_text_from_pdf_with_images_exception_raised(self, mock_rmtree):
         self.pdf_file = open(PATH_TO_PDF_FILE_WITH_IMAGES)
         self.document_clipper_pdf_reader = DocumentClipperPdfReader(self.pdf_file)
         self.assertRaises(Exception, self.document_clipper_pdf_reader.pdf_to_text,
                           self._images_to_text_method_mocked_with_exception())
+        self.assertEqual(1, len(mock_rmtree.call_args_list))
 
     @patch('os.remove')
     def test_fix_pdf_ok(self, mock_os_remove):