add test coverage for packages/special/__init__.py

since it now contains substantive logic

Author: rolandwalker

changelog.md

@@ -1,3 +1,11 @@
+Upcoming (TBD)
+==============
+
+Internal
+--------
+* Add test coverage for `special/__init__.py`.
+
+
 1.74.0 (2026/06/06)
 ==============
 

test/pytests/test_special_init.py

@@ -0,0 +1,81 @@
+from __future__ import annotations
+
+import importlib
+import sys
+from types import ModuleType
+
+import pytest
+
+
+def reload_special(monkeypatch: pytest.MonkeyPatch, *, llm_off: bool) -> ModuleType:
+    if llm_off:
+        monkeypatch.setenv('MYCLI_LLM_OFF', '1')
+    else:
+        monkeypatch.delenv('MYCLI_LLM_OFF', raising=False)
+    sys.modules.pop('mycli.packages.special', None)
+    return importlib.import_module('mycli.packages.special')
+
+
+def test_special_init_exports_public_names(monkeypatch: pytest.MonkeyPatch) -> None:
+    special = reload_special(monkeypatch, llm_off=False)
+
+    for name in special.__all__:
+        assert hasattr(special, name)
+
+
+def test_special_init_reexports_special_command_api(monkeypatch: pytest.MonkeyPatch) -> None:
+    special = reload_special(monkeypatch, llm_off=False)
+    special_main = importlib.import_module('mycli.packages.special.main')
+
+    assert special.execute is special_main.execute
+    assert special.special_command is special_main.special_command
+    assert special.CommandNotFound is special_main.CommandNotFound
+
+
+def test_special_init_reexports_io_state_api(monkeypatch: pytest.MonkeyPatch) -> None:
+    special = reload_special(monkeypatch, llm_off=False)
+    iocommands = importlib.import_module('mycli.packages.special.iocommands')
+
+    assert special.set_pager_enabled is iocommands.set_pager_enabled
+    assert special.is_pager_enabled is iocommands.is_pager_enabled
+    assert special.write_tee is iocommands.write_tee
+
+
+def test_special_init_reexports_dbcommands(monkeypatch: pytest.MonkeyPatch) -> None:
+    special = reload_special(monkeypatch, llm_off=False)
+    dbcommands = importlib.import_module('mycli.packages.special.dbcommands')
+
+    assert special.list_databases is dbcommands.list_databases
+    assert special.list_tables is dbcommands.list_tables
+    assert special.status is dbcommands.status
+
+
+def test_special_init_uses_llm_implementation_when_enabled(monkeypatch: pytest.MonkeyPatch) -> None:
+    special = reload_special(monkeypatch, llm_off=False)
+    llm = importlib.import_module('mycli.packages.special.llm')
+
+    assert special.FinishIteration is llm.FinishIteration
+    assert special.is_llm_command is llm.is_llm_command
+    assert special.handle_llm is llm.handle_llm
+    assert special.sql_using_llm is llm.sql_using_llm
+
+
+def test_special_init_uses_llm_stubs_when_disabled(monkeypatch: pytest.MonkeyPatch) -> None:
+    special = reload_special(monkeypatch, llm_off=True)
+
+    assert special.is_llm_command(r'\llm prompt') is False
+    with pytest.raises(special.FinishIteration) as handle_exc:
+        special.handle_llm(cast_args := object())
+    with pytest.raises(special.FinishIteration) as sql_exc:
+        special.sql_using_llm(cast_args)
+
+    assert handle_exc.value.results is None
+    assert sql_exc.value.results is None
+
+
+def test_special_init_stub_finish_iteration_stores_results(monkeypatch: pytest.MonkeyPatch) -> None:
+    special = reload_special(monkeypatch, llm_off=True)
+
+    error = special.FinishIteration(results=['done'])
+
+    assert error.results == ['done']