Làm thế nào để khẳng định rằng một phương pháp được trang trí với python unittest?

Question

Tôi có một người trang trí và tôi muốn khẳng định rằng một số phương pháp nhất định trong mã của tôi được trang trí với nó.

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 

Làm thế nào để khẳng định với unittest (unitttest2) mà my_method có @decorator và không ai lấy nó, và nó đã không bị lãng quên?

Answer 1

Bạn có thể làm điều đó bằng cách dựa vào trang trí của bạn để đánh dấu chức năng trình bao bọc với thuộc tính, sau đó bạn xác nhận.

Thực hành tốt là để người trang trí đặt thuộc tính __wrapped__ trỏ đến hàm ban đầu trên trình bao bọc được trả về.

như sau:

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 

và sau đó, trên mã thử nghiệm của bạn:

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

Answer 2

Nếu vì một số lý do bạn không thể sửa đổi trang trí, bạn cũng có thể thử kiểm tra đối với một số đặc trưng của một biến đóng.

Trong ví dụ của bạn, bạn biết rằng bản gốc my_method là biến chỉ đóng bởi các trang trí, vì vậy bạn có thể:

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