直接对实例的 __enter__ 或 __exit__ 方法赋值装饰后函数无效，因为 with 语句实际调用的是类定义中的绑定方法（而非实例属性）；需通过修改类方法或使用 unittest.mock.patch 动态替换才能生效。

直接对实例的 `__enter__` 或 `__exit__` 方法赋值装饰后函数无效，因为 `with` 语句实际调用的是类定义中的绑定方法（而非实例属性）；需通过修改类方法或使用 `unittest.mock.patch` 动态替换才能生效。

在 Python 中，with 语句的执行机制依赖于对象所属类上定义的 __enter__ 和 __exit__ 方法，而非实例字典中动态设置的同名属性。因此，即使你执行了：

mgr.__enter__ = deco(mgr.__enter__)

该赋值仅将装饰后的函数存入实例 mgr.__dict__，但 with mgr: 在底层仍会通过 type(mgr).__enter__(mgr) 查找并调用类方法，完全忽略实例上的覆盖——这就是为何你的 In wrapper 未在 __enter__/__exit__ 调用时打印。

✅ 正确做法：使用 unittest.mock.patch 动态修饰类方法

patch 是专为测试场景设计的安全方案：它临时替换类属性，并在退出上下文时自动还原，避免污染全局状态。以下是推荐实现：

from unittest.mock import patch

class contMgr():
    def __init__(self):
        print("In __init__")

    def __enter__(self):
        print("In __enter__")
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        print("In __exit__")

def deco(func):
    def wrapper(*args, **kwargs):
        print("In wrapper")
        return func(*args, **kwargs)
    return wrapper

def test():
    print("Blah, blah, blah")

# 创建实例
mgr = contMgr()
print("------------------------")

# ✅ 正确：patch 类方法（注意路径为 "__main__.contMgr.__enter__"）
with (
    patch("__main__.contMgr.__enter__", deco(contMgr.__enter__)),
    patch("__main__.contMgr.__exit__", deco(contMgr.__exit__))
):
    with mgr:
        test()

print("------------------------")
# ❌ 原始行为（无 patch）不受影响
with mgr:
    test()

? 关键细节：

• patch 的目标路径必须是类的全限定名（如 "__main__.contMgr.__enter__"），而非 mgr.__enter__；
• deco(contMgr.__enter__) 需传入未绑定的类方法（即 contMgr.__enter__），而非 mgr.__enter__（已绑定实例，会导致 self 重复传入）；
• 使用 with 形式 patch(...) 可确保异常安全还原，无需手动恢复。

⚠️ 注意事项与替代方案

• 不推荐永久修改类方法（如 contMgr.__enter__ = deco(contMgr.__enter__)）：会影响所有后续实例，破坏封装且难以维护；

• 避免 types.MethodType 手动绑定：虽技术上可行，但 with 仍优先查找类属性，无法绕过 MRO 查找逻辑；

• 若需长期装饰，应在类定义时直接装饰：

  class contMgr():
      @deco
      def __enter__(self): ...
  
      @deco
      def __exit__(self, ...): ...

  但此方式不满足“临时测试”需求。

总之，针对测试场景下的临时增强，unittest.mock.patch 是最简洁、可靠且符合 Python 惯例的解决方案。