Sử dụng bản vá giả để mô phỏng một phương thức cá thể

Tôi đang cố gắng mô phỏng thứ gì đó trong khi thử nghiệm ứng dụng Django bằng cách sử dụng thư viện thử nghiệm Mock được đặt tên theo trí tưởng tượng . Tôi dường như không thể làm cho nó hoạt động được, tôi đang cố gắng làm điều này:

models.py

from somelib import FooClass

class Promotion(models.Model):
    foo = models.ForeignKey(FooClass)
    def bar(self):
       print "Do something I don't want!"


test.py

class ViewsDoSomething(TestCase):
    view = 'my_app.views.do_something'

    def test_enter_promotion(self):
        @patch.object(my_app.models.FooClass, 'bar')
        def fake_bar(self, mock_my_method):
            print "Do something I want!"
            return True

        self.client.get(reverse(view))

Tôi đang làm gì sai?

Để thêm vào câu trả lời của Kit, hãy chỉ định đối số thứ 3 để patch.object()cho phép chỉ định đối tượng / phương thức bị chế nhạo. Nếu không, một MagicMockđối tượng mặc định được sử dụng.

    def fake_bar(self):
        print "Do something I want!"
        return True

    @patch.object(my_app.models.FooClass, 'bar', fake_bar)
    def test_enter_promotion(self):
        self.client.get(reverse(view))
        # Do something I want!

Lưu ý rằng, nếu bạn chỉ định các đối tượng chế giễu, sau đó mặc định MagicMock()là không còn thông qua vào đối tượng vá - ví dụ như không còn:

def test_enter_promotion(self, mock_method):

nhưng thay vì:

def test_enter_promotion(self):

http://www.voidspace.org.uk/python/mock/patch.html#patch-object

Ah Tôi đã bối rối về nơi để áp dụng trang trí vá lỗi đó. Đã sửa:

class ViewsDoSomething(TestCase):
    view = 'my_app.views.do_something'

    @patch.object(my_app.models.FooClass, 'bar')
    def test_enter_promotion(self, mock_method):
        self.client.get(reverse(view))