Allow SQL functions in SQL write queries

Closes #2751
This commit is contained in:
Simon Willison 2026-05-28 10:22:16 -07:00
commit 51dab16149
6 changed files with 91 additions and 15 deletions

View file

@ -50,10 +50,19 @@ def test_decision_for_write_sql_operation_rejects_vacuum():
assert decision.message == "VACUUM is not allowed in user-supplied SQL"
def test_decision_for_write_sql_operation_reports_unsupported_functions():
def test_decision_for_write_sql_operation_ignores_functions():
decision = decision_for_write_sql_operation(
Operation("function", "function", None, None, None, target="upper")
)
assert isinstance(decision, IgnoreWriteSqlOperation)
assert decision.reason == "SQL function"
def test_decision_for_write_sql_operation_reports_unsupported_operations():
decision = decision_for_write_sql_operation(
Operation("unknown", "unknown", None, None, None)
)
assert isinstance(decision, UnsupportedWriteSqlOperation)
assert decision.message == "Unsupported SQL operation: function function"
assert decision.message == "Unsupported SQL operation: unknown unknown"