mirror of
				https://github.com/python/cpython.git
				synced 2025-10-31 13:41:24 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			101 lines
		
	
	
	
		
			3.9 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			101 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()
 | 
