如何使用 unittest.mock 在 python class 上测试实例方法的实现？

Question

I have the following class:

class RefdataMapping(object):
    def _init_(LONG_LIST_OF_ARGUMENTS):
        # Complex stuff

    def is_hybrid_default_mapping(self, mapping):
       # Code I want to test

I want to test the implementation of is_hybrid_default_mapping without having to completely instantiate an object of RefdataMapping.

My humble stab at this have resulted in this:

def test_is_hybrid_default_mapping():
    mappings = [{"a": "*", "b": "*", "c": "*"}]
    mock = Mock(spec=RefDataMapping)
    res = mock.is_hybrid_default_mapping(mappings[0])
    assert res == False

but since res only returns an instance of the mocked method, the test fails.

FAILED tests/test_basic.py::test_is_hybrid_default_mapping - AssertionError: assert <Mock name='mock.is_hybrid_default_mapping()' id='139681656266272'> == False

How do I call the method without having to instantiate the class with all its complexity without adding inheritance or interfaces to this implementation?

Answer 1

Thanks to @jonrsharpe , this turned out to be really simple:

Call the method using the mocked object as the self argument.

def test_is_hybrid_default_mapping():
    mappings = [{"a": "*", "b": "*", "c": "*"}]
    mock = Mock(spec=RefDataMapping)
    res = RefDataMapping.is_hybrid_default_mapping(mock, mappings[0])
    assert res == False