問題描述

我熟悉其他語言中的其他模擬庫，例如 Java 中的 Mockito，但 Python 的 mock 庫讓我感到困惑.

我想測試以下課程.

類 MyClassUnderTest(object):def 子方法(自我，*args):do_dangerous_things()def main_method(self):self.submethod("什么都沒有.")

在我的測試中，我想確保在執行 main_method 時調用了 submethod 并且使用正確的參數調用它.我不希望 submethod 運行，因為它會做危險的事情.

我完全不確定如何開始使用它.Mock 的文檔非常難以理解，我不確定要模擬什么或如何模擬它.

我怎樣才能模擬 submethod 函數，同時保留 main_method 中的功能?

我想你要找的是 mock.patch.object

用 mock.patch.object(MyClassUnderTest, "submethod") 作為 submethod_mocked:submethod_mocked.return_value = 13MyClassUnderTest().main_method()submethod_mocked.assert_call_once_with(user_id, 100, self.context,self.account_type)

這里是小描述

 patch.object(目標, 屬性, new=DEFAULT,規格=無，創建=假，規格集=無，autospec=None, new_callable=None, **kwargs)

<塊引用>

使用模擬對象修補對象(目標)上的命名成員(屬性).

I'm familiar with other mocking libraries in other languages such as Mockito in Java, but Python's mock library confuses the life out of me.

I have the following class which I would like to test.

class MyClassUnderTest(object):

    def submethod(self, *args):
       do_dangerous_things()

    def main_method(self):
       self.submethod("Nothing.")

In my tests, I'd like to make sure that the submethod was called when main_method was executed and that it was called with the right arguments. I don't want submethod to run, as it does dangerous things.

I'm entirely unsure as to how to get started with this. Mock's documentation is incredibly hard to understand and I'm not sure what to even mock or how to mock it.

How can I mock the submethod function, while leaving the functionality in main_method alone?

I think what you are looking for is mock.patch.object

with mock.patch.object(MyClassUnderTest, "submethod") as submethod_mocked:
    submethod_mocked.return_value = 13
    MyClassUnderTest().main_method()
    submethod_mocked.assert_called_once_with(user_id, 100, self.context,
                                             self.account_type)

Here is small description

 patch.object(target, attribute, new=DEFAULT, 
              spec=None, create=False, spec_set=None, 
              autospec=None, new_callable=None, **kwargs)

patch the named member (attribute) on an object (target) with a mock object.

這篇關于僅模擬對象上的單個方法的文章就介紹到這了，希望我們推薦的答案對大家有所幫助，也希望大家多多支持html5模板網！