Your IP : 216.73.216.213
�
f f, � @ s, d d l m Z m Z m Z m Z e d � 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 d � Z e d k r(e � n d S)� )�verbose� run_unittest�
import_module�
reap_children�fcntlZtermiosNs 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/python34/lib64/python3.4/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
�setUp5 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 ZsigZframer 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
| t
j � } t
j
| t
j | t j
B� y& t j | d � } | j d | � Wn= t k
r6} z | j t j k r$� n WYd d } ~ Xn Xt
j
| t
j | � t 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 ttyi � 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�isattyr ZF_GETFLZF_SETFL�
O_NONBLOCK�read� assertEqual�errnoZEAGAIN�write�
TEST_STRING_1r �
TEST_STRING_2�close)r � master_fdZ
slave_nameZslave_fdZ
orig_flags�s1�e�s2r r r
�
test_basicC s@
zPtyTest.test_basicc
C s t d � t j � \ } } | t j k r� t j d � sT t d � t j d � n 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 � n t d | � xb y t j
| d
� } Wn t k
r'PYn X| s2Pn t j j
t | j d d � d d �� q� Wt j | d � \ } } | d ?} t d | | | f � | d k r�| j d � nT | d
k r�| j d � n8 | d k r�| j d � n | d k r| j d � n 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_forkr sV
zPtyTest.test_forkN)�__name__�
__module__� __qualname__r r r r. rA r r r r
r 4 s
/r 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 sf | 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 QXt 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 QXd 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 z t t t � Wd t � Xd S)N)r rE r r )r r r r
� test_main' s rm �__main__)Z test.supportr r r r r r% r r r; rJ r rU r r' r( r r ZTestCaser rE rm rB r r r r
�<module> s* "
�_