29 lines
840 B
Python
29 lines
840 B
Python
|
import importlib
|
||
|
|
from functools import wraps
|
||
|
|
|
||
|
|
|
||
|
|
def patch_test(original_method):
|
||
|
|
"""
|
||
|
|
A decorator that patches an Odoo test method with a new one.
|
||
|
|
The original_method is a direct reference to the method to be patched.
|
||
|
|
"""
|
||
|
|
def decorator(new_method):
|
||
|
|
@wraps(new_method)
|
||
|
|
def wrapper(*args, **kwargs):
|
||
|
|
return new_method(*args, **kwargs)
|
||
|
|
|
||
|
|
# Extract module and class names
|
||
|
|
module_name = original_method.__module__
|
||
|
|
class_name = original_method.__qualname__.split('.')[0]
|
||
|
|
|
||
|
|
# Import the module
|
||
|
|
module = importlib.import_module(module_name)
|
||
|
|
# Get the class
|
||
|
|
cls = getattr(module, class_name)
|
||
|
|
|
||
|
|
# Replace the original method with the new one
|
||
|
|
setattr(cls, original_method.__name__, wrapper)
|
||
|
|
|
||
|
|
return wrapper
|
||
|
|
return decorator
|