diff options
Diffstat (limited to 'Lib/unittest/test')
| -rw-r--r-- | Lib/unittest/test/test_discovery.py | 13 | ||||
| -rw-r--r-- | Lib/unittest/test/test_program.py | 1 |
2 files changed, 14 insertions, 0 deletions
diff --git a/Lib/unittest/test/test_discovery.py b/Lib/unittest/test/test_discovery.py index 52a711a..e688f8e 100644 --- a/Lib/unittest/test/test_discovery.py +++ b/Lib/unittest/test/test_discovery.py @@ -231,6 +231,19 @@ class TestDiscovery(unittest.TestCase): program.parseArgs(['something']) self.assertTrue(self.called) + def test_command_line_handling_discover_by_default_with_options(self): + program = TestableTestProgram() + program.module = None + + args = ['something', '-v', '-b', '-v', '-c', '-f'] + self.called = False + def do_discovery(argv): + self.called = True + self.assertEqual(argv, args[1:]) + program._do_discovery = do_discovery + program.parseArgs(args) + self.assertTrue(self.called) + def test_command_line_handling_do_discovery_too_many_arguments(self): class Stop(Exception): diff --git a/Lib/unittest/test/test_program.py b/Lib/unittest/test/test_program.py index ec805cc..a50a245 100644 --- a/Lib/unittest/test/test_program.py +++ b/Lib/unittest/test/test_program.py @@ -99,6 +99,7 @@ class InitialisableProgram(unittest.TestProgram): defaultTest = None testRunner = None testLoader = unittest.defaultTestLoader + module = '__main__' progName = 'test' test = 'test' def __init__(self, *args): |
