带有语句的猴子补丁python

Question

我正在使用 py.test 进行我的 python 单元测试。考虑以下代码：

def mytest():
    "Test method"
    print "Before with statement"
    with TestClass('file.zip', 'r') as test_obj:
        print "This shouldn't print after patching."
        # some operation on object.
    print "After with statement."

是否可以对 TestClass 类进行猴子补丁，以便 with 块中的代码变为 noop？

比如打补丁后的输出应该是：

Before with statement
After with statement

我知道我可以修补 mytest 函数本身，但这是为了获得更好的测试覆盖率。

我已经尝试过，以下几行中的某些内容但无法正常工作。

class MockTestClass(object):
    def __init__(self, *args):
        print "__init__ called."

    def __enter__(self):
        print "__enter__ called."
        raise TestException("Yeah done with it! Get lost now.")

    def __exit__(self, type, value, traceback):
        print "__exit__ called."

module_name.setattr('TestClass',  MockTestClass)

Answer 1

从@Peter 的回答中可以清楚地看出，我们不能将整个块设为noop。 我最终为我的用例做了以下工作。

# Module foo.py
class Foo(object):
    def __init__(self):
        print "class inited"

    def __enter__(self):
        print "entered class"
        return None

    def foo(self):
        raise Exception("Not implemented")

    def __exit__(self, type, value, traceback):
        print "exited class"
        return True

----------------------------
# Module FooTest
import foo

class FooTest(object):
    def __init__(self):
        print "class inited"

    def __enter__(self):
        print "entered class"
        return None

    def __exit__(self, type, value, traceback):
        print "exited class"
        return True

try:
    foo.Foo()
    print "It shouldn't print"
except:
    print "Expected exception"
setattr(foo, 'Foo', FooTest)
print "Patched"
with foo.Foo() as a:
    a.foo()
    print "It shouldn't print"
print 'Test passed!'

Answer 2

我认为 Python 语言规范不允许您尝试做的事情。

正如您在PEP-343 中看到的，“with”语句的定义不允许任何提前退出上下文的尝试：

mgr = (EXPR)
exit = type(mgr).__exit__  # Not calling it yet
value = type(mgr).__enter__(mgr)
exc = True
try:
    try:
        VAR = value  # Only if "as VAR" is present
        BLOCK
    except:
        # The exceptional case is handled here
        exc = False
        if not exit(mgr, *sys.exc_info()):
            raise
        # The exception is swallowed if exit() returns true
finally:
    # The normal and non-local-goto cases are handled here
    if exc:
        exit(mgr, None, None, None)

有人提议将其更改为您需要的函数 (PEP-377)，但已被拒绝。