mirror of
https://github.com/simonw/sqlite-utils.git
synced 2026-09-12 19:44:13 +02:00
792157d120
- Add type: ignore for monkey-patching Database.__init__ in conftest - Fix CLI test to pass string "2" instead of integer to Click invoke - Add type: ignore for optional sqlean import - Fix add_geometry_column test to use "XY" instead of integer 2 - Add type: ignore for click.Context as context manager - Add type: ignore for enable_fts test that intentionally omits argument - Add type: ignore for sys._called_from_test dynamic attribute - Fix rows_from_file test type error for intentional wrong argument - Handle None from pm.get_hookcallers in plugins.py 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
from sqlite_utils.utils import rows_from_file, Format, RowError
|
|
from io import BytesIO, StringIO
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"input,expected_format",
|
|
(
|
|
(b"id,name\n1,Cleo", Format.CSV),
|
|
(b"id\tname\n1\tCleo", Format.TSV),
|
|
(b'[{"id": "1", "name": "Cleo"}]', Format.JSON),
|
|
),
|
|
)
|
|
def test_rows_from_file_detect_format(input, expected_format):
|
|
rows, format = rows_from_file(BytesIO(input))
|
|
assert format == expected_format
|
|
rows_list = list(rows)
|
|
assert rows_list == [{"id": "1", "name": "Cleo"}]
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"ignore_extras,extras_key,expected",
|
|
(
|
|
(True, None, [{"id": "1", "name": "Cleo"}]),
|
|
(False, "_rest", [{"id": "1", "name": "Cleo", "_rest": ["oops"]}]),
|
|
# expected of None means expect an error:
|
|
(False, False, None),
|
|
),
|
|
)
|
|
def test_rows_from_file_extra_fields_strategies(ignore_extras, extras_key, expected):
|
|
try:
|
|
rows, format = rows_from_file(
|
|
BytesIO(b"id,name\r\n1,Cleo,oops"),
|
|
format=Format.CSV,
|
|
ignore_extras=ignore_extras,
|
|
extras_key=extras_key,
|
|
)
|
|
list_rows = list(rows)
|
|
except RowError:
|
|
if expected is None:
|
|
# This is fine,
|
|
return
|
|
else:
|
|
# We did not expect an error
|
|
raise
|
|
assert list_rows == expected
|
|
|
|
|
|
def test_rows_from_file_error_on_string_io():
|
|
with pytest.raises(TypeError) as ex:
|
|
rows_from_file(StringIO("id,name\r\n1,Cleo")) # type: ignore[arg-type]
|
|
assert ex.value.args == (
|
|
"rows_from_file() requires a file-like object that supports peek(), such as io.BytesIO",
|
|
)
|