直接对实例的 __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 惯例的解决方案。
