mirror of
				https://github.com/python/cpython.git
				synced 2025-10-31 13:41:24 +00:00 
			
		
		
		
	 49fd7fa443
			
		
	
	
		49fd7fa443
		
	
	
	
	
		
			
			number of tests, all because of the codecs/_multibytecodecs issue described here (it's not a Py3K issue, just something Py3K discovers): http://mail.python.org/pipermail/python-dev/2006-April/064051.html Hye-Shik Chang promised to look for a fix, so no need to fix it here. The tests that are expected to break are: test_codecencodings_cn test_codecencodings_hk test_codecencodings_jp test_codecencodings_kr test_codecencodings_tw test_codecs test_multibytecodec This merge fixes an actual test failure (test_weakref) in this branch, though, so I believe merging is the right thing to do anyway.
		
			
				
	
	
		
			139 lines
		
	
	
	
		
			4.6 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			139 lines
		
	
	
	
		
			4.6 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| import pty, os, sys, signal
 | |
| from test.test_support import verbose, TestFailed, TestSkipped
 | |
| 
 | |
| TEST_STRING_1 = "I wish to buy a fish license.\n"
 | |
| TEST_STRING_2 = "For my pet fish, Eric.\n"
 | |
| 
 | |
| # Solaris (at least 2.9 and 2.10) seem to have a fickle isatty(). The first
 | |
| # test below, testing the result of os.openpty() for tty-ness, sometimes
 | |
| # (but not always) fails. The second isatty test, in the sub-process, always
 | |
| # works. Allow that fickle first test to fail on these platforms, since it
 | |
| # doesn't actually affect functionality.
 | |
| fickle_isatty = ["sunos5"]
 | |
| 
 | |
| if verbose:
 | |
|     def debug(msg):
 | |
|         print msg
 | |
| else:
 | |
|     def debug(msg):
 | |
|         pass
 | |
| 
 | |
| def normalize_output(data):
 | |
|     # Some operating systems do conversions on newline.  We could possibly
 | |
|     # fix that by doing the appropriate termios.tcsetattr()s.  I couldn't
 | |
|     # figure out the right combo on Tru64 and I don't have an IRIX box.
 | |
|     # So just normalize the output and doc the problem O/Ses by allowing
 | |
|     # certain combinations for some platforms, but avoid allowing other
 | |
|     # differences (like extra whitespace, trailing garbage, etc.)
 | |
| 
 | |
|     # This is about the best we can do without getting some feedback
 | |
|     # from someone more knowledgable.
 | |
| 
 | |
|     # OSF/1 (Tru64) apparently turns \n into \r\r\n.
 | |
|     if data.endswith('\r\r\n'):
 | |
|         return data[:-3] + '\n'
 | |
| 
 | |
|     # IRIX apparently turns \n into \r\n.
 | |
|     if data.endswith('\r\n'):
 | |
|         return data[:-2] + '\n'
 | |
| 
 | |
|     return data
 | |
| 
 | |
| # Marginal testing of pty suite. Cannot do extensive 'do or fail' testing
 | |
| # because pty code is not too portable.
 | |
| 
 | |
| def test_basic_pty():
 | |
|     try:
 | |
|         debug("Calling master_open()")
 | |
|         master_fd, slave_name = pty.master_open()
 | |
|         debug("Got master_fd '%d', slave_name '%s'"%(master_fd, slave_name))
 | |
|         debug("Calling slave_open(%r)"%(slave_name,))
 | |
|         slave_fd = pty.slave_open(slave_name)
 | |
|         debug("Got slave_fd '%d'"%slave_fd)
 | |
|     except OSError:
 | |
|         # " An optional feature could not be imported " ... ?
 | |
|         raise TestSkipped, "Pseudo-terminals (seemingly) not functional."
 | |
| 
 | |
|     if not os.isatty(slave_fd) and sys.platform not in fickle_isatty:
 | |
|         raise TestFailed, "slave_fd is not a tty"
 | |
| 
 | |
|     debug("Writing to slave_fd")
 | |
|     os.write(slave_fd, TEST_STRING_1)
 | |
|     s1 = os.read(master_fd, 1024)
 | |
|     sys.stdout.write(normalize_output(s1))
 | |
| 
 | |
|     debug("Writing chunked output")
 | |
|     os.write(slave_fd, TEST_STRING_2[:5])
 | |
|     os.write(slave_fd, TEST_STRING_2[5:])
 | |
|     s2 = os.read(master_fd, 1024)
 | |
|     sys.stdout.write(normalize_output(s2))
 | |
| 
 | |
|     os.close(slave_fd)
 | |
|     os.close(master_fd)
 | |
| 
 | |
| def handle_sig(sig, frame):
 | |
|     raise TestFailed, "isatty hung"
 | |
| 
 | |
| # isatty() and close() can hang on some platforms
 | |
| # set an alarm before running the test to make sure we don't hang forever
 | |
| old_alarm = signal.signal(signal.SIGALRM, handle_sig)
 | |
| signal.alarm(10)
 | |
| 
 | |
| try:
 | |
|     test_basic_pty()
 | |
| finally:
 | |
|     # remove alarm, restore old alarm handler
 | |
|     signal.alarm(0)
 | |
|     signal.signal(signal.SIGALRM, old_alarm)
 | |
| 
 | |
| # basic pty passed.
 | |
| 
 | |
| debug("calling pty.fork()")
 | |
| pid, master_fd = pty.fork()
 | |
| if pid == pty.CHILD:
 | |
|     # stdout should be connected to a tty.
 | |
|     if not os.isatty(1):
 | |
|         debug("Child's fd 1 is not a tty?!")
 | |
|         os._exit(3)
 | |
| 
 | |
|     # After pty.fork(), the child should already be a session leader.
 | |
|     # (on those systems that have that concept.)
 | |
|     debug("In child, calling os.setsid()")
 | |
|     try:
 | |
|         os.setsid()
 | |
|     except OSError:
 | |
|         # Good, we already were session leader
 | |
|         debug("Good: OSError was raised.")
 | |
|         pass
 | |
|     except AttributeError:
 | |
|         # Have pty, but not setsid() ?
 | |
|         debug("No setsid() available ?")
 | |
|         pass
 | |
|     except:
 | |
|         # We don't want this error to propagate, escaping the call to
 | |
|         # os._exit() and causing very peculiar behavior in the calling
 | |
|         # regrtest.py !
 | |
|         # Note: could add traceback printing here.
 | |
|         debug("An unexpected error was raised.")
 | |
|         os._exit(1)
 | |
|     else:
 | |
|         debug("os.setsid() succeeded! (bad!)")
 | |
|         os._exit(2)
 | |
|     os._exit(4)
 | |
| else:
 | |
|     debug("Waiting for child (%d) to finish."%pid)
 | |
|     (pid, status) = os.waitpid(pid, 0)
 | |
|     res = status >> 8
 | |
|     debug("Child (%d) exited with status %d (%d)."%(pid, res, status))
 | |
|     if res == 1:
 | |
|         raise TestFailed, "Child raised an unexpected exception in os.setsid()"
 | |
|     elif res == 2:
 | |
|         raise TestFailed, "pty.fork() failed to make child a session leader."
 | |
|     elif res == 3:
 | |
|         raise TestFailed, "Child spawned by pty.fork() did not have a tty as stdout"
 | |
|     elif res != 4:
 | |
|         raise TestFailed, "pty.fork() failed for unknown reasons."
 | |
| 
 | |
| os.close(master_fd)
 | |
| 
 | |
| # pty.fork() passed.
 |