python - 将公共(public)属性添加到 Behave 方法

Question

使用伟大的Behave框架，但由于缺乏 OOP 技能而遇到麻烦。

Behave 有一个内置的上下文命名空间，其中的对象可以在测试执行步骤之间共享。在初始化我的 WebDriver session 之后，我一直在我的步骤之间传递它，使用这个 context 来保存所有内容。功能很好，但正如您在下面看到的，它一点也不干。

如何/在哪里可以将这些属性添加到 step_impl() 或 context 一次？

环境.py

from selenium import webdriver

def before_feature(context, scenario):
    """Initialize WebDriver instance"""

    driver = webdriver.PhantomJS(service_args=service_args, desired_capabilities=dcap)

    """
    Do my login thing..
    """

    context.driver = driver
    context.wait = wait
    context.expected_conditions = expected_conditions
    context.xenv = env_data

steps.py

@given('that I have opened the blah page')
def step_impl(context):

    driver = context.driver
    wait = context.wait
    expected_conditions = context.expected_conditions
    xenv = context.xenv

    driver.get("http://domain.com")
    driver.find_element_by_link_text("blah").click()
    wait.until(expected_conditions.title_contains("Blah page"))

@given(u'am on the yada subpage')
def step_impl(context):
    driver = context.driver
    wait = context.wait
    expected_conditions = context.expected_conditions
    xenv = context.xenv

    if driver.title is not "MySubPage/":
        driver.get("http://domain.MySubPage/")
        wait.until(expected_conditions.title_contains("Blah | SubPage"))

@given(u'that I have gone to another page')
def step_impl(context):
    driver = context.driver
    wait = context.wait
    expected_conditions = context.expected_conditions
    xenv = context.xenv

    driver.get("http://domain.com/MyOtherPahge/")

Answer 1

首先你可以跳过这个解包并在任何地方使用 context 属性，比如 context.driver.get("http://domain.com")

如果你不喜欢它并且你真的想要局部变量，你可以使用元组解包来使代码更好:

import operator
def example_step(context):
    driver, xenv = operator.attrgetter('driver', 'xenv')(context)

你可以像这样分解出默认的属性列表，但这让整个事情有点含蓄:

import operator

def unpack(context, field_list=('driver', 'xenv')):
    return operator.attrgetter(*field_list)(context)

def example_step(context):
    driver, xenv = unpack(context)

如果您仍然不喜欢它，您可以使用 globals() 进行修改。例如创建一个这样的函数:

def unpack(context, loc, field_list):
    for field in field_list:
        loc[field]  = getattr(context, field, None)

并在您的步骤中使用它:

def example_step(context):
    unpack(context, globals(), ('driver', 'xenv'))

    # now you can use driver and xenv local variables
    driver.get('http://domain.com')

这将减少代码中的重复，但它非常隐含并且可能很危险。所以不建议这样做。

我只会使用元组解包。它简单明了，因此不会导致其他错误。