give toolbar tests the sqlexecute property

Now that the prompt string can have data fetched from the connection,
and now that the toolbar can have custom format strings, there are
assertions in get_prompt() that we should satisfy in toolbar tests, by
setting the sqlexecute property.

Author: rolandwalker

changelog.md

@@ -4,6 +4,7 @@ Upcoming (TBD)
 Internal
 ---------
 * Require `sqlglot` 30.x.
+* Connect toolbar tests to the test database.
 
 
 1.65.0 (2026/03/16)

test/test_clitoolbar.py

@@ -1,20 +1,60 @@
+# type: ignore
+
 from prompt_toolkit.shortcuts import PromptSession
 
 from mycli.clitoolbar import create_toolbar_tokens_func
 from mycli.main import MyCli
+from mycli.sqlexecute import SQLExecute
+from test.utils import HOST, PASSWORD, PORT, USER, dbtest
 
 
+@dbtest
 def test_create_toolbar_tokens_func_initial():
     m = MyCli()
+    m.sqlexecute = SQLExecute(
+        None,
+        USER,
+        PASSWORD,
+        HOST,
+        PORT,
+        None,
+        None,
+        None,
+        None,
+        None,
+        None,
+        None,
+        None,
+        None,
+        None,
+    )
     m.prompt_app = PromptSession()
     iteration = 0
     f = create_toolbar_tokens_func(m, lambda: iteration == 0, m.toolbar_format)
     result = f()
     assert any("right-arrow accepts full-line suggestion" in token for token in result)
 
 
+@dbtest
 def test_create_toolbar_tokens_func_short():
     m = MyCli()
+    m.sqlexecute = SQLExecute(
+        None,
+        USER,
+        PASSWORD,
+        HOST,
+        PORT,
+        None,
+        None,
+        None,
+        None,
+        None,
+        None,
+        None,
+        None,
+        None,
+        None,
+    )
     m.prompt_app = PromptSession()
     iteration = 1
     f = create_toolbar_tokens_func(m, lambda: iteration == 0, m.toolbar_format)