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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
# Copyright © 2017 Lucas Hoffmann
# Copyright © 2018 Dylan Baker
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
import email.parser
import email.policy
import os
import tempfile
import unittest
from unittest import mock
import sys
from alot.db import envelope
from alot.account import Account
SETTINGS = {
'user_agent': 'agent',
}
test_account = Account('batman@batcave.org', message_id_domain='example.com')
class TestEnvelope(unittest.TestCase):
def assertEmailEqual(self, first, second):
with self.subTest('body'):
self.assertEqual(first.is_multipart(), second.is_multipart())
if not first.is_multipart():
self.assertEqual(first.get_payload(), second.get_payload())
else:
for f, s in zip(first.walk(), second.walk()):
if f.is_multipart() or s.is_multipart():
self.assertEqual(first.is_multipart(),
second.is_multipart())
else:
self.assertEqual(f.get_payload(), s.get_payload())
with self.subTest('headers'):
self.assertListEqual(first.values(), second.values())
def test_setitem_stores_text_unchanged(self):
"Just ensure that the value is set and unchanged"
e = envelope.Envelope(account=test_account)
e['Subject'] = 'sm\xf8rebr\xf8d'
self.assertEqual(e['Subject'], 'sm\xf8rebr\xf8d')
def _test_mail(self, envelope):
mail = envelope.construct_mail()
raw = mail.as_string(policy=email.policy.SMTP)
actual = email.parser.Parser().parsestr(raw)
self.assertEmailEqual(mail, actual)
@mock.patch('alot.db.envelope.settings', SETTINGS)
def test_construct_mail_simple(self):
"""Very simple envelope with a To, From, Subject, and body."""
headers = {
'From': 'foo@example.com',
'To': 'bar@example.com',
'Subject': 'Test email',
}
e = envelope.Envelope(account=test_account,
headers={k: [v] for k, v in headers.items()},
bodytext='Test')
self._test_mail(e)
@mock.patch('alot.db.envelope.settings', SETTINGS)
def test_construct_mail_with_attachment(self):
"""Very simple envelope with a To, From, Subject, body and attachment.
"""
headers = {
'From': 'foo@example.com',
'To': 'bar@example.com',
'Subject': 'Test email',
}
e = envelope.Envelope(account=test_account,
headers={k: [v] for k, v in headers.items()},
bodytext='Test')
with tempfile.NamedTemporaryFile(mode='wt', delete=False) as f:
f.write('blah')
self.addCleanup(os.unlink, f.name)
e.attach(f.name)
self._test_mail(e)
@mock.patch('alot.db.envelope.settings', SETTINGS)
def test_parse_template(self):
"""Tests multi-line header and body parsing"""
raw = (
'From: foo@example.com\n'
'To: bar@example.com,\n'
' baz@example.com\n'
'Subject: Fwd: Test email\n'
'\n'
'Some body content: which is not a header.\n'
)
envlp = envelope.Envelope(account=test_account)
envlp.parse_template(raw)
self.assertDictEqual(envlp.headers, {
'From': ['foo@example.com'],
'To': ['bar@example.com, baz@example.com'],
'Subject': ['Fwd: Test email']
})
self.assertEqual(envlp.body_txt,
'Some body content: which is not a header.')
@mock.patch('alot.db.envelope.settings', SETTINGS)
def test_construct_encoding(self):
headers = {
'From': 'foo@example.com',
'To': 'bar@example.com',
'Subject': 'Test email héhé',
}
e = envelope.Envelope(account=test_account,
headers={k: [v] for k, v in headers.items()},
bodytext='Test')
mail = e.construct_mail()
raw = mail.as_string(policy=email.policy.SMTP, maxheaderlen=sys.maxsize)
actual = email.parser.Parser().parsestr(raw)
self.assertEqual('Test email =?utf-8?b?aMOpaMOp?=', actual['Subject'])
|
