Added db.index_foreign_keys() method plus test

This commit is contained in:
Simon Willison 2019-06-30 16:38:14 -07:00
commit 61164d98cf
3 changed files with 21 additions and 7 deletions

View file

@ -361,6 +361,16 @@ def test_add_foreign_key_guess_table(fresh_db):
)
def test_index_foreign_keys(fresh_db):
test_add_foreign_key_guess_table(fresh_db)
assert [] == fresh_db["dogs"].indexes
fresh_db.index_foreign_keys()
assert [["breed_id"]] == [i.columns for i in fresh_db["dogs"].indexes]
# Calling it a second time should do nothing
fresh_db.index_foreign_keys()
assert [["breed_id"]] == [i.columns for i in fresh_db["dogs"].indexes]
@pytest.mark.parametrize(
"extra_data,expected_new_columns",
[