Add test

tanloong · tanloong · commit 782a59931f8e · 2025-05-05T12:19:59.000+08:00
diff --git a/Lib/test/test_sqlite3/test_cli.py b/Lib/test/test_sqlite3/test_cli.py
@@ -1,11 +1,13 @@
 """sqlite3 CLI tests."""
+import re
 import sqlite3
 import unittest
 
 from sqlite3.__main__ import main as cli
+from test.support import captured_stdout, captured_stderr, captured_stdin, requires_subprocess
+from test.support.import_helper import import_module
 from test.support.os_helper import TESTFN, unlink
-from test.support import captured_stdout, captured_stderr, captured_stdin
-
+from test.support.pty_helper import run_pty
 
 class CommandLineInterface(unittest.TestCase):
 
@@ -153,5 +155,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()
diff --git a/Misc/NEWS.d/next/Library/2025-05-05-03-14-08.gh-issue-133390.AuTggn.rst b/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.

Original file line number	Diff line number	Diff line change
`@@ -1 +1 @@`
`1`		-Support keyword completion for :mod:`sqlite3` command-line interface.
	`1`	+Support keyword completion in the :mod:`sqlite3` command-line interface.