2016-01-06 6 views
4

Чтобы программно объединить менеджеров контекста, я использую следующий код:python2.7 contextlib.ExitStack эквивалент

== helpers.py ==

from contextlib import nested 
import mock 

def multiple_patch(obj_to_be_patch, *methods): 
    return nested(
     *[mock.patch.object(obj_to_be_patch, method) for method in methods] 
    ) 

== tests.py ==

def test_foo(self): 
    with helpers.multiple_patch(Foo, "method1", "method2", "method3", "method3") as mocks: 
     mock_method1 = mocks[0] 
     .... 
     # asserts on mocks 

Поскольку я застрял в этой версии python, я не могу использовать contextlib.ExitStack и contextlib.nested устарел.

Благодаря

+0

Вы должны принять ответ. –

ответ

 Смежные вопросы

  • Нет связанных вопросов^_^