Mercurial > cpython-lru_cache-weakref

annotate Lib/test/test_zipimport_support.py @ 54294:dba767bc3155 legacy-trunk

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
remove mention of old ctypes version
author Benjamin Peterson <benjamin@python.org>
date Sat, 23 May 2009 20:59:09 +0000
parents edd72437cec3
children 51102abdec50 6768b2350c57
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
50162
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
1 # This test module covers support in various parts of the standard library
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
2 # for working with modules located inside zipfiles
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
3 # The tests are centralised in this fashion to make it easy to drop them
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
4 # if a platform doesn't support zipimport
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
5 import unittest
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
6 import test.test_support
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
7 import os
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
8 import os.path
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
9 import sys
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
10 import textwrap
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
11 import zipfile
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
12 import zipimport
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
13 import doctest
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
14 import inspect
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
15 import linecache
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
16 import pdb
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
17
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
18 verbose = test.test_support.verbose
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
19
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
20 # Library modules covered by this test set
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
21 # pdb (Issue 4201)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
22 # inspect (Issue 4223)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
23 # doctest (Issue 4197)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
24
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
25 # Other test modules with zipimport related tests
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
26 # test_zipimport (of course!)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
27 # test_cmd_line_script (covers the zipimport support in runpy)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
28
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
29 # Retrieve some helpers from other test cases
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
30 from test import test_doctest, sample_doctest
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
31 from test.test_importhooks import ImportHooksBaseTestCase
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
32 from test.test_cmd_line_script import temp_dir, _run_python, \
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
33 _spawn_python, _kill_python, \
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
34 _make_test_script, \
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
35 _compile_test_script, \
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
36 _make_test_zip, _make_test_pkg
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
37
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
38
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
39 def _run_object_doctest(obj, module):
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
40 # Direct doctest output (normally just errors) to real stdout; doctest
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
41 # output shouldn't be compared by regrtest.
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
42 save_stdout = sys.stdout
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
43 sys.stdout = test.test_support.get_original_stdout()
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
44 try:
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
45 finder = doctest.DocTestFinder(verbose=verbose, recurse=False)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
46 runner = doctest.DocTestRunner(verbose=verbose)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
47 # Use the object's fully qualified name if it has one
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
48 # Otherwise, use the module's name
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
49 try:
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
50 name = "%s.%s" % (obj.__module__, obj.__name__)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
51 except AttributeError:
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
52 name = module.__name__
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
53 for example in finder.find(obj, name, module):
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
54 runner.run(example)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
55 f, t = runner.failures, runner.tries
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
56 if f:
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
57 raise test.test_support.TestFailed("%d of %d doctests failed" % (f, t))
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
58 finally:
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
59 sys.stdout = save_stdout
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
60 if verbose:
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
61 print 'doctest (%s) ... %d tests with zero failures' % (module.__name__, t)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
62 return f, t
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
63
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
64
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
65
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
66 class ZipSupportTests(ImportHooksBaseTestCase):
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
67 # We use the ImportHooksBaseTestCase to restore
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
68 # the state of the import related information
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
69 # in the sys module after each test
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
70 # We also clear the linecache and zipimport cache
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
71 # just to avoid any bogus errors due to name reuse in the tests
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
72 def setUp(self):
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
73 linecache.clearcache()
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
74 zipimport._zip_directory_cache.clear()
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
75 ImportHooksBaseTestCase.setUp(self)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
76
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
77
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
78 def test_inspect_getsource_issue4223(self):
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
79 test_src = "def foo(): pass\n"
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
80 with temp_dir() as d:
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
81 init_name = _make_test_script(d, '__init__', test_src)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
82 name_in_zip = os.path.join('zip_pkg',
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
83 os.path.basename(init_name))
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
84 zip_name, run_name = _make_test_zip(d, 'test_zip',
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
85 init_name, name_in_zip)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
86 os.remove(init_name)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
87 sys.path.insert(0, zip_name)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
88 import zip_pkg
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
89 self.assertEqual(inspect.getsource(zip_pkg.foo), test_src)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
90
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
91 def test_doctest_issue4197(self):
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
92 # To avoid having to keep two copies of the doctest module's
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
93 # unit tests in sync, this test works by taking the source of
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
94 # test_doctest itself, rewriting it a bit to cope with a new
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
95 # location, and then throwing it in a zip file to make sure
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
96 # everything still works correctly
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
97 test_src = inspect.getsource(test_doctest)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
98 test_src = test_src.replace(
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
99 "from test import test_doctest",
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
100 "import test_zipped_doctest as test_doctest")
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
101 test_src = test_src.replace("test.test_doctest",
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
102 "test_zipped_doctest")
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
103 test_src = test_src.replace("test.sample_doctest",
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
104 "sample_zipped_doctest")
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
105 sample_src = inspect.getsource(sample_doctest)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
106 sample_src = sample_src.replace("test.test_doctest",
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
107 "test_zipped_doctest")
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
108 with temp_dir() as d:
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
109 script_name = _make_test_script(d, 'test_zipped_doctest',
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
110 test_src)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
111 zip_name, run_name = _make_test_zip(d, 'test_zip',
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
112 script_name)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
113 z = zipfile.ZipFile(zip_name, 'a')
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
114 z.writestr("sample_zipped_doctest.py", sample_src)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
115 z.close()
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
116 if verbose:
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
117 zip_file = zipfile.ZipFile(zip_name, 'r')
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
118 print 'Contents of %r:' % zip_name
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
119 zip_file.printdir()
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
120 zip_file.close()
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
121 os.remove(script_name)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
122 sys.path.insert(0, zip_name)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
123 import test_zipped_doctest
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
124 # Some of the doc tests depend on the colocated text files
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
125 # which aren't available to the zipped version (the doctest
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
126 # module currently requires real filenames for non-embedded
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
127 # tests). So we're forced to be selective about which tests
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
128 # to run.
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
129 # doctest could really use some APIs which take a text
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
130 # string or a file object instead of a filename...
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
131 known_good_tests = [
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
132 test_zipped_doctest.SampleClass,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
133 test_zipped_doctest.SampleClass.NestedClass,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
134 test_zipped_doctest.SampleClass.NestedClass.__init__,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
135 test_zipped_doctest.SampleClass.__init__,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
136 test_zipped_doctest.SampleClass.a_classmethod,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
137 test_zipped_doctest.SampleClass.a_property,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
138 test_zipped_doctest.SampleClass.a_staticmethod,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
139 test_zipped_doctest.SampleClass.double,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
140 test_zipped_doctest.SampleClass.get,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
141 test_zipped_doctest.SampleNewStyleClass,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
142 test_zipped_doctest.SampleNewStyleClass.__init__,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
143 test_zipped_doctest.SampleNewStyleClass.double,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
144 test_zipped_doctest.SampleNewStyleClass.get,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
145 test_zipped_doctest.old_test1,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
146 test_zipped_doctest.old_test2,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
147 test_zipped_doctest.old_test3,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
148 test_zipped_doctest.old_test4,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
149 test_zipped_doctest.sample_func,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
150 test_zipped_doctest.test_DocTest,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
151 test_zipped_doctest.test_DocTestParser,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
152 test_zipped_doctest.test_DocTestRunner.basics,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
153 test_zipped_doctest.test_DocTestRunner.exceptions,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
154 test_zipped_doctest.test_DocTestRunner.option_directives,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
155 test_zipped_doctest.test_DocTestRunner.optionflags,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
156 test_zipped_doctest.test_DocTestRunner.verbose_flag,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
157 test_zipped_doctest.test_Example,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
158 test_zipped_doctest.test_debug,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
159 test_zipped_doctest.test_pdb_set_trace,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
160 test_zipped_doctest.test_pdb_set_trace_nested,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
161 test_zipped_doctest.test_testsource,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
162 test_zipped_doctest.test_trailing_space_in_test,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
163 test_zipped_doctest.test_DocTestSuite,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
164 test_zipped_doctest.test_DocTestFinder,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
165 ]
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
166 # These remaining tests are the ones which need access
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
167 # to the data files, so we don't run them
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
168 fail_due_to_missing_data_files = [
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
169 test_zipped_doctest.test_DocFileSuite,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
170 test_zipped_doctest.test_testfile,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
171 test_zipped_doctest.test_unittest_reportflags,
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
172 ]
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
173 for obj in known_good_tests:
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
174 _run_object_doctest(obj, test_zipped_doctest)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
175
50185
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
176 def test_doctest_main_issue4197(self):
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
177 test_src = textwrap.dedent("""\
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
178 class Test:
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
179 ">>> 'line 2'"
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
180 pass
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
181
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
182 import doctest
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
183 doctest.testmod()
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
184 """)
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
185 pattern = 'File "%s", line 2, in %s'
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
186 with temp_dir() as d:
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
187 script_name = _make_test_script(d, 'script', test_src)
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
188 exit_code, data = _run_python(script_name)
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
189 expected = pattern % (script_name, "__main__.Test")
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
190 if verbose:
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
191 print "Expected line", expected
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
192 print "Got stdout:"
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
193 print data
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
194 self.assert_(expected in data)
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
195 zip_name, run_name = _make_test_zip(d, "test_zip",
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
196 script_name, '__main__.py')
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
197 exit_code, data = _run_python(zip_name)
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
198 expected = pattern % (run_name, "__main__.Test")
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
199 if verbose:
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
200 print "Expected line", expected
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
201 print "Got stdout:"
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
202 print data
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
203 self.assert_(expected in data)
edd72437cec3 Merged revisions 67790 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents: 50162
diff changeset
204
50162
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
205 def test_pdb_issue4201(self):
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
206 test_src = textwrap.dedent("""\
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
207 def f():
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
208 pass
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
209
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
210 import pdb
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
211 pdb.runcall(f)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
212 """)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
213 with temp_dir() as d:
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
214 script_name = _make_test_script(d, 'script', test_src)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
215 p = _spawn_python(script_name)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
216 p.stdin.write('l\n')
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
217 data = _kill_python(p)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
218 self.assert_(script_name in data)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
219 zip_name, run_name = _make_test_zip(d, "test_zip",
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
220 script_name, '__main__.py')
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
221 p = _spawn_python(zip_name)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
222 p.stdin.write('l\n')
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
223 data = _kill_python(p)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
224 self.assert_(run_name in data)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
225
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
226
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
227 def test_main():
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
228 test.test_support.run_unittest(ZipSupportTests)
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
229 test.test_support.reap_children()
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
230
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
231 if __name__ == '__main__':
4264cf00a75c Merged revisions 67750-67751 via svnmerge from
Nick Coghlan <ncoghlan@gmail.com>
parents:
diff changeset
232 test_main()