remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-07-26 19:04:32 +02:00
Code Issues Projects Releases Packages Wiki Activity

WIP --extras-key --ignore-extras - does not work yet, refs #444

Browse source
This commit is contained in:
Simon Willison 2022-06-14 21:17:49 -07:00
commit e52925bbb9
3 changed files with 88 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

36
tests/test_cli_insert.py
View file

@ -222,6 +222,42 @@ def test_insert_csv_tsv(content, options, db_path, tmpdir):
assert [{"foo": "1", "bar": "2", "baz": "cat,dog"}] == list(db["data"].rows)
@pytest.mark.parametrize(
"options,expected_rows,expected_error",
[
(["--csv", "--ignore-extras"], [{"id": "1", "name": "Cleo"}], None),
(
["--csv", "--extras-key", "x"],
[{"id": "1", "name": "Cleo", "x": '["extra"]'}],
None,
),
(["--csv"], None, "Oh no an extra field found"),
(
["--csv", "--extras-key", "x", "--ignore-extras"],
None,
"Error: --ignore-extras and --extras-key cannot be used together\n",
),
],
)
def test_insert_csv_with_extra_fields(
options, expected_rows, expected_error, db_path, tmpdir
):
db = Database(db_path)
file_path = str(tmpdir / "insert.csv")
open(file_path, "w").write("id,name\n1,Cleo,extra")
result = CliRunner().invoke(
cli.cli,
["insert", db_path, "data", file_path] + options,
catch_exceptions=False,
)
if expected_error:
assert result.exit_code == 2
assert result.output == expected_error
else:
assert result.exit_code == 0
assert list(db["data"].rows) == expected_rows
@pytest.mark.parametrize(
"options",
(