Використання макетної виправлення для знущання над методом екземпляра

Я намагаюся щось знущатись під час тестування програми Django, використовуючи образотворчу бібліотеку Mock testing . Здається, я не можу змусити це працювати, я намагаюся зробити це:

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))

Що я роблю не так?

Щоб додати до відповіді Кіта, вказавши 3-й аргумент, що patch.object()дозволяє вказати знущаний об’єкт / метод. В іншому випадку використовується MagicMockоб’єкт за замовчуванням .

    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!

Зверніть увагу , що, якщо ви вкажете знущальний об'єкт, то за замовчуванням MagicMock()буде більше не передається в виправленої об'єкт - наприклад , більше не:

def test_enter_promotion(self, mock_method):

але замість цього:

def test_enter_promotion(self):

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

Ах, я був збентежений, де застосувати декоратор патчів. Виправлено:

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))