python - 运行 Python unittest，以便在成功时不打印任何内容，仅在失败时打印 AssertionError()

Question

我有一个标准单元测试格式的测试模块

class my_test(unittest.TestCase):

    def test_1(self):
        [tests]

    def test_2(self):
        [tests]
  etc....

我的公司有一个专有的测试工具，可以将我的模块作为命令行脚本执行，它会捕获我的模块引发的任何错误，但如果成功则要求我的模块静音。

因此，我正在尝试找到一种方法来裸露地运行我的测试模块，这样如果我所有的测试都通过了，那么屏幕上不会打印任何内容，如果测试因 AssertionError 而失败，则该错误将通过标准管道传输Python 错误堆栈(就像普通 Python 脚本中的任何其他错误一样。)

docs提倡使用 unittest.main() 函数来运行给定模块中的所有测试，例如

if __name__ == "__main__":
    unittest.main()

问题是这将测试结果包装在 unittest 的 harness 中，因此即使所有测试都成功，它仍然会在屏幕上打印一些绒毛，如果有错误，它不会简单地作为通常的 python 错误转储, 还穿着安全带。

我尝试使用

将输出重定向到备用流

with open('.LOG','a') as logf:
    suite = unittest.TestLoader().loadTestsFromTestCase(my_test)
    unittest.TextTestRunner(stream = logf).run(suite)

这里的问题是一切都通过管道传输到日志文件(包括所有错误通知)。因此，当我的公司使用 harness 运行该模块时，它成功完成，因为据它所知，没有出现错误(因为它们都通过管道传输到日志文件)。

关于如何构建一个抑制所有错误并通过普通 Python 错误堆栈传送错误的测试运行器，有什么建议吗？一如既往，如果您认为有更好的方法来解决这个问题，请告诉我。

编辑:

这是我最终用来解决这个问题的方法。首先，我在我的测试类中添加了一个“get_test_names()”方法:

class my_test(unittest.TestCase):
  etc....
    @staticmethod
    def get_test_names():
        """Return the names of all the test methods for this class."""
        test_names = [ member[0] for memeber in inspect.getmembers(my_test)
                       if 'test_' in member[0] ]

然后我将对 unittest.main() 的调用替换为以下内容:

# Unittest catches all errors raised by the test cases, and returns them as 
# formatted strings inside a TestResult object. In order for the test 
# harness to catch these errors they need to be re-raised, and so I am defining 
# this CompareError class to do that. 
# For each code error, a CompareError will be raised, with the original error 
# stack as the argument. For test failures (i.e. assertion errors) an 
# AssertionError is raised.
class CompareError(Exception):
    def __init__(self,err):
        self.err = err
    def __str__(self):
        return repr(self.err)

# Collect all tests into a TestSuite()
all_tests = ut.TestSuite()
for test in my_test.get_test_names():
    all_tests.addTest(my_test(test))
# Define a TestResult object and run tests
results = ut.TestResult()
all_tests.run(results)
# Re-raise any script errors
for error in results.errors:
    raise CompareError(error[1])
# Re-raise any test failures 
for failure in results.failures:
    raise AssertionError(failure[1])

Answer 1

我想到了这个。如果您能够更改命令行，您可能会删除内部 io 重定向。

import sys, inspect, traceback

# redirect stdout,
# can be replaced by testharness.py > /dev/null at console
class devnull():
    def write(self, data):
        pass

f = devnull()
orig_stdout = sys.stdout
sys.stdout = f

class TestCase():
    def test_1(self):
        print 'test_1'

    def test_2(self):
        raise AssertionError, 'test_2'

    def test_3(self):
        print 'test_3'


if __name__ == "__main__":
    testcase = TestCase()
    testnames =  [ t[0] for t in inspect.getmembers(TestCase)
                        if t[0].startswith('test_') ]

    for testname in testnames:
        try:
            getattr(testcase, testname)()
        except AssertionError, e:
            print >> sys.stderr, traceback.format_exc()

# restore
sys.stdout = orig_stdout