亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

Python補丁上下文管理器返回對象

Python補丁上下文管理器返回對象

Helenr 2022-07-26 11:04:49
我正在嘗試修補執行數據庫查找并返回如下對象的上下文管理器: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這對我來說毫無意義。我做錯了什么,有沒有更好的方法來模擬上下文管理器?
查看完整描述

1 回答

?
30秒到達戰場

TA貢獻1828條經驗 獲得超6個贊

以下應該有效:

_mocked_ctx_manager.return_value.__enter__.return_value = client

_mocked_ctx_manager返回一個上下文管理器。因此,您需要設置__enter__.return_valueof _mocked_ctx_manager.return_value。

我發現以下文章很有幫助:Surrender Python Mocking!我現在有你。


查看完整回答
反對 回復 2022-07-26
  • 1 回答
  • 0 關注
  • 85 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號