Add test

Author: tanloong

Lib/test/test_sqlite3/test_cli.py

@@ -1,17 +1,20 @@
 """sqlite3 CLI tests."""
+import re
 import sqlite3
 import unittest
 
 from sqlite3.__main__ import main as cli
+from test.support.import_helper import import_module
 from test.support.os_helper import TESTFN, unlink
+from test.support.pty_helper import run_pty
 from test.support import (
     captured_stdout,
     captured_stderr,
     captured_stdin,
     force_not_colorized,
+    requires_subprocess,
 )
 
-
 class CommandLineInterface(unittest.TestCase):
 
     def _do_test(self, *args, expect_success=True):
@@ -159,5 +162,36 @@ def test_interact_on_disk_file(self):
         self.assertIn("(0,)\n", out)
 
 
+@requires_subprocess()
+class Completer(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        # Ensure that the readline module is loaded
+        # If this fails, the test is skipped because SkipTest will be raised
+        readline = import_module("readline")
+        if readline.backend == "editline":
+            raise unittest.SkipTest("libedit readline is not supported")
+
+    def test_keyword_completion(self):
+        script = "from sqlite3.__main__ import main; main()"
+        # List candidates starting with 'S', there should be multiple matches.
+        # Then add 'EL' and complete 'SEL' to 'SELECT'.  Quit console in the end
+        # to let run_pty() return.
+        input = b"S\t\tEL\t 1;\n.quit\n"
+        output = run_pty(script, input)
+        # Remove control sequences that colorize typed prefix 'S'
+        output = re.sub(rb"\x1b\[[0-9;]*[mK]", b"", output)
+        self.assertIn(b"SELECT", output)
+        self.assertIn(b"SET", output)
+        self.assertIn(b"SAVEPOINT", output)
+        self.assertIn(b"(1,)", output)
+
+        # Keywords are completed in upper case for even lower case user input
+        input = b"sel\t\t 1;\n.quit\n"
+        output = run_pty(script, input)
+        output = re.sub(rb"\x1b\[[0-9;]*[mK]", b"", output)
+        self.assertIn(b"SELECT", output)
+        self.assertIn(b"(1,)", output)
+
 if __name__ == "__main__":
     unittest.main()

Misc/NEWS.d/next/Library/2025-05-05-03-14-08.gh-issue-133390.AuTggn.rst

@@ -1 +1 @@
-Support keyword completion for :mod:`sqlite3` command-line interface.
+Support keyword completion in the :mod:`sqlite3` command-line interface.