Merge pull request #1270 from leonelcamara/test_week5b

Test Mail.Attachment sending test_tools.py itself
2016-04-06 20:57:06 -05:00
parent 4d1a4c48e6 0629df71ef
commit 3c69716672
1 changed files with 15 additions and 0 deletions
@@ -161,6 +161,21 @@ class TestMail(unittest.TestCase):
                                  message='world'))
        TestMail.DummySMTP.inbox.pop()

+    def test_attachment(self):
+        module_file = os.path.abspath(__file__)
+        mail = Mail()
+        mail.settings.server = 'smtp.example.com:25'
+        mail.settings.sender = 'you@example.com'
+        self.assertTrue(mail.send(to=['somebody@example.com'],
+                                  subject='hello',
+                                  message='world',
+                                  attachments=Mail.Attachment(module_file)))
+        message = TestMail.DummySMTP.inbox.pop()
+        import email
+        parsed_msg = email.message_from_string(message.payload)
+        attachment = parsed_msg.get_payload(1).get_payload(decode=True)
+        with open(module_file, 'rb') as mf:
+            self.assertEqual(attachment.decode('utf-8'), mf.read().decode('utf-8'))

 # TODO class TestRecaptcha(unittest.TestCase):