Your IP : 216.73.216.85
��Yf�+ � @ s d d l m Z m Z m Z e d � d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z
d d l Z d Z d Z
e r� d d � Z n d d � Z d d
� Z Gd d � d e j � Z Gd
d � d e j � Z d d � Z e d k re j � d S)� )�verbose�
import_module�
reap_childrenZtermiosNs I wish to buy a fish license.
s For my pet fish, Eric.
c C s t | � d S)N)�print)�msg� r �2/opt/alt/python35/lib64/python3.5/test/test_pty.py�debug s r c C s d S)Nr )r r r r r s c C sB | j d � r | j d d � S| j d � r> | j d d � S| S)Ns
s
s
)�endswith�replace)�datar r r �normalize_output s
r
c @ sL e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d S) �PtyTestc C s, t j t j | j � | _ t j d � d S)N�
)�signal�SIGALRM�
handle_sig� old_alarm�alarm)�selfr r r �setUp4 s z
PtyTest.setUpc C s' t j d � t j t j | j � d S)Nr )r r r r )r r r r �tearDown: s
zPtyTest.tearDownc C s | j d � d S)Nz isatty hung)�fail)r Zsig�framer r r r ? s zPtyTest.handle_sigc ! C s� yb t d � t j � \ } } t d | | f � t d | f � t j | � } t d | � Wn! t k
r� t j d � � Yn X| j t j | � d � t j
| � } zw t j | d � y& t j | d � } | j
d | � Wn: t k
r#} z | j t j k r� WYd d } ~ Xn XWd t j | | � Xt d
� t j | t � t j | d � } | j
d t | � � t d � t j | t d d
� � t j | t d
d � � t j | d � } | j
d t | � � t j | � t j | � d S)NzCalling master_open()z#Got master_fd '%d', slave_name '%s'zCalling slave_open(%r)zGot slave_fd '%d'z,Pseudo-terminals (seemingly) not functional.zslave_fd is not a ttyFi � zWriting to slave_fds I wish to buy a fish license.
zWriting chunked output� s For my pet fish, Eric.
)r �ptyZ master_openZ
slave_open�OSError�unittestZSkipTestZ
assertTrue�os�isatty� get_blocking� set_blocking�read� assertEqual�errnoZEAGAIN�write�
TEST_STRING_1r
�
TEST_STRING_2�close)r � master_fdZ
slave_nameZslave_fdZblocking�s1�e�s2r r r �
test_basicB sB
zPtyTest.test_basicc
C s t d � t j � \ } } | t j k r� t j d � sQ t d � t j d � t d � y t j � WnU t k
r� t d � YnQ t k
r� t d � Yn6 t d � t j d � Yn Xt d � t j d
� t j d � nt d | � x_ y t j
| d
� } Wn t k
r$PYn X| s,Pt j j
t | j d d � d d �� q� Wt j | d � \ } } | d ?} t d | | | f � | d k r�| j d � nQ | d
k r�| j d � n5 | d k r�| j d � n | d k r�| j d � t j | � d S)Nzcalling pty.fork()� zChild's fd 1 is not a tty?!� z In child, calling os.setsid()zGood: OSError was raised.zNo setsid() available?z An unexpected error was raised.z os.setsid() succeeded! (bad!)� � z!Waiting for child (%d) to finish.�P s
s
�encoding�asciir � z&Child (%d) exited with status %d (%d).z3Child raised an unexpected exception in os.setsid()z1pty.fork() failed to make child a session leader.z8Child spawned by pty.fork() did not have a tty as stdoutz&pty.fork() failed for unknown reasons.)r r �forkZCHILDr r �_exit�setsidr �AttributeErrorr# �sys�stdoutr& �strr �waitpidr r) )r �pidr* r Zstatus�resr r r � test_forks sV
zPtyTest.test_forkN)�__name__�
__module__� __qualname__r r r r. rA r r r r r 3 s
1r c @ sj e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
d S)�
SmallPtyTestsz)These tests don't spawn children or hang.c C sL t j | _ t j | _ t j | _ g | _ g | _ g | _ g | _
d S)N) r � STDIN_FILENO�orig_stdin_fileno�
STDOUT_FILENO�orig_stdout_fileno�select�orig_pty_select�fds�files�select_rfds_lengths�select_rfds_results)r r r r r � s zSmallPtyTests.setUpc C s� | j t _ | j t _ | j t _ x4 | j D]) } y | j � Wq. t k
rV Yq. Xq. Wx7 | j
D], } y t j | � Wqe t k
r� Yqe Xqe Wd S)N) rG r rF rI rH rK rJ rM r) r rL r )r �file�fdr r r r � s
zSmallPtyTests.tearDownc C s t j � } | j j | � | S)N)r �piperL �extend)r Zpipe_fdsr r r �_pipe� s zSmallPtyTests._pipec C s t j � } | j j | � | S)N)�socket�
socketpairrM rS )r rV r r r � _socketpair� s zSmallPtyTests._socketpairc C s; | j | j j d � t | � � | j j d � g g f S)Nr )r$ rN �pop�lenrO )r �rfdsZwfdsZxfdsr r r � _mock_select� s "zSmallPtyTests._mock_selectc C sg | j � \ } } | t _ | j � \ } } | t _ | j � } d d � | D� } t j | d d � t j | d � | j t _ | j j
d � | j j
| | d g � | j j
d � | j t
� � t j | d � Wd QRXt j | | d g g g d � d } | j | | d g | � | j t j | d � d � | j t j | d d � d � d S)
z6Test the normal data case on both master_fd and stdin.c S s g | ] } | j � � q Sr )�fileno)�.0�sr r r �
<listcomp>� s z4SmallPtyTests.test__copy_to_each.<locals>.<listcomp>r/ s from masters
from stdinr1 r N� )rT r rH rF rW r r& r[ rJ rN �appendrO � assertRaises�
IndexError�_copyr$ r# )r �read_from_stdout_fd�mock_stdout_fd�
mock_stdin_fd�write_to_stdin_fdrV �mastersrZ r r r �test__copy_to_each� s$ & z SmallPtyTests.test__copy_to_eachc
C s� | j � \ } } | t _ | j � \ } } | t _ | j � } d d � | D� } t j | d � | d j � t j | � | j t _ | j j
d � | j j
| | d g � | j j
d � | j t
� � t j | d � Wd QRXd S)z9Test the empty read EOF case on both master_fd and stdin.c S s g | ] } | j � � q Sr )r\ )r] r^ r r r r_ s z7SmallPtyTests.test__copy_eof_on_all.<locals>.<listcomp>r/ r1 r N)rT r rH rF rW r r) r[ rJ rN ra rO rb rc rd )r re rf rg rh rV ri r r r �test__copy_eof_on_all s
z#SmallPtyTests.test__copy_eof_on_allN) rB rC rD �__doc__r r rT rW r[ rj rk r r r r rE � s rE c C s t � d S)N)r r r r r �tearDownModule( s rm �__main__)Z test.supportr r r r% r r r; rJ r rU r r' r( r r
ZTestCaser rE rm rB �mainr r r r �<module> s(
�_