Chế giễu một lớp: Mock () hay patch ()?

Tôi đang sử dụng giả lập với Python và tự hỏi cách tiếp cận nào trong hai cách tiếp cận đó tốt hơn (đọc: thêm pythonic).

Phương pháp một : Chỉ cần tạo một đối tượng giả và sử dụng nó. Mã trông giống như:

def test_one (self):
    mock = Mock()
    mock.method.return_value = True 
    self.sut.something(mock) # This should called mock.method and checks the result. 
    self.assertTrue(mock.method.called)

Phương pháp hai : Sử dụng bản vá để tạo mô hình. Mã trông giống như:

@patch("MyClass")
def test_two (self, mock):
    instance = mock.return_value
    instance.method.return_value = True
    self.sut.something(instance) # This should called mock.method and checks the result. 
    self.assertTrue(instance.method.called)

Cả hai phương pháp đều làm điều tương tự. Tôi không chắc về sự khác biệt.

Bất cứ ai có thể khai sáng cho tôi?

mock.patchlà một sinh vật rất khác với mock.Mock. patch thay thế lớp bằng một đối tượng giả và cho phép bạn làm việc với thể hiện giả. Hãy xem đoạn mã này:

>>> class MyClass(object):
...   def __init__(self):
...     print 'Created MyClass@{0}'.format(id(self))
... 
>>> def create_instance():
...   return MyClass()
... 
>>> x = create_instance()
Created MyClass@4299548304
>>> 
>>> @mock.patch('__main__.MyClass')
... def create_instance2(MyClass):
...   MyClass.return_value = 'foo'
...   return create_instance()
... 
>>> i = create_instance2()
>>> i
'foo'
>>> def create_instance():
...   print MyClass
...   return MyClass()
...
>>> create_instance2()
<mock.Mock object at 0x100505d90>
'foo'
>>> create_instance()
<class '__main__.MyClass'>
Created MyClass@4300234128
<__main__.MyClass object at 0x100505d90>

patchthay thế MyClasstheo cách cho phép bạn kiểm soát việc sử dụng lớp trong các hàm mà bạn gọi. Sau khi bạn vá một lớp, các tham chiếu đến lớp được thay thế hoàn toàn bằng thể hiện giả.

mock.patchthường được sử dụng khi bạn đang kiểm tra một thứ gì đó tạo ra một phiên bản mới của một lớp bên trong kiểm tra. mock.Mockcác trường hợp rõ ràng hơn và được ưu tiên hơn. Nếu self.sut.somethingphương thức của bạn tạo một thể hiện MyClassthay vì nhận một thể hiện dưới dạng một tham số, thì mock.patchở đây sẽ phù hợp.

Tôi có một video YouTube về điều này.

Câu trả lời ngắn gọn: Sử dụng mockkhi bạn đang đi qua thứ mà bạn muốn chế nhạo, và patchnếu bạn không. Trong số hai, mô hình giả được ưa thích hơn hẳn vì nó có nghĩa là bạn đang viết mã bằng cách tiêm phụ thuộc thích hợp.

Ví dụ ngớ ngẩn:

# Use a mock to test this.
my_custom_tweeter(twitter_api, sentence):
    sentence.replace('cks','x')   # We're cool and hip.
    twitter_api.send(sentence)

# Use a patch to mock out twitter_api. You have to patch the Twitter() module/class 
# and have it return a mock. Much uglier, but sometimes necessary.
my_badly_written_tweeter(sentence):
    twitter_api = Twitter(user="XXX", password="YYY")
    sentence.replace('cks','x') 
    twitter_api.send(sentence)