Fix for Mock.configure_mock

mock.py

"""
Extensions to the standard Python mock library.
Use it instead of the library itself.
"""
from __future__ import absolute_import

try:
    from unittest.mock import *
except ImportErorr:
    from mock import *


class _ConfigureMockMixin(object):
    """Mixin for mock classes that makes :meth:`configure_mock`
    return the mock object.

    This allows to unambiguously separate attributes to be set on mock
    from the regular ``*Mock`` constructor parameters, like ``name``
    or ``return_value``, while still keeping the mock definition as
    a single expression (and thus making it usable e.g. as ``new=`` argument
    for :func:`mock.patch`).

    Example::

        mock_foo = Mock(name="Foo mock name") \
            .configure_mock(name="Foo object name")
    """
    def configure_mock(self, **attrs):
       super(_ConfigureMockMixin, self).configure_mock(**attrs)
       return self


# Mock classes

class MagicMock(_ConfigureMockMixin, MagicMock):
    pass


class Mock(_ConfigureMockMixin, Mock):
    pass


class NonCallableMagicMock(_ConfigureMockMixin, NonCallableMagicMock):
    pass


class NonCallableMock(_ConfigureMockMixin, NonCallableMock):
    pass


del _ConfigureMockMixin