remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-09-08 01:24:20 +02:00
Code Issues Projects Releases Packages Wiki Activity

Add -o/--column-order option to insert and upsert

Browse source
This commit is contained in:
Chris (ChrisJr404) 2026-08-24 23:56:30 -04:00
commit 7dff7ceea5
No known key found for this signature in database
4 changed files with 50 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

21
tests/test_cli_insert.py
View file

@ -79,6 +79,27 @@ def test_insert_json_flatten_nl(tmpdir):
]
@pytest.mark.parametrize("command", ("insert", "upsert"))
@pytest.mark.parametrize(
"args,expected_columns",
(
([], ["id", "name", "age"]),
(["-o", "name"], ["name", "id", "age"]),
(["--column-order", "age", "-o", "name"], ["age", "name", "id"]),
),
)
def test_insert_column_order(db_path, tmpdir, command, args, expected_columns):
json_path = str(tmpdir / "dog.json")
with open(json_path, "w") as fp:
fp.write(json.dumps({"id": 1, "name": "Cleo", "age": 4}))
result = CliRunner().invoke(
cli.cli, [command, db_path, "dogs", json_path, "--pk", "id"] + args
)
assert result.exit_code == 0
columns = [c.name for c in Database(db_path).table("dogs").columns]
assert columns == expected_columns
@pytest.mark.parametrize(
"args,expected_pks",
(