remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-07-31 22:44:11 +02:00
Code Issues Projects Releases Packages Wiki Activity

First attempt at .extract_expand() method, refs #239

Browse source
This commit is contained in:
Simon Willison 2021-02-25 08:36:03 -08:00
commit 8d641ab08a
2 changed files with 42 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

24
tests/test_extract.py
View file

@ -1,5 +1,6 @@
from sqlite_utils.db import Index, InvalidColumns
from sqlite_utils.db import Index, InvalidColumns, ForeignKey
import itertools
import json
import pytest
@ -171,3 +172,24 @@ def test_extract_error_on_incompatible_existing_lookup_table(fresh_db):
fresh_db["species2"].insert({"id": 1, "common_name": 3.5})
with pytest.raises(InvalidColumns):
fresh_db["tree"].extract("common_name", table="species2")
def test_extract_expand(fresh_db):
fresh_db["trees"].insert(
{"id": 1, "species": '{"id": 5, "name": "Tree 1", "common_name": "Palm"}'},
pk="id",
)
assert fresh_db.table_names() == ["trees"]
fresh_db["trees"].extract_expand(
"species", expand=json.loads, table="species", pk="id"
)
assert set(fresh_db.table_names()) == {"trees", "species"}
assert list(fresh_db["trees"].rows) == [{"id": 1, "species_id": 5}]
assert list(fresh_db["species"].rows) == [
{"id": 5, "name": "Tree 1", "common_name": "Palm"}
]
assert fresh_db["trees"].foreign_keys == [
ForeignKey(
table="trees", column="species_id", other_table="species", other_column="id"
)
]