publish muk_utils - 12.0

muk_utils/__manifest__.py

@@ -19,7 +19,7 @@
 { 
     "name": "MuK Utils",
     "summary": """Utility Features""",
-    "version": '12.0.1.6.20',   
+    "version": '12.0.1.6.21',   
     "category": 'Extra Tools',   
     "license": "AGPL-3",
     "author": "MuK IT",

muk_utils/tests/common.py

@@ -105,9 +105,11 @@ def multi_users(users=[['base.user_root', True], ['base.user_admin', True]], res
         return wrapper
     return decorator
 
-def track_function(func):
+def track_function(max_query_count=None, max_query_time=None, max_time=None, return_tracking=False):
+    def decorator(func):
         @functools.wraps(func)
         def wrapper(*args, **kwargs):
+            tracking_parameters = [func.__name__]
             threading.current_thread().query_time = 0
             threading.current_thread().query_count = 0
             threading.current_thread().perf_t0 = time.time()
@@ -124,9 +126,21 @@ def track_function(func):
                 message += " - %s Q %.3fs QT %.3fs OT %.3fs TT" % (
                     query_count, query_time, remaining_time, time_taken
                 )
+                tracking_parameters += [
+                    query_count, query_time, remaining_time, time_taken
+                ]
+                if max_query_count and query_count > max_query_count:
+                    raise AssertionError("More than %s queries" % max_query_count)
+                if max_query_time and query_time > max_query_time:
+                    raise AssertionError("Queries took longer than %.3fs" % max_query_time)
+                if max_time and time_taken > max_time:
+                    raise AssertionError("Function took longer than %.3fs" % max_time)
             _logger.info(message)
+            if return_tracking:
+                return result, tracking_parameters
             return result
         return wrapper
+    return decorator
 
 #----------------------------------------------------------
 # Test Cases