1 files changed, 43 insertions, 0 deletions
diff --git a/utils/libcxx/test/tracing.py b/utils/libcxx/test/tracing.py
new file mode 100644
index 000000000..c590ba3ef
--- /dev/null
+++ b/utils/libcxx/test/tracing.py
@@ -0,0 +1,43 @@
+#===----------------------------------------------------------------------===##
+#
+#                     The LLVM Compiler Infrastructure
+#
+# This file is dual licensed under the MIT and the University of Illinois Open
+# Source Licenses. See LICENSE.TXT for details.
+#
+#===----------------------------------------------------------------------===##
+
+import os
+import inspect
+
+
+def trace_function(function, log_calls, log_results, label=''):
+    def wrapper(*args, **kwargs):
+        kwarg_strs = ['{}={}'.format(k, v) for (k, v) in kwargs]
+        arg_str = ', '.join([str(a) for a in args] + kwarg_strs)
+        call_str = '{}({})'.format(function.func_name, arg_str)
+
+        # Perform the call itself, logging before, after, and anything thrown.
+        try:
+            if log_calls:
+                print('{}: Calling {}'.format(label, call_str))
+            res = function(*args, **kwargs)
+            if log_results:
+                print('{}: {} -> {}'.format(label, call_str, res))
+            return res
+        except Exception as ex:
+            if log_results:
+                print('{}: {} raised {}'.format(label, call_str, type(ex)))
+            raise ex
+
+    return wrapper
+
+
+def trace_object(obj, log_calls, log_results, label=''):
+    for name, member in inspect.getmembers(obj):
+        if inspect.ismethod(member):
+            # Skip meta-functions, decorate everything else
+            if not member.func_name.startswith('__'):
+                setattr(obj, name, trace_function(member, log_calls,
+                                                  log_results, label))
+    return obj