Вернуть статус Python Unittest - PullRequest
Новая
       12

Вернуть статус Python Unittest

0 голосов
/ 03 мая 2018

Я пытаюсь вызвать unittest из другого файла python и оценить код завершения. Я смог использовать unittest.TestLoader().loadTestsFromModule и unittest.TextTestRunner.run для вызова unittest из другого файла 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

Обновление

Чтобы задать вопрос проще, мне нужно установить выделенную переменную ниже на выделенное значение.

goal

1 Ответ

0 голосов
/ 08 мая 2018

Вы упомянули, что пытались реализовать 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)
...