Làm thế nào để vá classmethod với autospec trong lớp unmocked?

Question

Tôi muốn khẳng định rằng một classmethod trong một lớp Python gọi một classmethod với một tập hợp các đối số nhất định. Tôi muốn classmethod giả lập là "spec-ed", vì vậy nó phát hiện nếu nó được gọi với số lượng sai các đối số.

Khi tôi vá lớp học bằng cách sử dụng patch.object(.., autospec=True, ..), classmethod được thay thế bằng NonCallableMagicMock và tăng lỗi khi tôi cố gắng gọi nó.

from mock import patch 

class A(object): 

    @classmethod 
    def api_meth(cls): 
     return cls._internal_classmethod(1, 2, 3) 

    @classmethod 
    def _internal_classmethod(cls, n, m, o): 
     return sum(n, m, o) 

with patch.object(A, '_internal_classmethod') as p: 
    print(type(p).__name__) 

with patch.object(A, '_internal_classmethod', autospec=True) as p: 
    print(type(p).__name__) 

sản xuất đầu ra:

MagicMock 
NonCallableMagicMock 

Làm thế nào tôi có thể có được một mô hình đặc tả-ed cho _internal_classmethod khi lớp nó thuộc về không được chế giễu?

Answer 1

Sử dụng spec thay cho autospec và đặt trực tiếp.

with patch.object(A, '_internal_classmethod', spec=A._internal_classmethod) as p: 
    print(type(p).__name__) 

mang lại cho tôi

MagicMock 

cho đầu ra.

Answer 2

Có báo cáo lỗi nổi bật (google code link và python bug tracker link) để khắc phục vấn đề này. Cho đến khi bản sửa lỗi được kết hợp, bạn có thể thử những điều sau, đã làm việc cho tôi [On 2.7, mặc dù tôi nghĩ rằng nó cũng sẽ hoạt động trong 3.x].

def _patched_callable(obj): 
    "Monkeypatch to allow autospec'ed classmethods and staticmethods." 
    # See https://code.google.com/p/mock/issues/detail?id=241 and 
    # http://bugs.python.org/issue23078 for the relevant bugs this 
    # monkeypatch fixes 
    if isinstance(obj, type): 
     return True 
    if getattr(obj, '__call__', None) is not None: 
     return True 
    if (isinstance(obj, (staticmethod, classmethod)) 
     and mock._callable(obj.__func__)): 
     return True 
    return False 
_patched_callable._old_func = mock._callable 
mock._callable = _patched_callable 

Sau khỉ, bạn có thể sử dụng mock.patch bình thường và có phương pháp tĩnh và lớp được vá đúng cách.