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