需求
在使用pytest-assume的时候，它会等待测试函数执行结束才会抛出错误，这样我们可以执行更多，在最终的时候统一查看错误。
但是在进行ui自动化测试的时候，我需要在错误出现的时候就进行截图。
现状
之前使用pytest的assue进行测试脚本编写的时候，使用了pytest_runtest_makereport钩子函数就行了
通过测试执行状态判断再进行失败截图
@pytest.hookimpl(tryfirst=True,?hookwrapper=True)
def?pytest_runtest_makereport(item,?call):
????"""
????pytest?失败后执行
????:param?item:?测试用例
????:param?call:?测试步骤
????:return:
????"""
????out?=?yield
????result?=?out.get_result()
????logger.info(f"测试报告:{result}")
????logger.info(f"执行耗时:{call.duration}")
????if?result.outcome?in?['failed',?'error']:
????????for?k,?v?in?item.funcargs.items():
????????????if?hasattr(v,?'driver'):
????????????????attach_png(f'{TEST_PIC}/{int(time.time())}.png',?"失败截图",?v)
????????????????break
这个钩子传入了一个item对象，通过简单的调试就可以发现，我们需要的页面操作对象也在里面，所以我们可以方便的进行截图。
所以，理所当然的打开pytest-assume的源码看一下，有没有相关的钩子可以使用
def?pytest_assume_fail(lineno,?entry):
????"""
????Hook?to?manipulate?user-defined?data?in-case?of?assumption?failure.
????lineno:?Line?in?the?code?from?where?asumption?failed.
????entry:?The?assumption?failure?message?generated?from?assume()?call
????"""
????pass
找到了pytest_assume_fail，这下离我们失败截图这个需求，就差一步了，也就是拿到页面操作对象。
但是调试一下我们就可以发现，它给我们的两个参数lineno和entry
我们写一个简单的测试脚本查看一下
import?pytest
def?test_a():
????pytest.assume(0,?"错了")
lineno:失败的行数
11
entry:失败的内容
test_1.py:11:?AssumptionFailure
>>????pytest.assume(0,?"错了1")
AssertionError:?错了1
assert?0
测试函数中页面操作对象会通过fixture的方式传入，所以在流程中它肯定是能找到的
这时就比较容易想到常用的两个方法
locals()globals()
然而，找了一圈还是没找到页面操作的对象
这时候我在pytest-assume中发现了
(frame,?filename,?line,?funcname,?contextlist)?=?inspect.stack()[stack_level][0:5]
这样，就找到了这个页面操作对象！
现在就是仿照一下这段代码了[[inspect -检查对象#使用inspect查看调用堆栈]]
import?inspect
inspect.stack()
我们查看一下效果
它会依次打印我们的执行堆栈信息
我们根据测试脚本的名字，就可以很方便的找到页面对象了！
最终效果
确定当前测试脚本位置
遍历调用堆栈列表
根据脚本名称找到对应列表索引
查看是否有页面操作对象
失败截图并跳出
def?pytest_assume_fail(lineno,?entry):
????file_name?=?os.getcwd()?+?'/test_'
????for?i?in?inspect.stack():
????????if?file_name?in?i.filename:
????????????try:
????????????????for?k,?v?in?i.frame.f_locals.items():
????????????????????if?hasattr(v,?'driver'):
????????????????????????attach_png(f'{TEST_PIC}/{int(time.time())}.png',?"失败截图",?v)
????????????????????????break
????????????except?Exception:
????????????????pass