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
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
|
# coding:utf-8
import zlib
from nose.tools import eq_, ok_, assert_false
from six.moves import StringIO
from flanker.mime.message.headers import MimeHeaders, encoding
from tests import BILINGUAL
def headers_case_insensitivity_test():
h = MimeHeaders()
h['Content-Type'] = 1
eq_(1, h['Content-Type'])
eq_(1, h['conTenT-TyPE'])
ok_('cOnTenT-TyPE' in h)
ok_('Content-Type' in h)
eq_(1, h.get('Content-Type'))
eq_(None, h.get('Content-Type2'))
eq_([('Content-Type', 1)], h.items())
def headers_order_preserved_test():
headers = [('mime-version', '1'), ('rEceived', '2'), ('mime-version', '3'), ('ReceiveD', '4')]
h = MimeHeaders(headers)
# various types of iterations
should_be = [('Mime-Version', '1'), ('Received', '2'), ('Mime-Version', '3'), ('Received', '4')]
eq_(should_be, h.items())
ok_(isinstance(h.items(), list))
eq_(should_be, [p for p in h.iteritems()])
# iterate over keys
keys = ['Mime-Version', 'Received', 'Mime-Version', 'Received']
eq_(keys, [p for p in h])
eq_(keys, list(h.keys()))
def headers_boolean_test():
eq_(False, bool(MimeHeaders()))
eq_(True, bool(MimeHeaders([('A', 1)])))
def headers_to_string_test():
ok_(str(MimeHeaders([('A', 1)])))
def headers_multiple_values_test():
headers = [('mime-version', '1'), ('rEceived', '2'), ('mime-version', '3'), ('ReceiveD', '4')]
h = MimeHeaders(headers)
eq_(['1', '3'], h.getall('Mime-Version'))
# set re-sets all values for the message
h['Mime-Version'] = '5'
eq_(['5'], h.getall('Mime-Version'))
# use add to add more values
h.add('Received', '1')
eq_(['1', '2', '4'], h.getall('Received'))
# use prepend to insert header in the begining of the list
h.prepend('Received', '0')
eq_(['0', '1', '2', '4'], h.getall('Received'))
# delete removes it all!
del h['RECEIVED']
eq_([], h.getall('Received'))
def headers_length_test():
h = MimeHeaders()
eq_(0, len(h))
headers = [('mime-version', '1'), ('rEceived', '2'), ('mime-version', '3'), ('ReceiveD', '4')]
h = MimeHeaders(headers)
eq_(4, len(h))
def headers_alternation_test():
headers = [('mime-version', '1'), ('rEceived', '2'), ('mime-version', '3'), ('ReceiveD', '4')]
h = MimeHeaders(headers)
assert_false(h.have_changed())
h.prepend('Received', 'Yo')
ok_(h.have_changed())
h = MimeHeaders(headers)
del h['Mime-Version']
ok_(h.have_changed())
h = MimeHeaders(headers)
h['Mime-Version'] = 'a'
ok_(h.have_changed())
h = MimeHeaders(headers)
h.add('Mime-Version', 'a')
ok_(h.have_changed())
h = MimeHeaders(headers)
h.getall('Mime-Version')
h.get('o')
assert_false(h.have_changed())
def headers_transform_test():
headers = [('mime-version', '1'), ('rEceived', '2'), ('mime-version', '3'), ('ReceiveD', '4')]
h = MimeHeaders(headers)
# transform tracks whether anything actually changed
h.transform(lambda key,val: (key, val))
assert_false(h.have_changed())
# ok, now something have changed, make sure we've preserved order and did not collapse anything
h.transform(lambda key,val: ("X-{0}".format(key), "t({0})".format(val)))
ok_(h.have_changed())
eq_([('X-Mime-Version', 't(1)'), ('X-Received', 't(2)'), ('X-Mime-Version', 't(3)'), ('X-Received', 't(4)')], h.items())
def headers_transform_encodedword_test():
# Create a header with non-ascii characters that will be stored in encoded-word format.
headers = [('Subject', encoding.to_mime('Subject', u'Hello ✓'))]
h = MimeHeaders(headers)
# transform should decode it for us when we pass decode=True
h.transform(lambda key,val: (key, val.replace(u'✓', u'☃')), decode=True)
eq_(u'Hello ☃', h.get('Subject'))
def headers_parsing_empty_test():
h = MimeHeaders.from_stream(StringIO(""))
eq_(0, len(h))
def headers_parsing_ridiculously_long_line_test():
val = "abcdefg"*100000
header = "Hello: {0}\r\n".format(val)
MimeHeaders.from_stream(StringIO(header))
def headers_parsing_binary_stuff_survives_test():
value = zlib.compress(b"abcdefg")
header = "Hello: {0}\r\n".format(value)
ok_(MimeHeaders.from_stream(StringIO(header)))
def broken_sequences_test():
headers = StringIO(" hello this is a bad header\nGood: this one is ok")
headers = MimeHeaders.from_stream(headers)
eq_(1, len(headers))
eq_("this one is ok", headers["Good"])
def bilingual_message_test():
headers = MimeHeaders.from_stream(StringIO(BILINGUAL))
eq_(21, len(headers))
eq_(u"Simple text. How are you? Как ты поживаешь?", headers['Subject'])
received_headers = headers.getall('Received')
eq_(5, len(received_headers))
ok_('c2cs24435ybk' in received_headers[0])
def headers_roundtrip_test():
headers = MimeHeaders.from_stream(StringIO(BILINGUAL))
out = StringIO()
headers.to_stream(out)
headers2 = MimeHeaders.from_stream(StringIO(out.getvalue()))
eq_(21, len(headers2))
eq_(u"Simple text. How are you? Как ты поживаешь?", headers['Subject'])
received_headers = headers.getall('Received')
eq_(5, len(received_headers))
ok_('c2cs24435ybk' in received_headers[0])
eq_(headers['Content-Transfer-Encoding'],
headers2['Content-Transfer-Encoding'])
eq_(headers['DKIM-Signature'],
headers2['DKIM-Signature'])
def test_folding_combinations():
message = """From mrc@example.com Mon Feb 8 02:53:47 PST 1993\nTo: sasha\r\n continued\n line\nFrom: single line \r\nSubject: hello, how are you\r\n today?"""
headers = MimeHeaders.from_stream(StringIO(message))
eq_('sasha continued line', headers['To'])
eq_('single line ', headers['From'])
eq_("hello, how are you today?", headers['Subject'])
|
