Пример #1
0
    def testEmailParsing(self):
        # create an initial "from" email
        message = email.message_from_file(
            open("./testdata/mail3.txt", "r"))
        name, mto = parseaddr(message['to'])
        name, mfrom = parseaddr(message['from'])
        for part in message.walk():
            if part.get_content_maintype() == "multipart":
                continue # it's just a container
            if part.get_content_type() == "text/plain":
                body = part.get_payload(decode=True)
                break
        try:
            user = CustomUser.objects.get(
                email=mfrom)
        except CustomUser.DoesNotExist:
            user = CustomUser.objects.create(email=mfrom,
                                             username=mfrom)
            user.set_unusable_password()
            user.save()
        recipient, _ = CustomUser.objects.get_or_create(
            email=mto,
            username=mto)
        mail = Mail.objects.create(
            subject=message['subject'],
            mfrom=user,
            mto=recipient,
            message=body)

        # now we simulate a reply, and override a couple of relevant
        # headers 
        raw_email = email.message_from_file(
            open("./testdata/mail4.txt", "r"))
        new_mto = user.proxy_email
        raw_email.replace_header('in-reply-to', "<%s>" % mail.message_id)
        raw_email.replace_header('references', "<%s>" % mail.message_id)
        raw_email.replace_header('to', "%s <%s>" % (mail.message_id,
                                      new_mto))
        fp = StringIO(raw_email.as_string()) 
        response = make_response_from_file(fp)
        self.assertTrue(response)
        
        # now parse in an email that isn't a response to anything
        fp = open("./testdata/mail2.txt", "r")
        response = make_response_from_file(fp)
        self.assertEqual(response, EX_NOUSER)

        # and now try threading based on subject line as a fallback...
        raw_email = email.message_from_file(
            open("./testdata/mail2.txt", "r"))
        new_mto = user.proxy_email
        del(raw_email['in-reply-to'])
        del(raw_email['references'])
        raw_email.replace_header('subject', "Re:RE: re:%s" % mail.subject)
        raw_email.replace_header('to', "%s <%s>" % (mail.message_id,
                                      new_mto))
        fp = StringIO(raw_email.as_string()) 
        response = make_response_from_file(fp)
        self.assertTrue(response)
Пример #2
0
    def testSendAndReply(self):
        user = CustomUser.objects.create(email="*****@*****.**",
                                         username="******",
                                         needs_moderation=False)
        data = QueryDict("").copy()
        SUBJECT = "Test Send And Reply"
        data.update({'mto':'*****@*****.**',
                     'mfrom':user.email,
                     'subject':SUBJECT,
                     'message':'Hi'})
        form = MailForm(None, data)
        self.assertFalse(form.is_valid())
        data.update({'mto':'*****@*****.**',
                     'mfrom':user.email,
                     'subject':SUBJECT,
                     'name': 'One Baz',
                     'message':'Hi'})
        form = MailForm(None, data)
        self.assertTrue(form.is_valid())
        
        mail_obj = form.save()
        c = Client()
        response = c.get('/process', follow=True)
        self.assertContains(response, SUBJECT)
        self.assertEquals(mail.outbox[0].subject, SUBJECT)
        message_id = mail.outbox[0].extra_headers['Message-ID']
        REPLY_SUBJECT = "Re: %s" % SUBJECT
        response = EmailMessage(REPLY_SUBJECT,
                                "Hello back",
                                "*****@*****.**",
                                [user.proxy_email],
                                headers={'In-Reply-To':'<%s>' % message_id})
        response.send()
        self.assertEquals(mail.outbox[-1].subject, REPLY_SUBJECT)
        reply_fp = StringIO(mail.outbox[-1].message().as_string())
        response = make_response_from_file(reply_fp)
        self.assertTrue(response)
        # this was like a different person at the same company
        # replying to the message -- let's check that while we're here
        couser1 = CustomUser.objects.get(email='*****@*****.**')
        couser2 = CustomUser.objects.get(email='*****@*****.**')
        self.assertEqual(couser1.organisation, couser2.organisation)

        # two emails should have been created, threaded together
        mails = Mail.objects.order_by('created')
        self.assertEqual(mails[0].childwalk().next(), mails[1])
        # and an email should have been sent to the original sender,
        # with a proxied reply-to address
        self.assertEqual(mail.outbox[-1].to[0],
                         user.email)
        self.assertEqual(mail.outbox[-1].from_email,
                         couser2.email)
        self.assertEqual(mail.outbox[-1].extra_headers['Reply-To'],
                         couser2.proxy_email)