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

Database(..., execute_plugins=False) mechanism, refs #575

Browse source
This commit is contained in:
Simon Willison 2023-07-22 16:06:11 -07:00
commit 374a816c72
3 changed files with 24 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

28
tests/test_plugins.py
View file

@ -49,10 +49,16 @@ def test_prepare_connection():
conn.create_function("hello", 1, lambda name: f"Hello, {name}!")
db = Database(memory=True)
functions = db.execute(
"select distinct name from pragma_function_list order by 1"
).fetchall()
assert "hello" not in functions
def _functions(db):
return [
row[0]
for row in db.execute(
"select distinct name from pragma_function_list order by 1"
).fetchall()
]
assert "hello" not in _functions(db)
try:
plugins.pm.register(HelloFunctionPlugin(), name="HelloFunctionPlugin")
@ -62,18 +68,14 @@ def test_prepare_connection():
]
db = Database(memory=True)
functions = [
row[0]
for row in db.execute(
"select distinct name from pragma_function_list order by 1"
).fetchall()
]
assert "hello" in functions
assert "hello" in _functions(db)
result = db.execute('select hello("world")').fetchone()[0]
assert result == "Hello, world!"
# Test execute_plugins=False
db2 = Database(memory=True, execute_plugins=False)
assert "hello" not in _functions(db2)
finally:
plugins.pm.unregister(name="HelloFunctionPlugin")
assert plugins.get_plugins() == []