將常見屬性添加到Behave方法

Question

使用偉大的Behave框架，但遇到我缺乏OOP技能的麻煩。

行爲具有內置的上下文命名空間，其中可以在測試執行步驟之間共享對象。在初始化我的WebDriver會話之後，我一直使用這個context來保存所有內容。功能很好，但正如你可以在下面看到的，它是乾的。

如何/在哪裏可以將這些屬性添加到step_impl()或context一次？

environment.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') 

這將減少重複在你的代碼，但它是非常隱含，可能是危險的。所以不建議這樣做。

我只是使用tuple解包。它很簡單明確，所以不會引起額外的錯誤。

Answer 2

你可以定義一個裝飾器 '解壓' 的背景下，爲您和傳遞 '解壓' 值作爲參數：

environment.py

def before_feature(context, feature): 
    context.spam = 'spam' 

def after_feature(context, feature): 
    del context.spam 

test.feature

Scenario: Test global env 
    Then spam should be "spam" 

步驟。PY

def add_context_attrs(func): 
    @functools.wraps(func) # wrap it neatly 
    def wrapper(context, *args, **kwargs): # accept arbitrary args/kwargs 
     kwargs['spam'] = context.spam # unpack 'spam' and add it to the kwargs 
     return func(context, *args, **kwargs) # call the wrapped function 
    return wrapper 

@step('spam should be "{val}"') 
@add_context_attrs 
def assert_spam(context, val, spam): 
    assert spam == val 

Answer 3

堅持到乾燥的原則，我通常使用：
*背景故事：http://pythonhosted.org/behave/gherkin.html#background
*或環境控制：http://pythonhosted.org/behave/tutorial.html#environmental-controls