blob: 9eac5e5849cef349c8e7eb1c0c02c8659ff057dc (
plain)
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
|
"""
Tests for leap.mail.walk module
"""
import os.path
from email.parser import Parser
from leap.bitmask.mail import walk
CORPUS = {
'simple': 'rfc822.message',
'multimin': 'rfc822.multi-minimal.message',
'multisigned': 'rfc822.multi-signed.message',
'bounced': 'rfc822.bounce.message',
}
_here = os.path.dirname(__file__)
_parser = Parser()
# tests
def test_simple_mail():
msg = _parse('simple')
tree = walk.get_tree(msg)
assert len(tree['part_map']) == 0
assert tree['ctype'] == 'text/plain'
assert tree['multi'] is False
def test_multipart_minimal():
msg = _parse('multimin')
tree = walk.get_tree(msg)
assert tree['multi'] is True
assert len(tree['part_map']) == 1
first = tree['part_map'][1]
assert first['multi'] is False
assert first['ctype'] == 'text/plain'
def test_multi_signed():
msg = _parse('multisigned')
tree = walk.get_tree(msg)
assert tree['multi'] is True
assert len(tree['part_map']) == 2
_first = tree['part_map'][1]
_second = tree['part_map'][2]
assert len(_first['part_map']) == 3
assert(_second['multi'] is False)
def test_bounce_mime():
msg = _parse('bounced')
tree = walk.get_tree(msg)
ctypes = [tree['part_map'][index]['ctype']
for index in sorted(tree['part_map'].keys())]
third = tree['part_map'][3]
three_one_ctype = third['part_map'][1]['ctype']
assert three_one_ctype == 'multipart/signed'
assert ctypes == [
'text/plain',
'message/delivery-status',
'message/rfc822']
# utils
def _parse(name):
_str = _get_string_for_message(name)
return _parser.parsestr(_str)
def _get_string_for_message(name):
filename = os.path.join(_here, CORPUS[name])
with open(filename) as f:
msgstr = f.read()
return msgstr
|
