Your IP : 216.73.217.32
3
� f�- � @ s� d dl mZmZmZ ed� d dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z dZ
dZerzdd� Zndd� Zd d
� Zd d � ZG d
d� de j�ZG dd� de j�Zdd� Zedkr�e j� dS )� )�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/python36/lib64/python3.6/test/test_pty.py�debug s r c C s d S )Nr )r r r r r s c C s0 | j d�r| jdd�S | j d�r,| jdd�S | S )Ns
�
s
)�endswith�replace)�datar r r �normalize_output% s
r c C s t j| ddd�}|j� S )z8Read one line. May block forever if no newline is read.�rbF)�mode�closefd)�io�FileIO�readline)�fd�readerr r r � _readline: s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�PtyTestc C s> t j t j| j�}| jt j t j|� | jt jd� t jd� d S )Nr �
)�signal�SIGALRM�
handle_sigZ
addCleanup�alarm)�selfZ old_alarmr r r �setUpE s z
PtyTest.setUpc C s | j d� d S )Nz isatty hung)�fail)r Zsig�framer r r r M s zPtyTest.handle_sigc ! C s� yLt d� tj� \}}t d||f � t d|f � tj|�}t d| � W n tk
rj tjd��Y nX | jtj |�d� tj
|�}z^tj |d� y tj |d�}| j
d |� W n0 tk
r� } z|jtjkr҂ W Y d d }~X nX W d tj ||� X t d
� tj|t� t|�}| j
d t|�� t d � tj|td d
� � tj|td
d � � t|�}| j
dt|�� tj|� tj|� 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 r �
TEST_STRING_2�close)r � master_fdZ
slave_nameZslave_fdZblocking�s1�e�s2r r r �
test_basicP sB
zPtyTest.test_basicc
C s� t d� tj� \}}|tjkr�tjd�s:t d� tjd� t d� y tj� W nR tk
rj t d� Y nJ t k
r� t d� Y n0 t d� tjd� Y nX t d � tjd
� tjd � n�t d | � xPytj
|d
�}W n tk
r� P Y nX |� s�P t j j
t|jdd�dd�� q�W tj|d�\}}|d? }t d|||f � |dk�r^| jd� n@|d
k�rt| jd� n*|dk�r�| jd� n|d k�r�| jd� tj|� 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
r
�ascii)�encodingr � 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 r1 )r �pidr2 r
Zstatus�resr r r � test_fork� sV
zPtyTest.test_forkN)�__name__�
__module__� __qualname__r r r6 rI r r r r r D s 1r c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�
SmallPtyTestsz)These tests don't spawn children or hang.c C s4 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_| jt_| jt_x0| jD ]&}y |j� W q t k
rD Y q X q W x2| j
D ](}yt j|� W qR t k
rx Y qRX qRW d S )N) rO r$ rN rQ rP rS rR rU r1 r% rT r' )r �filer r r r �tearDown� s
zSmallPtyTests.tearDownc C s t j� }| jj|� |S )N)r' �piperT �extend)r Zpipe_fdsr r r �_pipe� s zSmallPtyTests._pipec C s t j� }| jj|� |S )N)�socket�
socketpairrU r[ )r r^ r r r � _socketpair� s zSmallPtyTests._socketpairc C s* | j | jjd�t|�� | jjd�g g fS )Nr )r, rV �pop�lenrW )r �rfdsZwfdsZxfdsr r r � _mock_select� s zSmallPtyTests._mock_selectc C s | j � \}}|t_| j � \}}|t_| j� }dd� |D �}tj|d d� tj|d� | jt_| j j
d� | j j
||d g� | j j
d� | j t
�� tj|d � W dQ R X tj||d gg g d�d }| j||d g|� | jtj|d �d� | jtj|d d �d� dS )
z6Test the normal data case on both master_fd and stdin.c S s g | ] }|j � �qS r )�fileno)�.0�sr r r �
<listcomp> s z4SmallPtyTests.test__copy_to_each.<locals>.<listcomp>r7 s from masters
from stdinr9 r N� )r\ r$ rP rN r_ r' r. rc rR rV �appendrW � assertRaises�
IndexError�_copyr, r+ )r �read_from_stdout_fd�mock_stdout_fd�
mock_stdin_fd�write_to_stdin_fdr^ �mastersrb r r r �test__copy_to_each s$ z SmallPtyTests.test__copy_to_eachc
C s� | j � \}}|t_| j � \}}|t_| j� }dd� |D �}|d j� tj|� | jt_| j j
d� | j j
||d g� | j j
d� | j t
�� tj|d � W dQ R X dS )z9Test the empty read EOF case on both master_fd and stdin.c S s g | ] }|j � �qS r )rd )re rf r r r rg $ s z7SmallPtyTests.test__copy_eof_on_all.<locals>.<listcomp>r7 r9 r N)r\ r$ rP rN r_ r1 r' rc rR rV ri rW rj rk rl )r rm rn ro rp r^ rq r r r �test__copy_eof_on_all s
z#SmallPtyTests.test__copy_eof_on_allN) rJ rK rL �__doc__r rY r\ r_ rc rr rs r r r r rM � s rM c C s
t � d S )N)r r r r r �tearDownModule5 s ru �__main__)Z test.supportr r r r- r$ r' rC rR r r] r r& r/ r0 r r r ZTestCaser rM ru rJ �mainr r r r �<module> s.
^