Question

我创建了一个自定义错误处理程序来处理失败时的屏幕截图

#error_handler.py

def screenshot_handler(func):
    def func_wrapper(self):
        try:
            return func(self)
        except Exception as e:
            print("screenshot") 
            return func(self)
    return func_wrapper

#page.py

@screenshot_handler
def assert_login(self,a):
    self.find_element(*DashboardPageLocators.AUTOREFRESH_BUTTON)
    return True

#steps.py
"""
i forced this to fail
"""

@then('map should display')
def step_impl(context):
    page = LoginPage(context)
    page.assert_login()

我想抓住步骤参数'map应该显示'，所以我创建了它的文件名。最明显的选择是复制字符串，但这样效率很低，是否有一个行为函数，我可以调用它来处理这个

Answer 1

我在ambient.py：

的after_step函数中执行此操作

def after_step(context, step):
    if step.status == 'failed':
        step_str = step.name
        # Take screen shot and upload to s3

您可以立即访问测试状态和步骤名称，而不必包含自己的错误处理装饰器。

step变量中有许多漂亮的东西：

(Pdb) pp(dir(step))

['__class__',
 '__delattr__',
 '__dict__',
 '__dir__',
 '__doc__',
 '__eq__',
 '__format__',
 '__ge__',
 '__getattribute__',
 '__gt__',
 '__hash__',
 '__init__',
 '__init_subclass__',
 '__le__',
 '__lt__',
 '__module__',
 '__ne__',
 '__new__',
 '__reduce__',
 '__reduce_ex__',
 '__repr__',
 '__setattr__',
 '__sizeof__',
 '__str__',
 '__subclasshook__',
 '__weakref__',
 'duration',
 'error_message',
 'exc_traceback',
 'exception',
 'filename',
 'keyword',
 'line',
 'location',
 'name',
 'replay',
 'reset',
 'run',
 'set_values',
 'status',
 'step_type',
 'store_exception_context',
 'table',
 'text',
 'type']

步骤字符串只是test.name

Python-Behave中屏幕截图的自定义错误处理程序

1 个答案: