From aeae28bda35b937d3212061c41e180fa72df17bb Mon Sep 17 00:00:00 2001
From: Harshit Jain <48647625+Harshit28j@users.noreply.github.com>
Date: Mon, 2 Mar 2026 17:15:34 +0530
Subject: [PATCH] Update tests/logging_callback_tests/test_gcs_pub_sub.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
---
 tests/logging_callback_tests/test_gcs_pub_sub.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/logging_callback_tests/test_gcs_pub_sub.py b/tests/logging_callback_tests/test_gcs_pub_sub.py
index 84211c449b7..3b940b4b739 100644
--- a/tests/logging_callback_tests/test_gcs_pub_sub.py
+++ b/tests/logging_callback_tests/test_gcs_pub_sub.py
@@ -82,6 +82,7 @@ def extract_pubsub_payload(mock_post: AsyncMock) -> dict:
     Handles base64 decoding of the Pub/Sub message data.
     """
     actual_request = mock_post.call_args[1]["json"]
+    assert "messages" in actual_request, f"Expected 'messages' key in Pub/Sub request, got keys: {list(actual_request.keys())}"
     encoded_message = actual_request["messages"][0]["data"]
     decoded_message = base64.b64decode(encoded_message).decode("utf-8")
     return json.loads(decoded_message)