Cap extension missing-key diagnostics length

Co-authored-by: Shri Sukhani <shrisukhani@users.noreply.github.com>

Author: cursoragent

hyperbrowser/client/managers/extension_utils.py

@@ -4,6 +4,8 @@
 from hyperbrowser.exceptions import HyperbrowserError
 from hyperbrowser.models.extension import ExtensionResponse
 
+_MAX_DISPLAYED_MISSING_KEYS = 20
+
 
 def _get_type_name(value: Any) -> str:
     return type(value).__name__
@@ -16,22 +18,27 @@ def _safe_stringify_key(value: object) -> str:
         return f"<unprintable {_get_type_name(value)}>"
 
 
+def _summarize_mapping_keys(mapping: Mapping[object, object]) -> str:
+    key_names = sorted(_safe_stringify_key(key) for key in mapping.keys())
+    if not key_names:
+        return "[]"
+    displayed_keys = key_names[:_MAX_DISPLAYED_MISSING_KEYS]
+    key_summary = ", ".join(displayed_keys)
+    remaining_key_count = len(key_names) - len(displayed_keys)
+    if remaining_key_count > 0:
+        key_summary = f"{key_summary}, ... (+{remaining_key_count} more)"
+    return f"[{key_summary}]"
+
+
 def parse_extension_list_response_data(response_data: Any) -> List[ExtensionResponse]:
     if not isinstance(response_data, Mapping):
         raise HyperbrowserError(
             f"Expected mapping response but got {_get_type_name(response_data)}"
         )
     if "extensions" not in response_data:
-        available_keys = ", ".join(
-            sorted(_safe_stringify_key(key) for key in response_data.keys())
-        )
-        if available_keys:
-            available_keys = f"[{available_keys}]"
-        else:
-            available_keys = "[]"
         raise HyperbrowserError(
             "Expected 'extensions' key in response but got "
-            f"{available_keys} keys"
+            f"{_summarize_mapping_keys(response_data)} keys"
         )
     if not isinstance(response_data["extensions"], list):
         raise HyperbrowserError(

tests/test_extension_utils.py

@@ -117,6 +117,21 @@ def test_parse_extension_list_response_data_missing_key_lists_available_keys():
         )
 
 
+def test_parse_extension_list_response_data_missing_key_limits_key_list_size():
+    payload = {f"key-{index:02d}": index for index in range(25)}
+
+    with pytest.raises(
+        HyperbrowserError,
+        match=(
+            "Expected 'extensions' key in response but got "
+            r"\[key-00, key-01, key-02, key-03, key-04, key-05, key-06, key-07,"
+            r" key-08, key-09, key-10, key-11, key-12, key-13, key-14, key-15,"
+            r" key-16, key-17, key-18, key-19, \.\.\. \(\+5 more\)\] keys"
+        ),
+    ):
+        parse_extension_list_response_data(payload)
+
+
 def test_parse_extension_list_response_data_missing_key_handles_unprintable_keys():
     class _BrokenStringKey:
         def __str__(self) -> str: