chore: update snapshots to new format

pyproject.toml

@@ -79,7 +79,7 @@ boto3-stubs = "^1.34.23"
 [tool.poetry.group.test.dependencies]
 pytest = ">=7.3.1,<9.0.0"
 pytest-asyncio = "^0.21.0"
-pytest-textual-snapshot = { git = "https://github.com/tconbeer/pytest-textual-snapshot.git", branch = "main" }
+pytest-textual-snapshot = { git = "https://github.com/tconbeer/pytest-textual-snapshot.git" }
 # extension tests require consistent version of duckdb.
 duckdb = "1.0.0"
 

src/harlequin/app.py

@@ -471,6 +471,7 @@ def update_internal_editor_state(
             self.editor = message.active_editor
         else:
             self.editor = self.editor_collection.current_editor
+        self.editor.focus()
 
     def on_text_area_selection_changed(self) -> None:
         if self._validate_selection():

tests/functional_tests/conftest.py

@@ -6,6 +6,7 @@
 from textual.worker import WorkerCancelled
 
 from harlequin.app import Harlequin
+from harlequin.autocomplete import HarlequinCompletion
 
 
 @pytest.fixture(autouse=True)
@@ -35,6 +36,117 @@ def mock_config_loader(monkeypatch: pytest.MonkeyPatch) -> None:
     )
 
 
+@pytest.fixture(autouse=True)
+def mock_completions(monkeypatch: pytest.MonkeyPatch) -> None:
+    KEYWORDS = [
+        "abort",
+        "all",
+        "alter",
+        "always",
+        "analyze",
+        "and",
+        "as",
+        "asc",
+        "begin",
+        "between",
+        "by",
+        "cascade",
+        "case",
+        "column",
+        "commit",
+        "create",
+        "database",
+        "delete",
+        "desc",
+        "distinct",
+        "drop",
+        "else",
+        "end",
+        "explain",
+        "from",
+        "group",
+        "groups",
+        "having",
+        "in",
+        "inner",
+        "insert",
+        "intersect",
+        "into",
+        "is",
+        "join",
+        "left",
+        "like",
+        "limit",
+        "null",
+        "on",
+        "order",
+        "outer",
+        "over",
+        "partition",
+        "row",
+        "savepoint",
+        "select",
+        "set",
+        "table",
+        "temp",
+        "temporary",
+        "then",
+        "union",
+        "update",
+        "using",
+        "values",
+        "view",
+        "when",
+        "where",
+        "window",
+    ]
+
+    FUNCTIONS = [
+        ("array_select", "fn"),
+        ("count", "agg"),
+        ("greatest", "fn"),
+        ("least", "fn"),
+        ("list_select", "fn"),
+        ("sqrt", "fn"),
+        ("sum", "agg"),
+    ]
+
+    keyword_completions = [
+        HarlequinCompletion(
+            label=kw_name, type_label="kw", value=kw_name, priority=100, context=None
+        )
+        for kw_name in KEYWORDS
+    ]
+
+    function_completions = [
+        HarlequinCompletion(
+            label=label, type_label=type_label, value=label, priority=1000, context=None
+        )
+        for label, type_label in FUNCTIONS
+    ]
+
+    completions = [*keyword_completions, *function_completions]
+    duckdb_completions = [
+        (
+            completion.label,
+            completion.type_label,
+            completion.priority,
+            completion.context,
+        )
+        for completion in completions
+    ]
+    monkeypatch.setattr(
+        "harlequin_sqlite.adapter.get_completion_data",
+        lambda *_: completions,
+        raising=True,
+    )
+    monkeypatch.setattr(
+        "harlequin_duckdb.adapter.get_completion_data",
+        lambda *_: duckdb_completions,
+        raising=True,
+    )
+
+
 @pytest.fixture
 def mock_pyperclip(monkeypatch: pytest.MonkeyPatch) -> MagicMock:
     mock = MagicMock()

tests/functional_tests/test_data_catalog.py

@@ -51,7 +51,6 @@ async def test_data_catalog(
             await pilot.pause()
         catalog = app.data_catalog
         assert not catalog.database_tree.show_root
-        snap_results.append(await app_snapshot(app, "Initialization"))
 
         # this test app has two databases attached.
         dbs = catalog.database_tree.root.children
@@ -71,6 +70,7 @@ async def test_data_catalog(
             await pilot.pause(0.1)
         assert len(dbs[0].children) == 2
         assert all(not node.is_expanded for node in dbs[0].children)
+        snap_results.append(await app_snapshot(app, "Initialization"))
 
         assert str(dbs[1].label) == "tiny db"
         assert dbs[0].is_expanded is False

tests/functional_tests/test_query_editor.py

@@ -5,8 +5,10 @@
 
 import pytest
 from textual.widgets.text_area import Selection
+from textual.widgets.tree import TreeNode
 
 from harlequin import Harlequin
+from harlequin.catalog import CatalogItem
 
 
 def transaction_button_visible(app: Harlequin) -> bool:
@@ -114,10 +116,6 @@ async def test_multiple_buffers(
         assert all(snap_results)
 
 
-@pytest.mark.xfail(
-    sys.platform in ("win32", "darwin"),
-    reason="Scroll bar is a different size.",
-)
 @pytest.mark.asyncio
 async def test_word_autocomplete(
     app_all_adapters: Harlequin,
@@ -131,6 +129,15 @@ async def test_word_autocomplete(
         while app.editor is None or app.editor_collection.word_completer is None:
             await pilot.pause()
 
+        # we need to let the data catalog load the root's children
+        while (
+            app.data_catalog.database_tree.loading
+            or not app.data_catalog.database_tree.root.children
+        ):
+            await pilot.pause()
+
+        app.editor.focus()
+
         await pilot.press("s")
         await pilot.pause()
         await wait_for_workers(app)
@@ -189,27 +196,32 @@ async def test_member_autocomplete(
     async with app.run_test() as pilot:
         await wait_for_workers(app)
 
+        async def _expand_and_wait(node: TreeNode[CatalogItem]) -> None:
+            node.expand()
+            while not node.children:
+                if getattr(node.data, "loaded", True):
+                    break
+                await pilot.pause()
+
         # we need to expand the data catalog to load items into the completer
         while (
             app.data_catalog.database_tree.loading
             or not app.data_catalog.database_tree.root.children
         ):
             await pilot.pause()
         for db_node in app.data_catalog.database_tree.root.children:
-            db_node.expand()
-            while not db_node.children:
-                if getattr(db_node.data, "loaded", True):
-                    break
-                await pilot.pause()
+            await _expand_and_wait(db_node)
             for schema_node in db_node.children:
-                schema_node.expand()
+                await _expand_and_wait(schema_node)
         await pilot.pause(1)
 
         # now the completer should be populated
         while app.editor is None or app.editor_collection.member_completer is None:
             await pilot.pause()
+
         app.editor.text = '"drivers"'
         app.editor.selection = Selection((0, 9), (0, 9))
+        app.editor.focus()
 
         await pilot.press("full_stop")
         await pilot.pause()