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

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 should display'，所以我从中创建了文件名。最明显的选择是复制字符串，但这效率很低，有没有我可以调用的行为函数来处理这个

Answer 1

我在 environments.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 中屏幕截图的自定义错误处理程序

Custom error handler for screenshots in Python-Behave

python

bdd

selenium

python-behave