windows - python unittest的返回状态

Question

我正在尝试从另一个 python 文件调用单元测试，并评估退出代码。我能够使用 unittest.TestLoader().loadTestsFromModule 和 unittest.TextTestRunner.run 从另一个 python 文件调用单元测试，但这会将整个结果返回到 cmd .我想简单地设置一个等于状态代码的变量，以便我可以对其进行评估。我能够找到一个方法 unittest.TestResult.wasSuccessful，但我在实现它时遇到了麻烦。当我将它添加到用例时，我得到以下 AttributeError: AttributeError: 'ConnectionTest' object has no attribute 'failures'

我在下面包含了一些代码示例和所需结果的模型，以说明我正在努力实现的目标。先感谢您。

""" Tests/ConnectionTest.py """

import unittest
from Connection import Connection


class ConnectionTest(unittest.TestCase):

    def test_connection(self):
        #my tests

    def test_pass(self):
        return unittest.TestResult.wasSuccessful(self)


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


""" StatusTest.py """

import unittest
import Tests.ConnectionTest as test
#import Tests.Test2 as test2
#import Tests.Test3 as test3
#import other unit tests ...

suite = unittest.TestLoader().loadTestsFromModule(test)
unittest.TextTestRunner(verbosity=2).run(suite)


""" Return True if unit test passed
"""
def test_passed(test):
    if test.test_pass() == 0:
        return True
    else:
        return False

""" Run unittest for each module before using it in code
"""
def main():
    tests = "test test2 test3".split()
    for test in tests:
        if test_passed(test):
            # do something
        else:
            # log failure
            pass

更新

为了更简单的问题，我需要将下面突出显示的变量设置为突出显示的值。

Answer 1

您提到您尝试实现 result.wasSuccessful，但希望执行以下操作:

result = unittest.TextTestRunner(verbosity=2).run(suite)
test_exit_code = int(not result.wasSuccessful())

当测试套件成功运行时，test_exit_code 的值将为 0，否则为 1。

如果您想禁用 TextTestRunner 的输出，您可以指定您自己的流，例如:

from io import StringIO

result = unittest.TextTestRunner(stream=StringIO(), verbosity=2).run(suite)