Added autospec=True to patch in tests

maykinmedia · Mar 5, 2024 · c04166a · c04166a
1 parent 26c5358
commit c04166a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/tests/test_process_html.py b/tests/test_process_html.py
@@ -10,8 +10,12 @@ class ProcessTestCase(TestCase):
     note the patched helpers are individually and more exhaustively tested elsewhere
     """
 
-    @patch("mail_editor.process.cid_for_bytes", return_value="MY_CID")
-    @patch("mail_editor.process.load_image", return_value=FileData(b"abc", "image/jpg"))
+    @patch("mail_editor.process.cid_for_bytes", return_value="MY_CID", autospec=True)
+    @patch(
+        "mail_editor.process.load_image",
+        return_value=FileData(b"abc", "image/jpg"),
+        autospec=True,
+    )
     def test_extract_images(self, m1, m2):
         html = """
             <html><body>
@@ -27,7 +31,7 @@ def test_extract_images(self, m1, m2):
         self.assertHTMLEqual(result.html, expected_html)
         self.assertEqual(result.cid_attachments, [("MY_CID", b"abc", "image/jpg")])
 
-    @patch("mail_editor.process.load_image", return_value=None)
+    @patch("mail_editor.process.load_image", return_value=None, autospec=True)
     def test_extract_images__keeps_absolute_url_when_not_loadable(self, m):
         html = """
             <html><body>

diff --git a/tests/test_send.py b/tests/test_send.py
@@ -56,8 +56,12 @@ def test_send_email(self):
         self.assertIn(str(_("Test mail sent from testcase with 111")), message.body)
         self.assertEqual(message.attachments, [])
 
-    @patch("mail_editor.process.cid_for_bytes", return_value="MY_CID")
-    @patch("mail_editor.process.load_image", return_value=FileData(b"abc", "image/jpg"))
+    @patch("mail_editor.process.cid_for_bytes", return_value="MY_CID", autospec=True)
+    @patch(
+        "mail_editor.process.load_image",
+        return_value=FileData(b"abc", "image/jpg"),
+        autospec=True,
+    )
     @override_settings(MAIL_EDITOR_CONF=CONFIG)
     def test_send_email_processed_content(self, m0, m1):
         subject_context = {"id": "111"}