Bir yöntemin python unittest ile süslendiğini nasıl iddia edebilirim?

Question

Bir dekoratörüm var ve kodumdaki bazı yöntemlerin onunla süslendiğini iddia etmek istiyorum.

import functools 

def decorator(func): 
    def _check_something(*args, **kwargs): 
     # some logic in here 
     return func(*args, **kwargs) 
    return functools.wraps(func)(_check_something) 

class MyClass(object): 

    @decorator 
    def my_method(foo, bar): 
     pass 

nasıl my_method@decorator ve kimse kaldırdığını ve bu unutulmuş değildi sahip olduğunu unittest (unitttest2) ile iddia ederim?

Answer 1

Sarıcı işlevini bir öznitelikle işaretlemek için dekoratörünüze güvenerek bunu yapabilirsiniz.

Dekoratör, iade paketindeki orijinal işlevin işaret ettiği bir __wrapped__ özniteliği ayarlamalıdır.

böylece:

def decorator(func): 
    @functools.wraps(func) 
    def _check_something(*args, **kwargs): 
     # some logic in here 
     return func(*args, **kwargs) 
    _check_something.__wrapped__ = func # <== add this 
    return _check_something 

ardından, test kodu: Herhangi bir nedenle dekoratör değiştiremezsiniz Eğer

assert getattr(MyClass.my_method, "__wrapped__").__name__ == 'my_method' 

Answer 2

, ayrıca bir bazı karakteristik kontrol deneyin olabilir kapalı değişken. Örnekte

, özgün my_method dekoratör tarafından kapatıldı tek değişken olduğunu biliyoruz, bu yüzden olabilir:

assert (my_method.__closure__ and 
      my_method.__closure__[0].cell_contents.__name__ == my_method.__name__)