Playing with Mock

mocked.py

import mock


class Car(object):

    def shift_down(self):
        print "Down"

    def shift_up(self):
        print "Up"

with mock.patch('__main__.Car') as mocked_car:
    my_car = mocked_car()
    my_car.shift_up()
    my_car.shift_down()
    my_car.foo()

    print my_car.method_calls


def hello():
    return "Hello"


with mock.patch.object(Car, 'shift_down') as mocked_shift_down:
    mocked_shift_down.return_value = hello()
    my_car = Car()
    my_car.shift_up()
    print my_car.shift_down()
    my_car.foo()

    print my_car.method_calls