1 files changed, 106 insertions, 0 deletions
diff --git a/kittystore/test/test_scrub.py b/kittystore/test/test_scrub.py
new file mode 100644
index 0000000..d590709
--- /dev/null
+++ b/kittystore/test/test_scrub.py
@@ -0,0 +1,106 @@
+# -*- coding: utf-8 -*-
+
+import unittest
+import email
+
+from mock import Mock
+
+from kittystore.scrub import Scrubber
+from kittystore.test import get_test_file
+
+
+class TestScrubber(unittest.TestCase):
+
+    def test_attachment_1(self):
+        with open(get_test_file("attachment-1.txt")) as email_file:
+            msg = email.message_from_file(email_file)
+        store = Mock()
+        scrubber = Scrubber("testlist@example.com", msg, store)
+        contents = scrubber.scrub()
+        self.assertEqual(store.add_attachment.call_count, 1)
+        store.add_attachment.assert_called_with(
+                'testlist@example.com', '505E5185.5040208@libero.it', 2,
+                'puntogil.vcf', 'text/x-vcard',
+                'begin:vcard\r\nfn:gil\r\nn:;gil\r\nversion:2.1\r\n'
+                'end:vcard\r\n\r\n')
+        self.assertEqual(contents,
+                "This is a test message.\r\n\r\n"
+                "-------------- next part --------------\n"
+                "Skipped content of type %(partctype)s\n"
+                "-------------- next part --------------\n"
+                "-- \ndevel mailing list\ndevel@lists.fedoraproject.org\n"
+                "https://admin.fedoraproject.org/mailman/listinfo/devel\n"
+                )
+        self.fail() # Fix the expected text above
+
+    def test_attachment_2(self):
+        with open(get_test_file("attachment-2.txt")) as email_file:
+            msg = email.message_from_file(email_file)
+        store = Mock()
+        scrubber = Scrubber("testlist@example.com", msg, store)
+        contents = scrubber.scrub()
+        self.assertEqual(store.add_attachment.call_count, 1)
+        store.add_attachment.assert_called_with(
+                'testlist@example.com', '50619B7A.2030404@thelounge.net', 3,
+                'signature.asc', 'application/pgp-signature',
+                '-----BEGIN PGP SIGNATURE-----\r\nVersion: GnuPG v1.4.12 '
+                '(GNU/Linux)\r\nComment: Using GnuPG with Mozilla - '
+                'http://www.enigmail.net/\r\n\r\niEYEARECAAYFAlBhm3oACgkQhmBj'
+                'z394AnmMnQCcC+6tWcqE1dPQmIdRbLXgKGVp\r\nEeUAn2OqtaXaXaQV7rx+'
+                'SmOldmSzcFw4\r\n=OEJv\r\n-----END PGP SIGNATURE-----\r\n')
+        self.assertEqual(contents,
+                u"This is a test message\r\nNon-ascii chars: Hofm\xfchlgasse"
+                u"\r\n-------------- next part --------------\n"
+                u"Skipped content of type %(partctype)s\n"
+                u"-------------- next part --------------\n"
+                u"-- \ndevel mailing list\ndevel@lists.fedoraproject.org\n"
+                u"https://admin.fedoraproject.org/mailman/listinfo/devel\n"
+                )
+        self.fail() # Fix the expected text above
+
+    def test_attachment_3(self):
+        with open(get_test_file("attachment-3.txt")) as email_file:
+            msg = email.message_from_file(email_file)
+        store = Mock()
+        scrubber = Scrubber("testlist@example.com", msg, store)
+        contents = scrubber.scrub()
+        self.assertEqual(store.add_attachment.call_count, 2)
+        args_1, args_2 = store.add_attachment.call_args_list
+        # HTML part
+        self.assertEqual(args_1[0][0:5], ("testlist@example.com",
+                "CACec3Lup8apbhUMcm_Ktn1dPxx4eWr2y1RV7ZSYhy0tzmjSrgQ@mail.gmail.com",
+                3, "attachment.html", "text/html"))
+        self.assertEqual(len(args_1[0][5]), 5812)
+        # Image attachment
+        self.assertEqual(args_2[0][0:5], ("testlist@example.com",
+                "CACec3Lup8apbhUMcm_Ktn1dPxx4eWr2y1RV7ZSYhy0tzmjSrgQ@mail.gmail.com",
+                4, "GeoffreyRoucourt.jpg", "image/jpeg"))
+        self.assertEqual(len(args_2[0][5]), 282180)
+        # Scrubbed content
+        self.assertEqual(contents,
+                u"This is a test message\r\n"
+                u"-------------- next part --------------\n"
+                u"Skipped content of type %(partctype)s\n"
+                )
+        self.fail() # Fix the expected text above
+
+    def test_html_email_1(self):
+        with open(get_test_file("html-email-1.txt")) as email_file:
+            msg = email.message_from_file(email_file)
+        store = Mock()
+        scrubber = Scrubber("testlist@example.com", msg, store)
+        contents = scrubber.scrub()
+        self.assertEqual(store.add_attachment.call_count, 1)
+        args = store.add_attachment.call_args[0]
+        # HTML part
+        self.assertEqual(args[0:5], ("testlist@example.com",
+                "016001cd9b3b$b71efed0$255cfc70$@fr",
+                2, "attachment.html", "text/html"))
+        self.assertEqual(len(args[5]), 5093)
+        # Scrubbed content
+        self.assertEqual(contents,
+                u"This is a test message\r\n"
+                u"Non-ASCII chars: r\xe9ponse fran\xe7ais \n"
+                )
+        self.fail() # Fix the expected text above
+