我正在尝试修补执行数据库查询并返回如下对象的上下文管理器：
class MyClass:
@contextlib.contextmanager
def client_ctx(self, id):
# hidrate from DB and yield object
yield client # instance of SQAlchemy model Client
def run(self, id):
with self.client_ctx(id) as cl:
# do work here
在这种情况下，客户端类是SQLAlchemy模型。
在我的测试中，我试图修补此方法，client_ctx以仅返回在测试中实例化的对象，如下所示：
@patch('MyClass.client_ctx')
def test_ctx(self, _mocked_ctx_manager):
myclass = MyClass()
client = Client(
id=1,
code='test-client')
_mocked_ctx_manager.__enter__.return_value = client
myclass.run(1)
我明白了：TypeError: Object of type MagicMock is not JSON serializable这对我来说毫无意义。我在做什么错，有没有更好的模拟上下文管理器的方法？
以下应该工作：
_mocked_ctx_manager.return_value.__enter__.return_value = client
您的_mocked_ctx_manager返回上下文管理器。因此，你需要设置__enter__.return_value的_mocked_ctx_manager.return_value。
我发现以下文章会有所帮助：投降Python模拟！我现在有你。
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句