Python code coverage for Lib/email/test/test_email_torture.py

#	count	content
1	n/a	# Copyright (C) 2002-2004 Python Software Foundation
2	n/a	#
3	n/a	# A torture test of the email package. This should not be run as part of the
4	n/a	# standard Python test suite since it requires several meg of email messages
5	n/a	# collected in the wild. These source messages are not checked into the
6	n/a	# Python distro, but are available as part of the standalone email package at
7	n/a	# http://sf.net/projects/mimelib
8	n/a
9	n/a	import sys
10	n/a	import os
11	n/a	import unittest
12	n/a	from io import StringIO
13	n/a	from types import ListType
14	n/a
15	n/a	from email.test.test_email import TestEmailBase
16	n/a	from test.support import TestSkipped, run_unittest
17	n/a
18	n/a	import email
19	n/a	from email import __file__ as testfile
20	n/a	from email.iterators import _structure
21	n/a
22	n/a	def openfile(filename):
23	n/a	from os.path import join, dirname, abspath
24	n/a	path = abspath(join(dirname(testfile), os.pardir, 'moredata', filename))
25	n/a	return open(path, 'r')
26	n/a
27	n/a	# Prevent this test from running in the Python distro
28	n/a	try:
29	n/a	openfile('crispin-torture.txt')
30	n/a	except IOError:
31	n/a	raise TestSkipped
32	n/a
33	n/a
34	n/a
35	n/a	class TortureBase(TestEmailBase):
36	n/a	def _msgobj(self, filename):
37	n/a	fp = openfile(filename)
38	n/a	try:
39	n/a	msg = email.message_from_file(fp)
40	n/a	finally:
41	n/a	fp.close()
42	n/a	return msg
43	n/a
44	n/a
45	n/a
46	n/a	class TestCrispinTorture(TortureBase):
47	n/a	# Mark Crispin's torture test from the SquirrelMail project
48	n/a	def test_mondo_message(self):
49	n/a	eq = self.assertEqual
50	n/a	neq = self.ndiffAssertEqual
51	n/a	msg = self._msgobj('crispin-torture.txt')
52	n/a	payload = msg.get_payload()
53	n/a	eq(type(payload), ListType)
54	n/a	eq(len(payload), 12)
55	n/a	eq(msg.preamble, None)
56	n/a	eq(msg.epilogue, '\n')
57	n/a	# Probably the best way to verify the message is parsed correctly is to
58	n/a	# dump its structure and compare it against the known structure.
59	n/a	fp = StringIO()
60	n/a	_structure(msg, fp=fp)
61	n/a	neq(fp.getvalue(), """\
62	n/a	multipart/mixed
63	n/a	text/plain
64	n/a	message/rfc822
65	n/a	multipart/alternative
66	n/a	text/plain
67	n/a	multipart/mixed
68	n/a	text/richtext
69	n/a	application/andrew-inset
70	n/a	message/rfc822
71	n/a	audio/basic
72	n/a	audio/basic
73	n/a	image/pbm
74	n/a	message/rfc822
75	n/a	multipart/mixed
76	n/a	multipart/mixed
77	n/a	text/plain
78	n/a	audio/x-sun
79	n/a	multipart/mixed
80	n/a	image/gif
81	n/a	image/gif
82	n/a	application/x-be2
83	n/a	application/atomicmail
84	n/a	audio/x-sun
85	n/a	message/rfc822
86	n/a	multipart/mixed
87	n/a	text/plain
88	n/a	image/pgm
89	n/a	text/plain
90	n/a	message/rfc822
91	n/a	multipart/mixed
92	n/a	text/plain
93	n/a	image/pbm
94	n/a	message/rfc822
95	n/a	application/postscript
96	n/a	image/gif
97	n/a	message/rfc822
98	n/a	multipart/mixed
99	n/a	audio/basic
100	n/a	audio/basic
101	n/a	message/rfc822
102	n/a	multipart/mixed
103	n/a	application/postscript
104	n/a	text/plain
105	n/a	message/rfc822
106	n/a	multipart/mixed
107	n/a	text/plain
108	n/a	multipart/parallel
109	n/a	image/gif
110	n/a	audio/basic
111	n/a	application/atomicmail
112	n/a	message/rfc822
113	n/a	audio/x-sun
114	n/a	""")
115	n/a
116	n/a
117	n/a	def _testclasses():
118	n/a	mod = sys.modules[__name__]
119	n/a	return [getattr(mod, name) for name in dir(mod) if name.startswith('Test')]
120	n/a
121	n/a
122	n/a	def suite():
123	n/a	suite = unittest.TestSuite()
124	n/a	for testclass in _testclasses():
125	n/a	suite.addTest(unittest.makeSuite(testclass))
126	n/a	return suite
127	n/a
128	n/a
129	n/a	def test_main():
130	n/a	for testclass in _testclasses():
131	n/a	run_unittest(testclass)
132	n/a
133	n/a
134	n/a
135	n/a	if __name__ == '__main__':
136	n/a	unittest.main(defaultTest='suite')