Mock patching from/import statement in Python Mock patching from/import statement in Python python python

Mock patching from/import statement in Python


If you're patching something in the same module, you can use __main__:

from mock import patchfrom collections import defaultdictwith patch('__main__.defaultdict'):    d = defaultdict()    print 'd:', d

If you're mocking something for an imported module, however, you'll want to use that module's name so the correct reference (or name) is patched:

# foo.pyfrom collections import defaultdictdef bar():    return defaultdict()# foo_test.py    from mock import patchfrom foo import barwith patch('foo.defaultdict'):    print bar()

The point here is that patch wants the full path to the thing it is patching. This just looks a little weird when patching something in the current module, since folks don't often use __main__ (or have to refer to the current module, for that matter).


patch works by patching names. You can't achieve anything by patching the name collections.defaultdict if you are using the name defaultdict (in the local namespace) to access the object. See the documentation at http://www.voidspace.org.uk/python/mock/patch.html#id1 .


The names can be very confusing in this case. We always want to mock a class definition in a namespace. The namespace is the module in which the import takes place. The class definition is the name used in that namespace.

Let's take a concrete example:

  • myproj.utilities module contains Actor class
  • myproj.application imports this as from myproj.utilities import Actor
  • my test should execute my.proj.application and mock our Actor

myproj.utilities.py

class Actor:    def __init__(name):        self.name = name

myproj.application.py

from myproj.utilities import Actorclass App:    def __init__(name):        self.actor = Actor(name)

test code

from mock import patchfrom myproj.application import Apptest:  # format: patch('<namespace>.<Class>')  # the namespace in which we with to mock  # the class definition we wish to mock  with patch('myproj.application.Actor'):      app = App('Someone')      print( type(app.actor) ) # expect a MagicMock

I tried several other approaches and this one works well for me. I have not tests the code above but instead generalized it from my own specific case. So, it might be a little off.