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
92
93
94
95
96
97
98
99
100
101
102
|
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
from email.message import Message
import tempfile
import shutil
import pytest
from muttdown.main import convert_tree
from muttdown.main import process_message
from muttdown.config import Config
@pytest.fixture
def basic_config():
return Config()
@pytest.fixture
def tempdir():
# workaround because pytest's bultin tmpdir fixture is broken on python 3.3
dirname = tempfile.mkdtemp()
try:
yield dirname
finally:
shutil.rmtree(dirname)
@pytest.fixture
def config_with_css(tempdir):
with open('%s/test.css' % tempdir, 'w') as f:
f.write('html, body, p { font-family: serif; }\n')
c = Config()
c.merge_config({'css_file': '%s/test.css' % tempdir})
return c
def test_unmodified_no_match(basic_config):
msg = Message()
msg['Subject'] = 'Test Message'
msg['From'] = 'from@example.com'
msg['To'] = 'to@example.com'
msg['Bcc'] = 'bananas'
msg.set_payload('This message has no sigil')
converted = process_message(msg, basic_config)
assert converted == msg
def test_simple_message(basic_config):
msg = MIMEMultipart()
msg['Subject'] = 'Test Message'
msg['From'] = 'from@example.com'
msg['To'] = 'to@example.com'
msg['Bcc'] = 'bananas'
msg.preamble = 'Outer preamble'
msg.attach(MIMEText("!m This is the main message body"))
attachment = MIMEText('this is an attachment', 'x-misc')
attachment.add_header('Content-Disposition', 'attachment')
msg.attach(attachment)
converted, _ = convert_tree(msg, basic_config)
assert converted['Subject'] == 'Test Message'
assert converted['From'] == 'from@example.com'
assert converted['To'] == 'to@example.com'
assert converted.get('Bcc', None) is None
assert isinstance(converted, MIMEMultipart)
assert converted.preamble == 'Outer preamble'
assert len(converted.get_payload()) == 2
alternatives_part = converted.get_payload()[0]
assert isinstance(alternatives_part, MIMEMultipart)
assert alternatives_part.get_content_type() == 'multipart/alternative'
assert len(alternatives_part.get_payload()) == 2
text_part = alternatives_part.get_payload()[0]
html_part = alternatives_part.get_payload()[1]
assert isinstance(text_part, MIMEText)
assert text_part.get_content_type() == 'text/plain'
assert isinstance(html_part, MIMEText)
assert html_part.get_content_type() == 'text/html'
attachment_part = converted.get_payload()[1]
assert isinstance(attachment_part, MIMEText)
assert attachment_part['Content-Disposition'] == 'attachment'
assert attachment_part.get_content_type() == 'text/x-misc'
def test_with_css(config_with_css):
msg = Message()
msg['Subject'] = 'Test Message'
msg['From'] = 'from@example.com'
msg['To'] = 'to@example.com'
msg['Bcc'] = 'bananas'
msg.set_payload('!m\n\nThis is a message')
converted, _ = convert_tree(msg, config_with_css)
assert isinstance(converted, MIMEMultipart)
assert len(converted.get_payload()) == 2
text_part = converted.get_payload()[0]
assert text_part.get_payload(decode=True) == b'!m\n\nThis is a message'
html_part = converted.get_payload()[1]
assert html_part.get_payload(decode=True) == b'<p style="font-family: serif">This is a message</p>'
|
