Tighten extension key-display import boundary assertions

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

Author: cursoragent

tests/test_extension_key_display_import_boundary.py

@@ -21,6 +21,23 @@ def _imports_format_string_key_for_error(module_text: str) -> bool:
     return False
 
 
+def _imports_safe_key_display_for_error(module_text: str) -> bool:
+    module_ast = ast.parse(module_text)
+    for node in module_ast.body:
+        if not isinstance(node, ast.ImportFrom):
+            continue
+        if node.module != "hyperbrowser.mapping_utils":
+            continue
+        if any(alias.name == "safe_key_display_for_error" for alias in node.names):
+            return True
+    return False
+
+
 def test_extension_key_display_helper_is_imported_from_display_utils():
     module_text = EXTENSION_UTILS_MODULE.read_text(encoding="utf-8")
     assert _imports_format_string_key_for_error(module_text)
+
+
+def test_extension_safe_key_display_helper_is_imported_from_mapping_utils():
+    module_text = EXTENSION_UTILS_MODULE.read_text(encoding="utf-8")
+    assert _imports_safe_key_display_for_error(module_text)