fix: render binary result values safely

sqlit/shared/ui/widgets_tables.py

@@ -21,8 +21,12 @@
 
 
 def normalize_arrow_value(value: Any) -> Any:
-    """Keep Arrow from inferring its binary UUID extension type."""
-    return str(value) if isinstance(value, UUID) else value
+    """Convert values Arrow cannot safely render as text."""
+    if isinstance(value, UUID):
+        return str(value)
+    if isinstance(value, (bytes, bytearray, memoryview)):
+        return f"0x{bytes(value).hex()}"
+    return value
 
 
 def _stringify_uuid_rows(rows: Iterable[Iterable[Any]]) -> list[tuple[Any, ...]]:

tests/ui/test_uuid_fastdatatable.py

@@ -1,4 +1,4 @@
-"""Regression coverage for UUID result values with PyArrow 24+."""
+"""Regression coverage for non-text result values with PyArrow 24+."""
 
 from __future__ import annotations
 
@@ -20,6 +20,39 @@
 
 _INVALID_UTF8_UUID = UUID("a0000000-0000-0000-0000-000000000000")
 
+_INVALID_UTF8_BINARY = b"\x00\x00\x00\x00\x00\x00\x81\xff"
+_BINARY_TEXT = "0x00000000000081ff"
+
+
+@pytest.mark.parametrize(
+    "value",
+    [
+        _INVALID_UTF8_BINARY,
+        bytearray(_INVALID_UTF8_BINARY),
+        memoryview(_INVALID_UTF8_BINARY),
+    ],
+    ids=["bytes", "bytearray", "memoryview"],
+)
+def test_binary_column_is_stringified_before_arrow_measurement(value: object) -> None:
+    table = SqlitDataTable(
+        data={"binary": [value]},
+        column_labels=["binary"],
+    )
+
+    assert table.backend is not None
+    assert table.backend.column_content_widths == [len(_BINARY_TEXT)]
+    assert table.get_cell_at(Coordinate(0, 0)) == _BINARY_TEXT
+
+
+def test_incrementally_added_binary_is_stringified() -> None:
+    table = SqlitDataTable(data={"binary": ["initial"]}, column_labels=["binary"])
+
+    table.add_rows([(_INVALID_UTF8_BINARY,)])
+
+    assert table.backend is not None
+    assert table.backend.column_content_widths == [len(_BINARY_TEXT)]
+    assert table.get_cell_at(Coordinate(1, 0)) == _BINARY_TEXT
+
 
 def test_uuid_column_is_stringified_before_arrow_measurement() -> None:
     table = SqlitDataTable(
@@ -73,3 +106,36 @@ async def test_decimal_incremental_backend_stringifies_uuid_column() -> None:
         assert app.results_table.get_cell_at(Coordinate(0, 2)) == str(
             _INVALID_UTF8_UUID
         )
+
+
+@pytest.mark.asyncio
+async def test_decimal_incremental_backend_stringifies_binary_column() -> None:
+    connection = create_test_connection("test-db", "sqlite")
+    services = build_test_services(
+        connection_store=MockConnectionStore([connection]),
+        settings_store=MockSettingsStore({"theme": "tokyo-night"}),
+    )
+    app = SSMSTUI(services=services)
+    rows = [
+        (index, Decimal(f"{index}.25"), _INVALID_UTF8_BINARY)
+        for index in range(1, 202)
+    ]
+
+    async with app.run_test(size=(120, 40)) as pilot:
+        await pilot.pause()
+        await app._display_query_results(
+            columns=["id", "amount", "binary"],
+            rows=rows,
+            row_count=len(rows),
+            truncated=False,
+            elapsed_ms=0,
+        )
+        for _ in range(3):
+            await pilot.pause(0.05)
+
+        assert app.results_table.row_count == len(rows)
+        assert app.results_table.backend is not None
+        assert app.results_table.backend.column_content_widths[2] == len(
+            _BINARY_TEXT
+        )
+        assert app.results_table.get_cell_at(Coordinate(0, 2)) == _BINARY_TEXT