From: Antoine Pitrou <solipsis@pitrou.net>
Date: Fri, 13 Sep 2013 22:39:15 +0000 (+0200)
Subject: Issue #19013: add a __main__ to unittest.test.testmock to ease CLI invocation
X-Git-Tag: v3.4.0a3~71
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=4b3c58c8a0b0c041854e80e9a01aeb5d77872a05;p=thirdparty%2FPython%2Fcpython.git

Issue #19013: add a __main__ to unittest.test.testmock to ease CLI invocation
---

diff --git a/Lib/unittest/test/testmock/__main__.py b/Lib/unittest/test/testmock/__main__.py
new file mode 100644
index 000000000000..24e1933a8ee1
--- /dev/null
+++ b/Lib/unittest/test/testmock/__main__.py
@@ -0,0 +1,18 @@
+import os
+import unittest
+
+
+def load_tests(loader, standard_tests, pattern):
+    # top level directory cached on loader instance
+    this_dir = os.path.dirname(__file__)
+    pattern = pattern or "test*.py"
+    # We are inside unittest.test, so the top-level is two notches up
+    top_level_dir = os.path.dirname(os.path.dirname(os.path.dirname(this_dir)))
+    package_tests = loader.discover(start_dir=this_dir, pattern=pattern,
+                                    top_level_dir=top_level_dir)
+    standard_tests.addTests(package_tests)
+    return standard_tests
+
+
+if __name__ == '__main__':
+    unittest.main()