test: add more debugging message

Author: shuoweil

tests/unit/session/test_read_gbq_colab.py

@@ -29,6 +29,19 @@ def test_read_gbq_colab_includes_label():
     """Make sure we can tell direct colab usage apart from regular read_gbq usage."""
     import bigframes.core.log_adapter as log_adapter
 
+    # Track when get_and_reset_api_methods is called
+    original_get_and_reset = log_adapter.get_and_reset_api_methods
+    reset_calls = []
+
+    def debug_get_and_reset(dry_run=False):
+        methods = original_get_and_reset(dry_run)
+        if methods:
+            reset_calls.append(methods)
+            print(f"get_and_reset_api_methods called with: {methods}")
+        return methods
+
+    log_adapter.get_and_reset_api_methods = debug_get_and_reset
+
     # Monkey patch to track if decorator is called
     original_add = log_adapter.add_api_method
     called_methods = []
@@ -48,11 +61,12 @@ def debug_add_api_method(name):
 
     # Ensure call stack is empty before calling the method
     log_adapter._call_stack.clear()
-    print(f"Call stack before call: {log_adapter._call_stack}")
 
     _ = session._read_gbq_colab("SELECT 'read-gbq-colab-test'")
-    print(f"Call stack after call: {log_adapter._call_stack}")
-    print(f"Methods tracked by decorator: {called_methods}")
+
+    print(f"Total reset calls: {len(reset_calls)}")
+    for i, methods in enumerate(reset_calls):
+        print(f"Reset call {i}: {methods}")
 
     configs = session._job_configs  # type: ignore