Python：在不调用 unittest.main() 的情况下运行 unittest.TestCase？

Question

我在 Python 的 unittest 中编写了一个小型测试套件：

class TestRepos(unittest.TestCase):

@classmethod
def setUpClass(cls):
    """Get repo lists from the svn server."""
    ...

def test_repo_list_not_empty(self):
    """Assert the the repo list is not empty"""
    self.assertTrue(len(TestRepoLists.all_repos)>0)

def test_include_list_not_empty(self):
    """Assert the the include list is not empty"""
    self.assertTrue(len(TestRepoLists.svn_dirs)>0)

...

if __name__ == '__main__':
    unittest.main(testRunner=xmlrunner.XMLTestRunner(output='tests', 
                                                 descriptions=True))

使用 the xmlrunner pacakge 将输出格式化为 Junit 测试。

我添加了一个用于切换 JUnit 输出的命令行参数：

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Validate repo lists.')
    parser.add_argument('--junit', action='store_true')
    args=parser.parse_args()
    print args
    if (args.junit):
        unittest.main(testRunner=xmlrunner.XMLTestRunner(output='tests', 
                                                     descriptions=True))
    else:
        unittest.main(TestRepoLists)

问题是在没有--junit 的情况下运行脚本可以工作，但使用--junit 调用它会与unittest 的参数发生冲突：

option --junit not recognized
Usage: test_lists_of_repos_to_branch.py [options] [test] [...]

Options:
  -h, --help       Show this message
  -v, --verbose    Verbose output
  ...

如何在不调用 unittest.main() 的情况下运行 unittest.TestCase？

Answer 1

您确实应该使用合适的测试运行程序（例如nose 或zope.testing）。在您的具体情况下，我会改用argparser.parse_known_args()：

if __name__ == '__main__':
    parser = argparse.ArgumentParser(add_help=False)
    parser.add_argument('--junit', action='store_true')
    options, args = parser.parse_known_args()

    testrunner = None
    if (options.junit):
        testrunner = xmlrunner.XMLTestRunner(output='tests', descriptions=True)
    unittest.main(testRunner=testrunner, argv=sys.argv[:1] + args)

请注意，我从您的参数解析器中删除了--help，因此--junit 选项被隐藏，但它不会再干扰unittest.main。我还将剩余的参数传递给unittest.main()。