1 files changed, 78 insertions, 0 deletions
diff --git a/tests/test_doctest.py b/tests/test_doctest.py
new file mode 100644
index 0000000..6ef1e09
--- /dev/null
+++ b/tests/test_doctest.py
@@ -0,0 +1,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()
+