#17066: test_robotparser now works with unittest test discovery. Patch by Zachary Ware.
diff --git a/Lib/test/test_robotparser.py b/Lib/test/test_robotparser.py
index 178761d..8c09e74 100644
--- a/Lib/test/test_robotparser.py
+++ b/Lib/test/test_robotparser.py
@@ -6,7 +6,10 @@
from test import support
class RobotTestCase(unittest.TestCase):
- def __init__(self, index, parser, url, good, agent):
+ def __init__(self, index=None, parser=None, url=None, good=None, agent=None):
+ # workaround to make unittest discovery work (see #17066)
+ if not isinstance(index, int):
+ return
unittest.TestCase.__init__(self)
if good:
self.str = "RobotTest(%d, good, %s)" % (index, url)
@@ -269,10 +272,11 @@
self.assertTrue(
parser.can_fetch("*", "http://www.python.org/robots.txt"))
-def test_main():
- support.run_unittest(NetworkTestCase)
- support.run_unittest(tests)
+def load_tests(loader, suite, pattern):
+ suite = unittest.makeSuite(NetworkTestCase)
+ suite.addTest(tests)
+ return suite
if __name__=='__main__':
- support.verbose = 1
- test_main()
+ support.use_resources = ['network']
+ unittest.main()