TypeError: несвязанный метод при попытке издеваться над classmethod

этот сценарий терпит неудачу:

import mock

class MyClass(object):

    @classmethod
    def my_method(cls):
        print('my_method')

def mocked_method(cls):
    print('I want this method to get called')

with mock.patch.object(MyClass, 'my_method', mocked_method):
    MyClass.my_method()

исключения:

Traceback (most recent call last):
  File "/home/foo/tmp/test_mocking_classmethod.py", line 14, in <module>
    MyClass.my_method()
TypeError: unbound method mocked_method() must be called with MyClass instance as first argument (got nothing instead)

1 ответов


функции Python дескрипторов и Python связывает их с экземпляром, на который они смотрят, или в случае classmethod, к классу. Потому что вы не использовали classmethod декоратор на функции замены, он связан неправильно (как обычный метод вместо этого, поэтому нет cls передается).

просто оберните цель в classmethod декоратор вручную:

with mock.patch.object(MyClass, 'my_method', classmethod(mocked_method)):
    MyClass.my_method()

здесь я применила @classmethod декоратор вручную, но вы могли бы также просто используйте его по назначению, как декоратор, непосредственно на целевой функции:

@classmethod
def mocked_method(cls):
    print('I want this method to get called')

with mock.patch.object(MyClass, 'my_method', mocked_method):
    MyClass.my_method()

демо:

>>> import mock
>>> class MyClass(object):
...     @classmethod
...     def my_method(cls):
...         print('my_method')
... 
>>> def mocked_method(cls):
...     print('I want this method to get called')
... 
>>> with mock.patch.object(MyClass, 'my_method', classmethod(mocked_method)):
...     MyClass.my_method()
... 
I want this method to get called