Question

我正在运行在执行期间创建大量大文件的测试。为此，如果测试已通过，我想删除tmpdir目录。但如果测试失败，则tmpdir的内容应保持不变。

我知道如何确定测试结果：

from _pytest.runner import runtestprotocol

def pytest_runtest_protocol(item, nextitem):
    reports = runtestprotocol(item, nextitem=nextitem)
    for report in reports:
        if report.when == 'call':
            # value will be set to passed or filed
            test_outcome = report.outcome
            # But what next?

return True

但我不知道如何找到创建的tmpdir目录的路径。

Answer 1

您应create a tmpdir fixture创建tempdir，将其传递到您的代码中，然后将其删除。

此外，必须将夹具设置为始终删除tempdir，即使失败也是如此。否则，您可能会留下一个不洁的状态，这可能会导致其他测试失败（用户没有注意到）。

相反，我推荐

使用--pdb在出现错误时放入Python调试器。夹具尚未清理，您可以检查文件。
Creating a custom option，允许您禁用tmpdir的清理。
创建一个自定义tmpdir fixture，将所有tmpfiles复制到用户可配置的位置（同样，使用自定义选项），然后清理tmpdir。

在任何情况下，不洁的tmpdir状态都是用户有意识的决定，并会防止意外的副作用。

Answer 2

您可以从实际项目的 funcargs 轻松检索您的tmpdir。

在你的情况下：

from _pytest.runner import runtestprotocol

def pytest_runtest_protocol(item, nextitem):
    reports = runtestprotocol(item, nextitem=nextitem)
    for report in reports:
        if report.when == 'call':
            # value will be set to passed or filed
            test_outcome = report.outcome
            # depending on test_outcome value, remove tmpdir
            if test_outcome is "OK for you":
               if 'tmpdir' in item.funcargs:
                  tmpdir = item.funcargs['tmpdir'] #retrieve tmpdir
                  if tmpdir.check(): tmpdir.remove()

return True

对于故事，item.funcargs是一个字典，其中包含传递给我们当前正在检查的测试项的{arguments：value}。因此，第一步是检查 tmpdir 确实是实际测试的arg，然后检索它。并在删除它之前最终检查它的存在。

希望这会有所帮助。

编辑：好像你的pytest_runtest_protocol（..）还没有完全初始化该项目。确保它是..

只需覆盖pytest_runtest_teardown（item），它就会在每个测试项运行完成后成功（成功或失败）。尝试添加类似的方法：

def pytest_runtest_teardown(item):
   if item.rep_call.passed:
      if 'tmpdir' in item.funcargs:
         tmpdir = item.funcargs['tmpdir'] #retrieve tmpdir
         if tmpdir.check(): tmpdir.remove()

而且，不要忘记以下内容（在文档中给出）以便轻松访问您的报告。

@pytest.hookimpl(tryfirst=True, hookwrapper=True)
def pytest_runtest_makereport(item, call,):
    # execute all other hooks to obtain the report object
    outcome = yield
    rep = outcome.get_result()
    # set an report attribute for each phase of a call, which can
    # be "setup", "call", "teardown"
    setattr(item, "rep_" + rep.when, rep)

成功测试用例后删除py.test tmpdir目录

2 个答案: