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

sqlite-utils json dogs.db "select * from dogs"

Browse source
This commit is contained in:
Simon Willison 2019-01-25 18:06:29 -08:00
commit 5466c9745d
4 changed files with 158 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

36
tests/test_cli.py
View file

@ -151,3 +151,39 @@ def test_csv(db_path):
cli.cli, ["csv", db_path, "select id, name, age from dogs", "--no-headers"]
)
assert "1,Cleo,4\n2,Pancakes,2\n" == result.output
@pytest.mark.parametrize(
"args,expected",
[
(
[],
'[{"id": 1, "name": "Cleo", "age": 4},\n {"id": 2, "name": "Pancakes", "age": 2}]',
),
(
["--nl"],
'{"id": 1, "name": "Cleo", "age": 4}\n{"id": 2, "name": "Pancakes", "age": 2}',
),
(["--arrays"], '[[1, "Cleo", 4],\n [2, "Pancakes", 2]]'),
(["--arrays", "--nl"], '[1, "Cleo", 4]\n[2, "Pancakes", 2]'),
],
)
def test_json(db_path, args, expected):
db = Database(db_path)
with db.conn:
db["dogs"].insert_all(
[
{"id": 1, "age": 4, "name": "Cleo"},
{"id": 2, "age": 2, "name": "Pancakes"},
]
)
result = CliRunner().invoke(
cli.cli, ["csv", db_path, "select id, name, age from dogs"]
)
assert 0 == result.exit_code
assert "id,name,age\n1,Cleo,4\n2,Pancakes,2\n" == result.output
# Test the no-headers option:
result = CliRunner().invoke(
cli.cli, ["json", db_path, "select id, name, age from dogs"] + args
)
assert expected == result.output.strip()