mirror of https://github.com/python/cpython
102 lines
3.9 KiB
Python
102 lines
3.9 KiB
Python
"""
|
|
Very minimal unittests for parts of the readline module.
|
|
"""
|
|
import os
|
|
import tempfile
|
|
import unittest
|
|
from test.support import import_module, unlink
|
|
from test.support.script_helper import assert_python_ok
|
|
|
|
# Skip tests if there is no readline module
|
|
readline = import_module('readline')
|
|
|
|
class TestHistoryManipulation (unittest.TestCase):
|
|
"""
|
|
These tests were added to check that the libedit emulation on OSX and the
|
|
"real" readline have the same interface for history manipulation. That's
|
|
why the tests cover only a small subset of the interface.
|
|
"""
|
|
|
|
@unittest.skipUnless(hasattr(readline, "clear_history"),
|
|
"The history update test cannot be run because the "
|
|
"clear_history method is not available.")
|
|
def testHistoryUpdates(self):
|
|
readline.clear_history()
|
|
|
|
readline.add_history("first line")
|
|
readline.add_history("second line")
|
|
|
|
self.assertEqual(readline.get_history_item(0), None)
|
|
self.assertEqual(readline.get_history_item(1), "first line")
|
|
self.assertEqual(readline.get_history_item(2), "second line")
|
|
|
|
readline.replace_history_item(0, "replaced line")
|
|
self.assertEqual(readline.get_history_item(0), None)
|
|
self.assertEqual(readline.get_history_item(1), "replaced line")
|
|
self.assertEqual(readline.get_history_item(2), "second line")
|
|
|
|
self.assertEqual(readline.get_current_history_length(), 2)
|
|
|
|
readline.remove_history_item(0)
|
|
self.assertEqual(readline.get_history_item(0), None)
|
|
self.assertEqual(readline.get_history_item(1), "second line")
|
|
|
|
self.assertEqual(readline.get_current_history_length(), 1)
|
|
|
|
@unittest.skipUnless(hasattr(readline, "append_history_file"),
|
|
"append_history not available")
|
|
def test_write_read_append(self):
|
|
hfile = tempfile.NamedTemporaryFile(delete=False)
|
|
hfile.close()
|
|
hfilename = hfile.name
|
|
self.addCleanup(unlink, hfilename)
|
|
|
|
# test write-clear-read == nop
|
|
readline.clear_history()
|
|
readline.add_history("first line")
|
|
readline.add_history("second line")
|
|
readline.write_history_file(hfilename)
|
|
|
|
readline.clear_history()
|
|
self.assertEqual(readline.get_current_history_length(), 0)
|
|
|
|
readline.read_history_file(hfilename)
|
|
self.assertEqual(readline.get_current_history_length(), 2)
|
|
self.assertEqual(readline.get_history_item(1), "first line")
|
|
self.assertEqual(readline.get_history_item(2), "second line")
|
|
|
|
# test append
|
|
readline.append_history_file(1, hfilename)
|
|
readline.clear_history()
|
|
readline.read_history_file(hfilename)
|
|
self.assertEqual(readline.get_current_history_length(), 3)
|
|
self.assertEqual(readline.get_history_item(1), "first line")
|
|
self.assertEqual(readline.get_history_item(2), "second line")
|
|
self.assertEqual(readline.get_history_item(3), "second line")
|
|
|
|
# test 'no such file' behaviour
|
|
os.unlink(hfilename)
|
|
with self.assertRaises(FileNotFoundError):
|
|
readline.append_history_file(1, hfilename)
|
|
|
|
# write_history_file can create the target
|
|
readline.write_history_file(hfilename)
|
|
|
|
|
|
class TestReadline(unittest.TestCase):
|
|
|
|
@unittest.skipIf(readline._READLINE_VERSION < 0x0600
|
|
and "libedit" not in readline.__doc__,
|
|
"not supported in this library version")
|
|
def test_init(self):
|
|
# Issue #19884: Ensure that the ANSI sequence "\033[1034h" is not
|
|
# written into stdout when the readline module is imported and stdout
|
|
# is redirected to a pipe.
|
|
rc, stdout, stderr = assert_python_ok('-c', 'import readline',
|
|
TERM='xterm-256color')
|
|
self.assertEqual(stdout, b'')
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|