1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
# -*- coding: utf-8 -*-
# pylint: disable=missing-docstring,unused-argument,too-many-arguments
import os
import unittest
import doctest
import getpass
import mercurial_extension_utils
import sys
USING_PY3 = sys.version_info >= (3, 0, 0)
# IMPORTANT NOTE:
#
# As I wanted doctests to be readable, most of them assume
# specific paths and names (for example some tests assume /home/lordvader
# as home directory). This is on purpose,
#
# >>> normalize_path("~/src")
# '/home/lordvader/src'
#
# is readable and fulfills documentation role well, whatever I could write
# instead to handle various accounts, would be unreadable mess.
#
# To make running tests possible, below we adapt docstrings
# before executing them.
class FixingUpDocTestParser(doctest.DocTestParser): # pylint: disable=no-init
PATTERN_HOME = '/home/lordvader'
PATTERN_NAME = 'lordvader'
TRUE_HOME = os.path.expanduser("~")
TRUE_NAME = getpass.getuser()
REL_TO_HOME = os.path.relpath(TRUE_HOME)
def get_doctest(self, string, globs, name, filename, lineno):
# Replace /home/lordvader with whatever true home is
# (and similar)
string = string \
.replace(self.PATTERN_HOME, self.TRUE_HOME) \
.replace(self.PATTERN_NAME, self.TRUE_NAME)
# Special fixup for ../../.. pointing at home
string = string.replace('"../../..', '"' + self.REL_TO_HOME)
return doctest.DocTestParser.get_doctest(
self, string, globs, name, filename, lineno)
def load_tests(loader, tests, pattern):
if os.name != 'nt':
if USING_PY3:
finder = doctest.DocTestFinder(parser=FixingUpDocTestParser())
suite = doctest.DocTestSuite(
mercurial_extension_utils,
test_finder=finder)
else:
suite = doctest.DocFileSuite(
"py2_doctests_mercurial_extension_utils.py",
module_relative=True,
globs=mercurial_extension_utils.__dict__,
parser=FixingUpDocTestParser())
else:
if USING_PY3:
raise Exception("TODO: py3 tests for Windows")
else:
suite = doctest.DocFileSuite(
"py2win_doctests_mercurial_extension_utils.py",
module_relative=True,
globs=mercurial_extension_utils.__dict__,
parser=FixingUpDocTestParser())
tests.addTests(suite)
return tests
if __name__ == "__main__":
unittest.main()
|
