github/datasette
1
0
Fork 0
mirror of https://github.com/simonw/datasette.git synced 2025-12-10 16:51:24 +01:00
Code Issues Projects Releases Packages Wiki Activity

Fix to support replacing a database, closes #2465

Browse source
This commit is contained in:
Simon Willison 2025-02-06 10:32:47 -08:00
commit f95ac19e71
2 changed files with 36 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

31
tests/test_internals_database.py
View file

@ -721,3 +721,34 @@ async def test_hidden_tables(app_client):
"r_parent",
"r_rowid",
]
@pytest.mark.asyncio
async def test_replace_database(tmpdir):
path1 = str(tmpdir / "data1.db")
(tmpdir / "two").mkdir()
path2 = str(tmpdir / "two" / "data1.db")
sqlite3.connect(path1).executescript(
"""
create table t (id integer primary key);
insert into t (id) values (1);
insert into t (id) values (2);
"""
)
sqlite3.connect(path2).executescript(
"""
create table t (id integer primary key);
insert into t (id) values (1);
"""
)
datasette = Datasette([path1])
db = datasette.get_database("data1")
count = (await db.execute("select count(*) from t")).first()[0]
assert count == 2
# Now replace that database
datasette.get_database("data1").close()
datasette.remove_database("data1")
datasette.add_database(Database(datasette, path2), "data1")
db2 = datasette.get_database("data1")
count = (await db2.execute("select count(*) from t")).first()[0]
assert count == 1