# -*- 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 forgemail.lib.util import parse_address, parse_message
from forgemail.lib.exc import AddressException
from forgemail.reactors.common_react import received_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, 'Wiki.msg.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)
def test_malformed_email_no_exception(self):
msg = MIMEText('Bad email, no Message-ID')
received_email('', msg.as_string())