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_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_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 collect(message):
collected = deque()
for p in message.walk(with_self=True):
for h in HEADERS:
if h in p.headers:
collected.append((h, p.headers[h]))
if p.content_type.is_delivery_status():
collected += collect_from_status(p.body)
return MimeHeaders(collected)
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_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_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_to_string_test():
ok_(str(MimeHeaders([('A', 1)])))
def headers_boolean_test():
eq_(False, bool(MimeHeaders()))
eq_(True, bool(MimeHeaders([('A', 1)])))
