blob: 105116a5c00b48a85e93646be7564f6d495a204b [file] [log] [blame]
Georg Brandlb533e262008-05-25 18:19:30 +00001import os
Nick Coghlan7bb30b72010-12-03 09:29:11 +00002import sys
Éric Araujoe64e51b2011-07-29 17:03:55 +02003import builtins
Antoine Pitrou916fc7b2013-05-19 15:44:54 +02004import contextlib
Georg Brandlb533e262008-05-25 18:19:30 +00005import difflib
Georg Brandlb533e262008-05-25 18:19:30 +00006import inspect
Nick Coghlan7bb30b72010-12-03 09:29:11 +00007import pydoc
Ezio Melottib185a042011-04-28 07:42:55 +03008import keyword
Antoine Pitrou916fc7b2013-05-19 15:44:54 +02009import pkgutil
Nick Coghlan7bb30b72010-12-03 09:29:11 +000010import re
11import string
Georg Brandlb533e262008-05-25 18:19:30 +000012import test.support
Nick Coghlan7bb30b72010-12-03 09:29:11 +000013import time
Ethan Furmanb0c84cd2013-10-20 22:37:39 -070014import types
Nick Coghlan7bb30b72010-12-03 09:29:11 +000015import unittest
Brian Curtin49c284c2010-03-31 03:19:28 +000016import xml.etree
Georg Brandld80d5f42010-12-03 07:47:22 +000017import textwrap
18from io import StringIO
Raymond Hettinger1103d052011-03-25 14:15:24 -070019from collections import namedtuple
Antoine Pitrouf7f54752011-07-15 22:42:12 +020020from test.script_helper import assert_python_ok
Antoine Pitroua6e81a22011-07-15 22:32:25 +020021from test.support import (
Ned Deily92a81a12011-10-06 14:19:03 -070022 TESTFN, rmtree,
Antoine Pitrou916fc7b2013-05-19 15:44:54 +020023 reap_children, reap_threads, captured_output, captured_stdout,
Stefan Krah5de32782014-01-18 23:18:39 +010024 captured_stderr, unlink, requires_docstrings
Antoine Pitroua6e81a22011-07-15 22:32:25 +020025)
Georg Brandlb533e262008-05-25 18:19:30 +000026from test import pydoc_mod
27
Victor Stinner62a68f22011-05-20 02:29:13 +020028try:
29 import threading
30except ImportError:
31 threading = None
32
Serhiy Storchaka9d0add02013-01-27 19:47:45 +020033if test.support.HAVE_DOCSTRINGS:
34 expected_data_docstrings = (
35 'dictionary for instance variables (if defined)',
36 'list of weak references to the object (if defined)',
37 ) * 2
38else:
39 expected_data_docstrings = ('', '', '', '')
40
Barry Warsaw28a691b2010-04-17 00:19:56 +000041expected_text_pattern = """
Georg Brandlb533e262008-05-25 18:19:30 +000042NAME
43 test.pydoc_mod - This is a test module for test_pydoc
Georg Brandlb533e262008-05-25 18:19:30 +000044%s
45CLASSES
46 builtins.object
47 A
48 B
49\x20\x20\x20\x20
50 class A(builtins.object)
51 | Hello and goodbye
52 |\x20\x20
53 | Methods defined here:
54 |\x20\x20
55 | __init__()
56 | Wow, I have no function!
57 |\x20\x20
58 | ----------------------------------------------------------------------
59 | Data descriptors defined here:
60 |\x20\x20
Serhiy Storchaka9d0add02013-01-27 19:47:45 +020061 | __dict__%s
Georg Brandlb533e262008-05-25 18:19:30 +000062 |\x20\x20
Serhiy Storchaka9d0add02013-01-27 19:47:45 +020063 | __weakref__%s
Georg Brandlb533e262008-05-25 18:19:30 +000064\x20\x20\x20\x20
65 class B(builtins.object)
66 | Data descriptors defined here:
67 |\x20\x20
Serhiy Storchaka9d0add02013-01-27 19:47:45 +020068 | __dict__%s
Georg Brandlb533e262008-05-25 18:19:30 +000069 |\x20\x20
Serhiy Storchaka9d0add02013-01-27 19:47:45 +020070 | __weakref__%s
Georg Brandlb533e262008-05-25 18:19:30 +000071 |\x20\x20
72 | ----------------------------------------------------------------------
73 | Data and other attributes defined here:
74 |\x20\x20
75 | NO_MEANING = 'eggs'
76
77FUNCTIONS
78 doc_func()
79 This function solves all of the world's problems:
80 hunger
81 lack of Python
82 war
83\x20\x20\x20\x20
84 nodoc_func()
85
86DATA
Alexander Belopolskya47bbf52010-11-18 01:52:54 +000087 __xyz__ = 'X, Y and Z'
Georg Brandlb533e262008-05-25 18:19:30 +000088
89VERSION
90 1.2.3.4
91
92AUTHOR
93 Benjamin Peterson
94
95CREDITS
96 Nobody
Alexander Belopolskya47bbf52010-11-18 01:52:54 +000097
98FILE
99 %s
Georg Brandlb533e262008-05-25 18:19:30 +0000100""".strip()
101
Serhiy Storchaka9d0add02013-01-27 19:47:45 +0200102expected_text_data_docstrings = tuple('\n | ' + s if s else ''
103 for s in expected_data_docstrings)
104
Barry Warsaw28a691b2010-04-17 00:19:56 +0000105expected_html_pattern = """
Georg Brandlb533e262008-05-25 18:19:30 +0000106<table width="100%%" cellspacing=0 cellpadding=2 border=0 summary="heading">
107<tr bgcolor="#7799ee">
108<td valign=bottom>&nbsp;<br>
109<font color="#ffffff" face="helvetica, arial">&nbsp;<br><big><big><strong><a href="test.html"><font color="#ffffff">test</font></a>.pydoc_mod</strong></big></big> (version 1.2.3.4)</font></td
110><td align=right valign=bottom
111><font color="#ffffff" face="helvetica, arial"><a href=".">index</a><br><a href="file:%s">%s</a>%s</font></td></tr></table>
112 <p><tt>This&nbsp;is&nbsp;a&nbsp;test&nbsp;module&nbsp;for&nbsp;test_pydoc</tt></p>
113<p>
114<table width="100%%" cellspacing=0 cellpadding=2 border=0 summary="section">
115<tr bgcolor="#ee77aa">
116<td colspan=3 valign=bottom>&nbsp;<br>
117<font color="#ffffff" face="helvetica, arial"><big><strong>Classes</strong></big></font></td></tr>
118\x20\x20\x20\x20
119<tr><td bgcolor="#ee77aa"><tt>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</tt></td><td>&nbsp;</td>
120<td width="100%%"><dl>
121<dt><font face="helvetica, arial"><a href="builtins.html#object">builtins.object</a>
122</font></dt><dd>
123<dl>
124<dt><font face="helvetica, arial"><a href="test.pydoc_mod.html#A">A</a>
125</font></dt><dt><font face="helvetica, arial"><a href="test.pydoc_mod.html#B">B</a>
126</font></dt></dl>
127</dd>
128</dl>
129 <p>
130<table width="100%%" cellspacing=0 cellpadding=2 border=0 summary="section">
131<tr bgcolor="#ffc8d8">
132<td colspan=3 valign=bottom>&nbsp;<br>
133<font color="#000000" face="helvetica, arial"><a name="A">class <strong>A</strong></a>(<a href="builtins.html#object">builtins.object</a>)</font></td></tr>
134\x20\x20\x20\x20
135<tr bgcolor="#ffc8d8"><td rowspan=2><tt>&nbsp;&nbsp;&nbsp;</tt></td>
136<td colspan=2><tt>Hello&nbsp;and&nbsp;goodbye<br>&nbsp;</tt></td></tr>
137<tr><td>&nbsp;</td>
138<td width="100%%">Methods defined here:<br>
139<dl><dt><a name="A-__init__"><strong>__init__</strong></a>()</dt><dd><tt>Wow,&nbsp;I&nbsp;have&nbsp;no&nbsp;function!</tt></dd></dl>
140
141<hr>
142Data descriptors defined here:<br>
143<dl><dt><strong>__dict__</strong></dt>
Serhiy Storchaka9d0add02013-01-27 19:47:45 +0200144<dd><tt>%s</tt></dd>
Georg Brandlb533e262008-05-25 18:19:30 +0000145</dl>
146<dl><dt><strong>__weakref__</strong></dt>
Serhiy Storchaka9d0add02013-01-27 19:47:45 +0200147<dd><tt>%s</tt></dd>
Georg Brandlb533e262008-05-25 18:19:30 +0000148</dl>
149</td></tr></table> <p>
150<table width="100%%" cellspacing=0 cellpadding=2 border=0 summary="section">
151<tr bgcolor="#ffc8d8">
152<td colspan=3 valign=bottom>&nbsp;<br>
153<font color="#000000" face="helvetica, arial"><a name="B">class <strong>B</strong></a>(<a href="builtins.html#object">builtins.object</a>)</font></td></tr>
154\x20\x20\x20\x20
155<tr><td bgcolor="#ffc8d8"><tt>&nbsp;&nbsp;&nbsp;</tt></td><td>&nbsp;</td>
156<td width="100%%">Data descriptors defined here:<br>
157<dl><dt><strong>__dict__</strong></dt>
Serhiy Storchaka9d0add02013-01-27 19:47:45 +0200158<dd><tt>%s</tt></dd>
Georg Brandlb533e262008-05-25 18:19:30 +0000159</dl>
160<dl><dt><strong>__weakref__</strong></dt>
Serhiy Storchaka9d0add02013-01-27 19:47:45 +0200161<dd><tt>%s</tt></dd>
Georg Brandlb533e262008-05-25 18:19:30 +0000162</dl>
163<hr>
164Data and other attributes defined here:<br>
165<dl><dt><strong>NO_MEANING</strong> = 'eggs'</dl>
166
167</td></tr></table></td></tr></table><p>
168<table width="100%%" cellspacing=0 cellpadding=2 border=0 summary="section">
169<tr bgcolor="#eeaa77">
170<td colspan=3 valign=bottom>&nbsp;<br>
171<font color="#ffffff" face="helvetica, arial"><big><strong>Functions</strong></big></font></td></tr>
172\x20\x20\x20\x20
173<tr><td bgcolor="#eeaa77"><tt>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</tt></td><td>&nbsp;</td>
174<td width="100%%"><dl><dt><a name="-doc_func"><strong>doc_func</strong></a>()</dt><dd><tt>This&nbsp;function&nbsp;solves&nbsp;all&nbsp;of&nbsp;the&nbsp;world's&nbsp;problems:<br>
175hunger<br>
176lack&nbsp;of&nbsp;Python<br>
177war</tt></dd></dl>
178 <dl><dt><a name="-nodoc_func"><strong>nodoc_func</strong></a>()</dt></dl>
179</td></tr></table><p>
180<table width="100%%" cellspacing=0 cellpadding=2 border=0 summary="section">
181<tr bgcolor="#55aa55">
182<td colspan=3 valign=bottom>&nbsp;<br>
183<font color="#ffffff" face="helvetica, arial"><big><strong>Data</strong></big></font></td></tr>
184\x20\x20\x20\x20
185<tr><td bgcolor="#55aa55"><tt>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</tt></td><td>&nbsp;</td>
Alexander Belopolskya47bbf52010-11-18 01:52:54 +0000186<td width="100%%"><strong>__xyz__</strong> = 'X, Y and Z'</td></tr></table><p>
Georg Brandlb533e262008-05-25 18:19:30 +0000187<table width="100%%" cellspacing=0 cellpadding=2 border=0 summary="section">
188<tr bgcolor="#7799ee">
189<td colspan=3 valign=bottom>&nbsp;<br>
190<font color="#ffffff" face="helvetica, arial"><big><strong>Author</strong></big></font></td></tr>
191\x20\x20\x20\x20
192<tr><td bgcolor="#7799ee"><tt>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</tt></td><td>&nbsp;</td>
193<td width="100%%">Benjamin&nbsp;Peterson</td></tr></table><p>
194<table width="100%%" cellspacing=0 cellpadding=2 border=0 summary="section">
195<tr bgcolor="#7799ee">
196<td colspan=3 valign=bottom>&nbsp;<br>
197<font color="#ffffff" face="helvetica, arial"><big><strong>Credits</strong></big></font></td></tr>
198\x20\x20\x20\x20
199<tr><td bgcolor="#7799ee"><tt>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</tt></td><td>&nbsp;</td>
200<td width="100%%">Nobody</td></tr></table>
Barry Warsaw28a691b2010-04-17 00:19:56 +0000201""".strip() # ' <- emacs turd
Georg Brandlb533e262008-05-25 18:19:30 +0000202
Serhiy Storchaka9d0add02013-01-27 19:47:45 +0200203expected_html_data_docstrings = tuple(s.replace(' ', '&nbsp;')
204 for s in expected_data_docstrings)
Georg Brandlb533e262008-05-25 18:19:30 +0000205
206# output pattern for missing module
207missing_pattern = "no Python documentation found for '%s'"
208
Benjamin Peterson0289b152009-06-28 17:22:03 +0000209# output pattern for module with bad imports
Brett Cannon679ecb52013-07-04 17:51:50 -0400210badimport_pattern = "problem in %s - ImportError: No module named %r"
Benjamin Peterson0289b152009-06-28 17:22:03 +0000211
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700212expected_dynamicattribute_pattern = """
213Help on class DA in module %s:
214
215class DA(builtins.object)
216 | Data descriptors defined here:
Charles-François Natali1a82f7e2013-10-21 14:46:34 +0200217 |\x20\x20
Ethan Furman3f2f1922013-10-22 07:30:24 -0700218 | __dict__%s
Charles-François Natali1a82f7e2013-10-21 14:46:34 +0200219 |\x20\x20
Ethan Furman3f2f1922013-10-22 07:30:24 -0700220 | __weakref__%s
Charles-François Natali1a82f7e2013-10-21 14:46:34 +0200221 |\x20\x20
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700222 | ham
Charles-François Natali1a82f7e2013-10-21 14:46:34 +0200223 |\x20\x20
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700224 | ----------------------------------------------------------------------
225 | Data and other attributes inherited from Meta:
Charles-François Natali1a82f7e2013-10-21 14:46:34 +0200226 |\x20\x20
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700227 | ham = 'spam'
228""".strip()
229
230expected_virtualattribute_pattern1 = """
231Help on class Class in module %s:
232
233class Class(builtins.object)
234 | Data and other attributes inherited from Meta:
Charles-François Natali1a82f7e2013-10-21 14:46:34 +0200235 |\x20\x20
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700236 | LIFE = 42
237""".strip()
238
239expected_virtualattribute_pattern2 = """
240Help on class Class1 in module %s:
241
242class Class1(builtins.object)
243 | Data and other attributes inherited from Meta1:
Charles-François Natali1a82f7e2013-10-21 14:46:34 +0200244 |\x20\x20
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700245 | one = 1
246""".strip()
247
248expected_virtualattribute_pattern3 = """
249Help on class Class2 in module %s:
250
251class Class2(Class1)
252 | Method resolution order:
253 | Class2
254 | Class1
255 | builtins.object
Charles-François Natali1a82f7e2013-10-21 14:46:34 +0200256 |\x20\x20
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700257 | Data and other attributes inherited from Meta1:
Charles-François Natali1a82f7e2013-10-21 14:46:34 +0200258 |\x20\x20
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700259 | one = 1
Charles-François Natali1a82f7e2013-10-21 14:46:34 +0200260 |\x20\x20
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700261 | ----------------------------------------------------------------------
262 | Data and other attributes inherited from Meta3:
Charles-François Natali1a82f7e2013-10-21 14:46:34 +0200263 |\x20\x20
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700264 | three = 3
Charles-François Natali1a82f7e2013-10-21 14:46:34 +0200265 |\x20\x20
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700266 | ----------------------------------------------------------------------
267 | Data and other attributes inherited from Meta2:
Charles-François Natali1a82f7e2013-10-21 14:46:34 +0200268 |\x20\x20
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700269 | two = 2
270""".strip()
271
272expected_missingattribute_pattern = """
273Help on class C in module %s:
274
275class C(builtins.object)
276 | Data and other attributes defined here:
Charles-François Natali1a82f7e2013-10-21 14:46:34 +0200277 |\x20\x20
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700278 | here = 'present!'
279""".strip()
280
Antoine Pitrouf7f54752011-07-15 22:42:12 +0200281def run_pydoc(module_name, *args, **env):
Georg Brandlb533e262008-05-25 18:19:30 +0000282 """
283 Runs pydoc on the specified module. Returns the stripped
284 output of pydoc.
285 """
Antoine Pitrouf7f54752011-07-15 22:42:12 +0200286 args = args + (module_name,)
Ned Deily92a81a12011-10-06 14:19:03 -0700287 # do not write bytecode files to avoid caching errors
288 rc, out, err = assert_python_ok('-B', pydoc.__file__, *args, **env)
Antoine Pitrouf7f54752011-07-15 22:42:12 +0200289 return out.strip()
Georg Brandlb533e262008-05-25 18:19:30 +0000290
291def get_pydoc_html(module):
292 "Returns pydoc generated output as html"
293 doc = pydoc.HTMLDoc()
294 output = doc.docmodule(module)
295 loc = doc.getdocloc(pydoc_mod) or ""
296 if loc:
297 loc = "<br><a href=\"" + loc + "\">Module Docs</a>"
298 return output.strip(), loc
299
300def get_pydoc_text(module):
301 "Returns pydoc generated output as text"
302 doc = pydoc.TextDoc()
303 loc = doc.getdocloc(pydoc_mod) or ""
304 if loc:
305 loc = "\nMODULE DOCS\n " + loc + "\n"
306
307 output = doc.docmodule(module)
308
Nick Coghlan7bb30b72010-12-03 09:29:11 +0000309 # clean up the extra text formatting that pydoc performs
Georg Brandlb533e262008-05-25 18:19:30 +0000310 patt = re.compile('\b.')
311 output = patt.sub('', output)
312 return output.strip(), loc
313
314def print_diffs(text1, text2):
315 "Prints unified diffs for two texts"
Nick Coghlan7bb30b72010-12-03 09:29:11 +0000316 # XXX now obsolete, use unittest built-in support
Ezio Melottid8b509b2011-09-28 17:37:55 +0300317 lines1 = text1.splitlines(keepends=True)
318 lines2 = text2.splitlines(keepends=True)
Georg Brandlb533e262008-05-25 18:19:30 +0000319 diffs = difflib.unified_diff(lines1, lines2, n=0, fromfile='expected',
320 tofile='got')
321 print('\n' + ''.join(diffs))
322
Nick Coghlan7bb30b72010-12-03 09:29:11 +0000323def get_html_title(text):
Nick Coghlanecace282010-12-03 16:08:46 +0000324 # Bit of hack, but good enough for test purposes
325 header, _, _ = text.partition("</head>")
326 _, _, title = header.partition("<title>")
327 title, _, _ = title.partition("</title>")
Nick Coghlan7bb30b72010-12-03 09:29:11 +0000328 return title
329
Georg Brandlb533e262008-05-25 18:19:30 +0000330
Antoine Pitrou916fc7b2013-05-19 15:44:54 +0200331class PydocBaseTest(unittest.TestCase):
332
333 def _restricted_walk_packages(self, walk_packages, path=None):
334 """
335 A version of pkgutil.walk_packages() that will restrict itself to
336 a given path.
337 """
338 default_path = path or [os.path.dirname(__file__)]
339 def wrapper(path=None, prefix='', onerror=None):
340 return walk_packages(path or default_path, prefix, onerror)
341 return wrapper
342
343 @contextlib.contextmanager
344 def restrict_walk_packages(self, path=None):
345 walk_packages = pkgutil.walk_packages
346 pkgutil.walk_packages = self._restricted_walk_packages(walk_packages,
347 path)
348 try:
349 yield
350 finally:
351 pkgutil.walk_packages = walk_packages
352
353
Georg Brandld2f38572011-01-30 08:37:19 +0000354class PydocDocTest(unittest.TestCase):
Georg Brandlb533e262008-05-25 18:19:30 +0000355
R. David Murray378c0cf2010-02-24 01:46:21 +0000356 @unittest.skipIf(sys.flags.optimize >= 2,
357 "Docstrings are omitted with -O2 and above")
Brett Cannon7a540732011-02-22 03:04:06 +0000358 @unittest.skipIf(hasattr(sys, 'gettrace') and sys.gettrace(),
359 'trace function introduces __locals__ unexpectedly')
Georg Brandlb533e262008-05-25 18:19:30 +0000360 def test_html_doc(self):
361 result, doc_loc = get_pydoc_html(pydoc_mod)
362 mod_file = inspect.getabsfile(pydoc_mod)
Benjamin Petersonc5e94642008-06-14 23:04:46 +0000363 if sys.platform == 'win32':
364 import nturl2path
365 mod_url = nturl2path.pathname2url(mod_file)
366 else:
367 mod_url = mod_file
Serhiy Storchaka9d0add02013-01-27 19:47:45 +0200368 expected_html = expected_html_pattern % (
369 (mod_url, mod_file, doc_loc) +
370 expected_html_data_docstrings)
Georg Brandlb533e262008-05-25 18:19:30 +0000371 if result != expected_html:
372 print_diffs(expected_html, result)
373 self.fail("outputs are not equal, see diff above")
374
R. David Murray378c0cf2010-02-24 01:46:21 +0000375 @unittest.skipIf(sys.flags.optimize >= 2,
376 "Docstrings are omitted with -O2 and above")
Brett Cannon7a540732011-02-22 03:04:06 +0000377 @unittest.skipIf(hasattr(sys, 'gettrace') and sys.gettrace(),
378 'trace function introduces __locals__ unexpectedly')
Georg Brandlb533e262008-05-25 18:19:30 +0000379 def test_text_doc(self):
380 result, doc_loc = get_pydoc_text(pydoc_mod)
Serhiy Storchaka9d0add02013-01-27 19:47:45 +0200381 expected_text = expected_text_pattern % (
382 (doc_loc,) +
383 expected_text_data_docstrings +
384 (inspect.getabsfile(pydoc_mod),))
Georg Brandlb533e262008-05-25 18:19:30 +0000385 if result != expected_text:
386 print_diffs(expected_text, result)
387 self.fail("outputs are not equal, see diff above")
388
Brian Curtin49c284c2010-03-31 03:19:28 +0000389 def test_issue8225(self):
390 # Test issue8225 to ensure no doc link appears for xml.etree
391 result, doc_loc = get_pydoc_text(xml.etree)
392 self.assertEqual(doc_loc, "", "MODULE DOCS incorrectly includes a link")
393
R David Murrayc43125a2012-04-23 13:23:57 -0400394 def test_non_str_name(self):
395 # issue14638
396 # Treat illegal (non-str) name like no name
397 class A:
398 __name__ = 42
399 class B:
400 pass
401 adoc = pydoc.render_doc(A())
402 bdoc = pydoc.render_doc(B())
403 self.assertEqual(adoc.replace("A", "B"), bdoc)
404
Georg Brandlb533e262008-05-25 18:19:30 +0000405 def test_not_here(self):
406 missing_module = "test.i_am_not_here"
407 result = str(run_pydoc(missing_module), 'ascii')
408 expected = missing_pattern % missing_module
409 self.assertEqual(expected, result,
410 "documentation for missing module found")
411
R. David Murray1f1b9d32009-05-27 20:56:59 +0000412 def test_input_strip(self):
413 missing_module = " test.i_am_not_here "
414 result = str(run_pydoc(missing_module), 'ascii')
415 expected = missing_pattern % missing_module.strip()
416 self.assertEqual(expected, result)
417
Ezio Melotti412c95a2010-02-16 23:31:04 +0000418 def test_stripid(self):
419 # test with strings, other implementations might have different repr()
420 stripid = pydoc.stripid
421 # strip the id
422 self.assertEqual(stripid('<function stripid at 0x88dcee4>'),
423 '<function stripid>')
424 self.assertEqual(stripid('<function stripid at 0x01F65390>'),
425 '<function stripid>')
426 # nothing to strip, return the same text
427 self.assertEqual(stripid('42'), '42')
428 self.assertEqual(stripid("<type 'exceptions.Exception'>"),
429 "<type 'exceptions.Exception'>")
430
Georg Brandld80d5f42010-12-03 07:47:22 +0000431 @unittest.skipIf(sys.flags.optimize >= 2,
432 'Docstrings are omitted with -O2 and above')
Brett Cannon7a540732011-02-22 03:04:06 +0000433 @unittest.skipIf(hasattr(sys, 'gettrace') and sys.gettrace(),
434 'trace function introduces __locals__ unexpectedly')
Georg Brandld80d5f42010-12-03 07:47:22 +0000435 def test_help_output_redirect(self):
436 # issue 940286, if output is set in Helper, then all output from
437 # Helper.help should be redirected
438 old_pattern = expected_text_pattern
439 getpager_old = pydoc.getpager
440 getpager_new = lambda: (lambda x: x)
441 self.maxDiff = None
442
443 buf = StringIO()
444 helper = pydoc.Helper(output=buf)
445 unused, doc_loc = get_pydoc_text(pydoc_mod)
446 module = "test.pydoc_mod"
447 help_header = """
448 Help on module test.pydoc_mod in test:
449
450 """.lstrip()
451 help_header = textwrap.dedent(help_header)
452 expected_help_pattern = help_header + expected_text_pattern
453
454 pydoc.getpager = getpager_new
455 try:
456 with captured_output('stdout') as output, \
457 captured_output('stderr') as err:
458 helper.help(module)
459 result = buf.getvalue().strip()
Serhiy Storchaka9d0add02013-01-27 19:47:45 +0200460 expected_text = expected_help_pattern % (
461 (doc_loc,) +
462 expected_text_data_docstrings +
463 (inspect.getabsfile(pydoc_mod),))
Georg Brandld80d5f42010-12-03 07:47:22 +0000464 self.assertEqual('', output.getvalue())
465 self.assertEqual('', err.getvalue())
466 self.assertEqual(expected_text, result)
467 finally:
468 pydoc.getpager = getpager_old
469
Raymond Hettinger1103d052011-03-25 14:15:24 -0700470 def test_namedtuple_public_underscore(self):
471 NT = namedtuple('NT', ['abc', 'def'], rename=True)
472 with captured_stdout() as help_io:
Terry Jan Reedy5c811642013-11-04 21:43:26 -0500473 pydoc.help(NT)
Raymond Hettinger1103d052011-03-25 14:15:24 -0700474 helptext = help_io.getvalue()
475 self.assertIn('_1', helptext)
476 self.assertIn('_replace', helptext)
477 self.assertIn('_asdict', helptext)
478
Victor Stinnere6c910e2011-06-30 15:55:43 +0200479 def test_synopsis(self):
480 self.addCleanup(unlink, TESTFN)
481 for encoding in ('ISO-8859-1', 'UTF-8'):
482 with open(TESTFN, 'w', encoding=encoding) as script:
483 if encoding != 'UTF-8':
484 print('#coding: {}'.format(encoding), file=script)
485 print('"""line 1: h\xe9', file=script)
486 print('line 2: hi"""', file=script)
487 synopsis = pydoc.synopsis(TESTFN, {})
488 self.assertEqual(synopsis, 'line 1: h\xe9')
489
Eric Snowaed5b222014-01-04 20:38:11 -0700490 def test_synopsis_sourceless(self):
491 expected = os.__doc__.splitlines()[0]
492 filename = os.__cached__
493 synopsis = pydoc.synopsis(filename)
494
495 self.assertEqual(synopsis, expected)
496
R David Murray455f2962013-03-19 00:00:33 -0400497 def test_splitdoc_with_description(self):
498 example_string = "I Am A Doc\n\n\nHere is my description"
499 self.assertEqual(pydoc.splitdoc(example_string),
500 ('I Am A Doc', '\nHere is my description'))
501
502 def test_is_object_or_method(self):
503 doc = pydoc.Doc()
504 # Bound Method
505 self.assertTrue(pydoc._is_some_method(doc.fail))
506 # Method Descriptor
507 self.assertTrue(pydoc._is_some_method(int.__add__))
508 # String
509 self.assertFalse(pydoc._is_some_method("I am not a method"))
510
511 def test_is_package_when_not_package(self):
512 with test.support.temp_cwd() as test_dir:
513 self.assertFalse(pydoc.ispackage(test_dir))
514
515 def test_is_package_when_is_package(self):
516 with test.support.temp_cwd() as test_dir:
517 init_path = os.path.join(test_dir, '__init__.py')
518 open(init_path, 'w').close()
519 self.assertTrue(pydoc.ispackage(test_dir))
520 os.remove(init_path)
521
R David Murrayac0cea52013-03-19 02:47:44 -0400522 def test_allmethods(self):
523 # issue 17476: allmethods was no longer returning unbound methods.
524 # This test is a bit fragile in the face of changes to object and type,
525 # but I can't think of a better way to do it without duplicating the
526 # logic of the function under test.
527
528 class TestClass(object):
529 def method_returning_true(self):
530 return True
531
532 # What we expect to get back: everything on object...
533 expected = dict(vars(object))
534 # ...plus our unbound method...
535 expected['method_returning_true'] = TestClass.method_returning_true
536 # ...but not the non-methods on object.
537 del expected['__doc__']
538 del expected['__class__']
539 # inspect resolves descriptors on type into methods, but vars doesn't,
540 # so we need to update __subclasshook__.
541 expected['__subclasshook__'] = TestClass.__subclasshook__
542
543 methods = pydoc.allmethods(TestClass)
544 self.assertDictEqual(methods, expected)
545
Georg Brandlb533e262008-05-25 18:19:30 +0000546
Antoine Pitrou916fc7b2013-05-19 15:44:54 +0200547class PydocImportTest(PydocBaseTest):
Ned Deily92a81a12011-10-06 14:19:03 -0700548
549 def setUp(self):
550 self.test_dir = os.mkdir(TESTFN)
551 self.addCleanup(rmtree, TESTFN)
552
553 def test_badimport(self):
554 # This tests the fix for issue 5230, where if pydoc found the module
555 # but the module had an internal import error pydoc would report no doc
556 # found.
557 modname = 'testmod_xyzzy'
558 testpairs = (
559 ('i_am_not_here', 'i_am_not_here'),
Brett Cannonfd074152012-04-14 14:10:13 -0400560 ('test.i_am_not_here_either', 'test.i_am_not_here_either'),
561 ('test.i_am_not_here.neither_am_i', 'test.i_am_not_here'),
562 ('i_am_not_here.{}'.format(modname), 'i_am_not_here'),
563 ('test.{}'.format(modname), 'test.{}'.format(modname)),
Ned Deily92a81a12011-10-06 14:19:03 -0700564 )
565
566 sourcefn = os.path.join(TESTFN, modname) + os.extsep + "py"
567 for importstring, expectedinmsg in testpairs:
568 with open(sourcefn, 'w') as f:
569 f.write("import {}\n".format(importstring))
570 result = run_pydoc(modname, PYTHONPATH=TESTFN).decode("ascii")
571 expected = badimport_pattern % (modname, expectedinmsg)
572 self.assertEqual(expected, result)
573
574 def test_apropos_with_bad_package(self):
575 # Issue 7425 - pydoc -k failed when bad package on path
576 pkgdir = os.path.join(TESTFN, "syntaxerr")
577 os.mkdir(pkgdir)
578 badsyntax = os.path.join(pkgdir, "__init__") + os.extsep + "py"
579 with open(badsyntax, 'w') as f:
580 f.write("invalid python syntax = $1\n")
Antoine Pitrou916fc7b2013-05-19 15:44:54 +0200581 with self.restrict_walk_packages(path=[TESTFN]):
582 with captured_stdout() as out:
583 with captured_stderr() as err:
584 pydoc.apropos('xyzzy')
585 # No result, no error
586 self.assertEqual(out.getvalue(), '')
587 self.assertEqual(err.getvalue(), '')
588 # The package name is still matched
589 with captured_stdout() as out:
590 with captured_stderr() as err:
591 pydoc.apropos('syntaxerr')
592 self.assertEqual(out.getvalue().strip(), 'syntaxerr')
593 self.assertEqual(err.getvalue(), '')
Ned Deily92a81a12011-10-06 14:19:03 -0700594
595 def test_apropos_with_unreadable_dir(self):
596 # Issue 7367 - pydoc -k failed when unreadable dir on path
597 self.unreadable_dir = os.path.join(TESTFN, "unreadable")
598 os.mkdir(self.unreadable_dir, 0)
599 self.addCleanup(os.rmdir, self.unreadable_dir)
600 # Note, on Windows the directory appears to be still
601 # readable so this is not really testing the issue there
Antoine Pitrou916fc7b2013-05-19 15:44:54 +0200602 with self.restrict_walk_packages(path=[TESTFN]):
603 with captured_stdout() as out:
604 with captured_stderr() as err:
605 pydoc.apropos('SOMEKEY')
606 # No result, no error
607 self.assertEqual(out.getvalue(), '')
608 self.assertEqual(err.getvalue(), '')
Ned Deily92a81a12011-10-06 14:19:03 -0700609
Eric Snowaed5b222014-01-04 20:38:11 -0700610 def test_modules(self):
611 # See Helper.listmodules().
612 num_header_lines = 2
613 num_module_lines_min = 5 # Playing it safe.
614 num_footer_lines = 3
615 expected = num_header_lines + num_module_lines_min + num_footer_lines
616
617 output = StringIO()
618 helper = pydoc.Helper(output=output)
619 helper('modules')
620 result = output.getvalue().strip()
621 num_lines = len(result.splitlines())
622
623 self.assertGreaterEqual(num_lines, expected)
624
625 def test_modules_search(self):
626 # See Helper.listmodules().
627 expected = 'pydoc - '
628
629 output = StringIO()
630 helper = pydoc.Helper(output=output)
631 with captured_stdout() as help_io:
632 helper('modules pydoc')
633 result = help_io.getvalue()
634
635 self.assertIn(expected, result)
636
Eric Snow5ea97502014-01-04 23:04:27 -0700637 @unittest.skip('some buildbots are not cooperating (#20123)')
Eric Snowaed5b222014-01-04 20:38:11 -0700638 def test_modules_search_builtin(self):
Eric Snow5ea97502014-01-04 23:04:27 -0700639 expected = 'gc - '
Eric Snowaed5b222014-01-04 20:38:11 -0700640
641 output = StringIO()
642 helper = pydoc.Helper(output=output)
643 with captured_stdout() as help_io:
Eric Snow5ea97502014-01-04 23:04:27 -0700644 helper('modules garbage')
Eric Snowaed5b222014-01-04 20:38:11 -0700645 result = help_io.getvalue()
646
647 self.assertTrue(result.startswith(expected))
648
649 def test_importfile(self):
650 loaded_pydoc = pydoc.importfile(pydoc.__file__)
651
Eric Snow3a62d142014-01-06 20:42:59 -0700652 self.assertIsNot(loaded_pydoc, pydoc)
Eric Snowaed5b222014-01-04 20:38:11 -0700653 self.assertEqual(loaded_pydoc.__name__, 'pydoc')
654 self.assertEqual(loaded_pydoc.__file__, pydoc.__file__)
Eric Snow3a62d142014-01-06 20:42:59 -0700655 self.assertEqual(loaded_pydoc.__spec__, pydoc.__spec__)
Eric Snowaed5b222014-01-04 20:38:11 -0700656
Ned Deily92a81a12011-10-06 14:19:03 -0700657
Georg Brandlb533e262008-05-25 18:19:30 +0000658class TestDescriptions(unittest.TestCase):
659
660 def test_module(self):
661 # Check that pydocfodder module can be described
662 from test import pydocfodder
663 doc = pydoc.render_doc(pydocfodder)
Benjamin Peterson577473f2010-01-19 00:09:57 +0000664 self.assertIn("pydocfodder", doc)
Georg Brandlb533e262008-05-25 18:19:30 +0000665
Georg Brandlb533e262008-05-25 18:19:30 +0000666 def test_class(self):
Nick Coghlan7bb30b72010-12-03 09:29:11 +0000667 class C: "New-style class"
Georg Brandlb533e262008-05-25 18:19:30 +0000668 c = C()
669
670 self.assertEqual(pydoc.describe(C), 'class C')
671 self.assertEqual(pydoc.describe(c), 'C')
672 expected = 'C in module %s object' % __name__
Benjamin Peterson577473f2010-01-19 00:09:57 +0000673 self.assertIn(expected, pydoc.render_doc(c))
Georg Brandlb533e262008-05-25 18:19:30 +0000674
Éric Araujoe64e51b2011-07-29 17:03:55 +0200675 def test_builtin(self):
676 for name in ('str', 'str.translate', 'builtins.str',
677 'builtins.str.translate'):
678 # test low-level function
679 self.assertIsNotNone(pydoc.locate(name))
680 # test high-level function
681 try:
682 pydoc.render_doc(name)
683 except ImportError:
684 self.fail('finding the doc of {!r} failed'.format(o))
685
686 for name in ('notbuiltins', 'strrr', 'strr.translate',
687 'str.trrrranslate', 'builtins.strrr',
688 'builtins.str.trrranslate'):
689 self.assertIsNone(pydoc.locate(name))
690 self.assertRaises(ImportError, pydoc.render_doc, name)
691
Stefan Krah5de32782014-01-18 23:18:39 +0100692 @requires_docstrings
693 def test_builtin_signatures(self):
Larry Hastings1abd7082014-01-16 14:15:03 -0800694 # test producing signatures from builtins
695 stat_sig = pydoc.render_doc(os.stat)
696 self.assertEqual(pydoc.plain(stat_sig).splitlines()[2],
697 'stat(path, *, dir_fd=None, follow_symlinks=True)')
698
Georg Brandlb533e262008-05-25 18:19:30 +0000699
Victor Stinner62a68f22011-05-20 02:29:13 +0200700@unittest.skipUnless(threading, 'Threading required for this test.')
Georg Brandld2f38572011-01-30 08:37:19 +0000701class PydocServerTest(unittest.TestCase):
Nick Coghlan7bb30b72010-12-03 09:29:11 +0000702 """Tests for pydoc._start_server"""
703
704 def test_server(self):
705
706 # Minimal test that starts the server, then stops it.
707 def my_url_handler(url, content_type):
708 text = 'the URL sent was: (%s, %s)' % (url, content_type)
709 return text
710
711 serverthread = pydoc._start_server(my_url_handler, port=0)
712 starttime = time.time()
713 timeout = 1 #seconds
714
715 while serverthread.serving:
716 time.sleep(.01)
717 if serverthread.serving and time.time() - starttime > timeout:
718 serverthread.stop()
719 break
720
721 self.assertEqual(serverthread.error, None)
722
723
Antoine Pitrou916fc7b2013-05-19 15:44:54 +0200724class PydocUrlHandlerTest(PydocBaseTest):
Nick Coghlan7bb30b72010-12-03 09:29:11 +0000725 """Tests for pydoc._url_handler"""
726
727 def test_content_type_err(self):
Nick Coghlan7bb30b72010-12-03 09:29:11 +0000728 f = pydoc._url_handler
Georg Brandld2f38572011-01-30 08:37:19 +0000729 self.assertRaises(TypeError, f, 'A', '')
730 self.assertRaises(TypeError, f, 'B', 'foobar')
Nick Coghlan7bb30b72010-12-03 09:29:11 +0000731
732 def test_url_requests(self):
733 # Test for the correct title in the html pages returned.
734 # This tests the different parts of the URL handler without
735 # getting too picky about the exact html.
736 requests = [
Georg Brandld2f38572011-01-30 08:37:19 +0000737 ("", "Pydoc: Index of Modules"),
738 ("get?key=", "Pydoc: Index of Modules"),
739 ("index", "Pydoc: Index of Modules"),
740 ("topics", "Pydoc: Topics"),
741 ("keywords", "Pydoc: Keywords"),
742 ("pydoc", "Pydoc: module pydoc"),
743 ("get?key=pydoc", "Pydoc: module pydoc"),
744 ("search?key=pydoc", "Pydoc: Search Results"),
745 ("topic?key=def", "Pydoc: KEYWORD def"),
746 ("topic?key=STRINGS", "Pydoc: TOPIC STRINGS"),
747 ("foobar", "Pydoc: Error - foobar"),
748 ("getfile?key=foobar", "Pydoc: Error - getfile?key=foobar"),
Nick Coghlan7bb30b72010-12-03 09:29:11 +0000749 ]
750
Antoine Pitrou916fc7b2013-05-19 15:44:54 +0200751 with self.restrict_walk_packages():
752 for url, title in requests:
753 text = pydoc._url_handler(url, "text/html")
754 result = get_html_title(text)
755 self.assertEqual(result, title, text)
756
757 path = string.__file__
758 title = "Pydoc: getfile " + path
759 url = "getfile?key=" + path
Nick Coghlan7bb30b72010-12-03 09:29:11 +0000760 text = pydoc._url_handler(url, "text/html")
761 result = get_html_title(text)
762 self.assertEqual(result, title)
763
Nick Coghlan7bb30b72010-12-03 09:29:11 +0000764
Ezio Melottib185a042011-04-28 07:42:55 +0300765class TestHelper(unittest.TestCase):
766 def test_keywords(self):
767 self.assertEqual(sorted(pydoc.Helper.keywords),
768 sorted(keyword.kwlist))
769
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700770class PydocWithMetaClasses(unittest.TestCase):
Ethan Furman3f2f1922013-10-22 07:30:24 -0700771 @unittest.skipIf(sys.flags.optimize >= 2,
772 "Docstrings are omitted with -O2 and above")
773 @unittest.skipIf(hasattr(sys, 'gettrace') and sys.gettrace(),
774 'trace function introduces __locals__ unexpectedly')
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700775 def test_DynamicClassAttribute(self):
776 class Meta(type):
777 def __getattr__(self, name):
778 if name == 'ham':
779 return 'spam'
780 return super().__getattr__(name)
781 class DA(metaclass=Meta):
782 @types.DynamicClassAttribute
783 def ham(self):
784 return 'eggs'
Ethan Furman3f2f1922013-10-22 07:30:24 -0700785 expected_text_data_docstrings = tuple('\n | ' + s if s else ''
786 for s in expected_data_docstrings)
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700787 output = StringIO()
788 helper = pydoc.Helper(output=output)
789 helper(DA)
Ethan Furman3f2f1922013-10-22 07:30:24 -0700790 expected_text = expected_dynamicattribute_pattern % (
791 (__name__,) + expected_text_data_docstrings[:2])
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700792 result = output.getvalue().strip()
793 if result != expected_text:
794 print_diffs(expected_text, result)
795 self.fail("outputs are not equal, see diff above")
796
Ethan Furman3f2f1922013-10-22 07:30:24 -0700797 @unittest.skipIf(sys.flags.optimize >= 2,
798 "Docstrings are omitted with -O2 and above")
799 @unittest.skipIf(hasattr(sys, 'gettrace') and sys.gettrace(),
800 'trace function introduces __locals__ unexpectedly')
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700801 def test_virtualClassAttributeWithOneMeta(self):
802 class Meta(type):
803 def __dir__(cls):
804 return ['__class__', '__module__', '__name__', 'LIFE']
805 def __getattr__(self, name):
806 if name =='LIFE':
807 return 42
808 return super().__getattr(name)
809 class Class(metaclass=Meta):
810 pass
811 output = StringIO()
812 helper = pydoc.Helper(output=output)
813 helper(Class)
814 expected_text = expected_virtualattribute_pattern1 % __name__
815 result = output.getvalue().strip()
816 if result != expected_text:
817 print_diffs(expected_text, result)
818 self.fail("outputs are not equal, see diff above")
819
Ethan Furman3f2f1922013-10-22 07:30:24 -0700820 @unittest.skipIf(sys.flags.optimize >= 2,
821 "Docstrings are omitted with -O2 and above")
822 @unittest.skipIf(hasattr(sys, 'gettrace') and sys.gettrace(),
823 'trace function introduces __locals__ unexpectedly')
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700824 def test_virtualClassAttributeWithTwoMeta(self):
825 class Meta1(type):
826 def __dir__(cls):
827 return ['__class__', '__module__', '__name__', 'one']
828 def __getattr__(self, name):
829 if name =='one':
830 return 1
831 return super().__getattr__(name)
832 class Meta2(type):
833 def __dir__(cls):
834 return ['__class__', '__module__', '__name__', 'two']
835 def __getattr__(self, name):
836 if name =='two':
837 return 2
838 return super().__getattr__(name)
839 class Meta3(Meta1, Meta2):
840 def __dir__(cls):
841 return list(sorted(set(
842 ['__class__', '__module__', '__name__', 'three'] +
843 Meta1.__dir__(cls) + Meta2.__dir__(cls))))
844 def __getattr__(self, name):
845 if name =='three':
846 return 3
847 return super().__getattr__(name)
848 class Class1(metaclass=Meta1):
849 pass
850 class Class2(Class1, metaclass=Meta3):
851 pass
852 fail1 = fail2 = False
853 output = StringIO()
854 helper = pydoc.Helper(output=output)
855 helper(Class1)
856 expected_text1 = expected_virtualattribute_pattern2 % __name__
857 result1 = output.getvalue().strip()
858 if result1 != expected_text1:
859 print_diffs(expected_text1, result1)
860 fail1 = True
861 output = StringIO()
862 helper = pydoc.Helper(output=output)
863 helper(Class2)
864 expected_text2 = expected_virtualattribute_pattern3 % __name__
865 result2 = output.getvalue().strip()
866 if result2 != expected_text2:
867 print_diffs(expected_text2, result2)
868 fail2 = True
869 if fail1 or fail2:
870 self.fail("outputs are not equal, see diff above")
871
Ethan Furman3f2f1922013-10-22 07:30:24 -0700872 @unittest.skipIf(sys.flags.optimize >= 2,
873 "Docstrings are omitted with -O2 and above")
874 @unittest.skipIf(hasattr(sys, 'gettrace') and sys.gettrace(),
875 'trace function introduces __locals__ unexpectedly')
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700876 def test_buggy_dir(self):
877 class M(type):
878 def __dir__(cls):
879 return ['__class__', '__name__', 'missing', 'here']
880 class C(metaclass=M):
881 here = 'present!'
882 output = StringIO()
883 helper = pydoc.Helper(output=output)
884 helper(C)
885 expected_text = expected_missingattribute_pattern % __name__
886 result = output.getvalue().strip()
887 if result != expected_text:
888 print_diffs(expected_text, result)
889 self.fail("outputs are not equal, see diff above")
890
Eric Snowaed5b222014-01-04 20:38:11 -0700891
Antoine Pitroua6e81a22011-07-15 22:32:25 +0200892@reap_threads
Georg Brandlb533e262008-05-25 18:19:30 +0000893def test_main():
Antoine Pitroua6e81a22011-07-15 22:32:25 +0200894 try:
895 test.support.run_unittest(PydocDocTest,
Ned Deily92a81a12011-10-06 14:19:03 -0700896 PydocImportTest,
Antoine Pitroua6e81a22011-07-15 22:32:25 +0200897 TestDescriptions,
898 PydocServerTest,
899 PydocUrlHandlerTest,
900 TestHelper,
Ethan Furmanb0c84cd2013-10-20 22:37:39 -0700901 PydocWithMetaClasses,
Antoine Pitroua6e81a22011-07-15 22:32:25 +0200902 )
903 finally:
904 reap_children()
Georg Brandlb533e262008-05-25 18:19:30 +0000905
906if __name__ == "__main__":
907 test_main()