有没有一种自定义的方式来收集pytest结果?

时间:2020-04-30 14:31:13

标签: python pytest

我想查看刚刚运行的测试结果,以便可以传递信息,例如进入自定义文件。稍后将使用它来自动解析pytest结果。

1 个答案:

答案 0 :(得分:0)

是的,有一种方法可以创建/使用conftest.py。在该示例中,将创建包含结果的字典,并在所有测试运行后打印结果。

from collections import OrderedDict
test_results = OrderedDict()

def get_current_test():
    """Just a helper function to extract the current test"""
    full_name = os.environ.get('PYTEST_CURRENT_TEST').split(' ')[0]
    test_file = full_name.split("::")[0].split('/')[-1].split('.py')[0]
    test_name = full_name.split("::")[1]
    return full_name, test_file, test_name


@pytest.hookimpl(tryfirst=True, hookwrapper=True)
def pytest_runtest_makereport(item, call):
    """The actual wrapper that gets called before and after every test"""
    global test_results
    outcome = yield
    rep = outcome.get_result()

    # only check the result of the test
    if rep.when == "call":
        full_name, test_file, test_name = get_current_test()
        test_name_msg = f"{test_name}_msg"
        if rep.failed:
            test_results[test_name] = "Failure"
            # return the last error msg either by pytest.fail or from any exception raised
            test_results[test_name_msg] = f"{call.excinfo.typename} - {call.excinfo.value}"
        else:
            test_results[test_name] = "Success"
            test_results[test_name_msg] = ''

def pytest_unconfigure(config):
    """Called when pytest is about to end. Can be used to print the result dict or 
    to pipe the data into a file"""
    print(test_results)