py.test - 如何使用上下文管理器在funcarg /夾具

Question

密切相關的：In python, is there a good idiom for using context managers in setup/teardown

---

我有一個在測試中使用固定的時間/時區上下文管理。我想要一個pytest funcarg（或夾具，我們使用pytest 2.2.3，但我可以向後翻譯）。我可以這樣做：

def pytest_funcarg__fixedTimezone(request): 
    # fix timezone to match Qld, no DST to worry about and matches all 
    # Eastern states in winter. 
    fixedTime = offsetTime.DisplacedRealTime(tz=' Australia/Brisbane') 

    def setup(): 
     fixedTime.__enter__() 
     return fixedTime 

    def teardown(fixedTime): 
     # this seems rather odd? 
     fixedTime.__exit__(None, None, None) 

...但它有點噁心。在相關的Q jsbueno中指出：問題是如果發生異常，代碼沒有規定可以正確調用對象的__exit__方法。

His answer使用元類方法。但是這對於pytest來說並不是那麼有用，在pytest中，測試通常只是函數，而不是類。那麼解決這個問題的最好方法是什麼？涉及runtest hooks的內容？

Answer 1

恐怕目前沒有在燈具中使用上下文管理器的優雅方式。但是，如果測試失敗的終結將運行：

import contextlib, pytest 

@contextlib.contextmanager 
def manager(): 
    print 'manager enter' 
    yield 42 
    print 'manager exit' 

@pytest.fixture 
def fix(request): 
    m = manager() 
    request.addfinalizer(lambda: m.__exit__(None, None, None)) 
    return m.__enter__() 

def test_foo(fix): 
    print fix 
    raise Exception('oops') 

如果你運行這個與pytest -s，你會看到__exit__()調用發生。

Answer 2

由於2.4,py.test有yield款式夾具支持。我們可以直接在其內部使用with上下文。

@pytest.yield_fixture 
def passwd(): 
    with open("/etc/passwd") as f: 
     yield f.readlines() 

自3.0，py.test棄用@pytest.yield_fixture使用。我們可以直接使用@pytest.fixture作爲上下文管理器。

@pytest.fixture 
def passwd(): 
    with open("/etc/passwd") as f: 
     yield f.readlines()