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()