Add __all__ attribute to Mock class for docs

This commit is contained in:
Seth House 2015-05-18 17:23:33 -06:00
parent 5f1a93d966
commit 6eeca46158

View file

@ -24,6 +24,8 @@ class Mock(object):
def __init__(self, *args, **kwargs):
pass
__all__ = []
def __call__(self, *args, **kwargs):
ret = Mock()
# If mocked function is used as a decorator, expose decorated function.
@ -94,6 +96,7 @@ MOCK_MODULES = [
'MySQLdb',
'MySQLdb.cursors',
'psutil',
'psutil.version_info',
'pycassa',
'pymongo',
'rabbitmq_server',