[bfa3c8]: Allura / allura / tests / test_mail_util.py History

Parent: [c98680] (diff)

Child: [e5b93e] (diff)

Download this file

test_mail_util.py    93 lines (78 with data), 3.4 kB 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
# -*- coding: utf-8 -*-

import unittest

from email.MIMEMultipart import MIMEMultipart

from email.MIMEText import MIMEText

from email import header

from email.parser import Parser



import tg

from nose.tools import raises, assert_equal

from ming.orm import ThreadLocalORMSession



from alluratest.controller import setup_basic_test, setup_global_objects

from allura.lib.utils import ConfigProxy



from allura.lib.mail_util import parse_address, parse_message

from allura.lib.exceptions import AddressException

from allura.tasks.mail_tasks import route_email



config = ConfigProxy(

    common_suffix='forgemail.domain',

    return_path='forgemail.return_path')



class TestReactor(unittest.TestCase):



    def setUp(self):

        setup_basic_test()

        setup_global_objects()

        ThreadLocalORMSession.flush_all()

        ThreadLocalORMSession.close_all()



    @raises(AddressException)

    def test_parse_address_bad_domain(self):

        parse_address('foo@bar.com')



    @raises(AddressException)

    def test_parse_address_bad_project(self):

        parse_address('foo@wiki.unicorns.p' + config.common_suffix)



    @raises(AddressException)

    def test_parse_address_missing_tool(self):

        parse_address('foo@test.p' + config.common_suffix)



    @raises(AddressException)

    def test_parse_address_bad_tool(self):

        parse_address('foo@hammer.test.p' + config.common_suffix)



    def test_parse_address_good(self):

        topic, project, app = parse_address('foo@wiki.test.p' + config.common_suffix)

        assert_equal(topic, 'foo')

        assert_equal(project.name, 'test')

        assert_equal(app.__class__.__name__, 'ForgeWikiApp')



    def test_unicode_simple_message(self):

        charset = 'utf-8'

        msg1 = MIMEText(u'''���� �������������������� ��������������

�������������� ���������������� ����������������

�������������� �� ����������; ��������������

������������ ���� �������� ������������ ����������

�� �������������� ������������������ ������������������;'''.encode(charset),

                        'plain',

                        charset)

        msg1['Message-ID'] = '<foo@bar.com>'

        s_msg = msg1.as_string()

        msg2 = parse_message(s_msg)

        assert isinstance(msg2['payload'], unicode)



    def test_unicode_complex_message(self):

        charset = 'utf-8'

        p1 = MIMEText(u'''���� �������������������� ��������������

�������������� ���������������� ����������������

�������������� �� ����������; ��������������

������������ ���� �������� ������������ ����������

�� �������������� ������������������ ������������������;'''.encode(charset),

                        'plain',

                        charset)

        p2 = MIMEText(u'''<p>���� �������������������� ��������������

�������������� ���������������� ����������������

�������������� �� ����������; ��������������

������������ ���� �������� ������������ ����������

�� �������������� ������������������ ������������������;</p>'''.encode(charset),

                        'plain',

                        charset)

        msg1 = MIMEMultipart()

        msg1['Message-ID'] = '<foo@bar.com>'

        msg1.attach(p1)

        msg1.attach(p2)

        s_msg = msg1.as_string()

        msg2 = parse_message(s_msg)

        for part in msg2['parts']:

            if part['payload'] is None: continue

            assert isinstance(part['payload'], unicode)