Mocking imported modules in Python

Question

I'm trying to implement unit tests for function that uses imported external objects.

For example helpers.py is:

import os
import pylons

def some_func(arg):
   ...
   var1 = os.path.exist(...)
   var2 = os.path.getmtime(...)
   var3 = pylons.request.environ['HTTP_HOST']
   ...

So when I'm creating unit test for it I do some mocking (minimock in my case) and replacing references to pylons.request and os.path:

import helpers
def test_some_func():
    helpers.pylons.request = minimock.Mock("pylons.request")
    helpers.pylons.request.environ = { 'HTTP_HOST': "localhost" }
    helpers.os.path = minimock.Mock(....)
    ...
    some_func(...)
    # assert
    ...

This does not look good for me.

Is there any other better way or strategy to substitute imported function/objects in Python?

Almad · Accepted Answer · 2010-03-04 18:39:09Z

2

Use voidspace's mocking library and it's patching/wrapping ability.

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

answered Mar 4, 2010 at 18:39

Almad

5,9437 gold badges38 silver badges53 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Peter Lyons · Accepted Answer · 2010-03-12 04:17:45Z

1

Well, in minimock there is an easier paradigm for this than what you are using above:

>>> from minimock import mock
>>> import os.path
>>> mock('os.path.isfile', returns=True)

See http://pypi.python.org/pypi/MiniMock#creating-mocks

Once you do that, any module that does os.path.isfile("blah") is going to get True back. You don't need to go and explicitly reassign the module-under-test's namespace.

answered Mar 12, 2010 at 4:17

Peter Lyons

147k32 gold badges285 silver badges281 bronze badges

Collectives™ on Stack Overflow

Mocking imported modules in Python

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related