add test

Author: edublancas

src/sql/magic.py

@@ -705,7 +705,7 @@ def get_query_type(command: str):
 
 def set_configs(ip, file_path, alternate_path):
     """Set user defined SqlMagic configuration settings"""
-    sql = ip.find_cell_magic("sql").__self__
+    sql = ip.find_cell_magic("jupysql").__self__
     user_configs, loaded_from = util.get_user_configs(file_path, alternate_path)
     default_configs = util.get_default_configs(sql)
     table_rows = []

src/tests/test_magic.py

@@ -23,6 +23,8 @@
 from sql.run.resultset import ResultSet
 from sql import magic
 from sql.warnings import JupySQLQuotedNamedParametersWarning
+from sql._testing import TestingShell
+from sql.magic import load_ipython_extension
 
 
 from conftest import runsql
@@ -2767,3 +2769,21 @@ def test_disabled_named_parameters_shows_disabled_warning(ip):
     )
 
     assert expected_warning in str(excinfo.value)
+
+
+def test_databricks_sql_magic_disabled(monkeypatch):
+    monkeypatch.setenv("DATABRICKS_RUNTIME_VERSION", "10.0.0")
+    ip = TestingShell.preconfigured_shell()
+    load_ipython_extension(ip)
+
+    ip.run_cell("%jupysql duckdb://")
+    ip.run_cell("%jupysql select 1")
+
+    with pytest.raises(UsageError) as excinfo_line:
+        ip.run_cell("%sql select 1")
+
+    with pytest.raises(UsageError) as excinfo_cell:
+        ip.run_cell("%%sql\nselect 1")
+
+    assert "Line magic function `%sql` not found" in str(excinfo_line.value)
+    assert "Cell magic `%%sql` not found" in str(excinfo_cell.value)