Your IP : 216.73.217.129
U
i�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 �� dS )� )�verbose�
import_module�
reap_childrenZtermiosN� I wish to buy a fish license.
� For my pet fish, Eric.
c C s t | � d S �N)�print��msg� r �2/opt/alt/python38/lib64/python3.8/test/test_pty.py�debug s r
c C s d S r r r r r r r
s c C s0 | � d�r| �dd�S | � d�r,| �dd�S | S )Ns
�
�
)�endswith�replace)�datar r r �normalize_output% s
r c C s t j| ddd�}|�� S )z8Read one line. May block forever if no newline is read.�rbF)�mode�closefd)�io�FileIO�readline)�fd�readerr r r � _readline9 s r c @ s8 e Zd Zdd� Zdd� Zedd� �Zdd� Zd d
� Zd S ) �PtyTestc C s` t � t j| j�}| �t j t j|� t � t j| j�}| �t j t j|� | �t jd� t �d� d S )Nr �
)�signal�SIGALRM�
handle_sigZ
addCleanup�SIGHUP�
handle_sighup�alarm)�selfZ old_alarmZ
old_sighupr r r �setUpD s z
PtyTest.setUpc C s | � d� d S )Nz isatty hung)�fail)r% Zsig�framer r r r! P s zPtyTest.handle_sigc C s d S r r )Zsignumr( r r r r# S s zPtyTest.handle_sighupc
C s� zLt d� t�� \}}t d||f � t d|f � t�|�}t d| � W n tk
rj t�d��Y nX | �t� |�d� t�
|�}z^t� |d� z t� |d�}| �
d |� W n0 tk
r� } z|jtjkr҂ W 5 d }~X Y nX W 5 t� ||� X t d
� t�|t� t|�}| �
d t|�� t d � t�|td d
� � t�|td
d � � t|�}| �
dt|�� t�|� t�|� 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_fdr zWriting chunked output� r )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_basicZ sF �
�
zPtyTest.test_basicc C s� t d� t�� \}}|tjkr�t�d�s:t d� t�d� t d� z t�� W nR tk
rj t d� Y nJ t k
r� t d� Y n0 t d� t�d� Y nX t d � t�d
� t�d � n�t d | � zt�
|d
�}W n tk
r� Y �q Y nX |�s�q t j �
t|�dd�dd�� q�t�|d�\}}|d? }t d|||f � |dk�r`| �d� n@|d
k�rv| �d� n*|dk�r�| �d� n|d k�r�| �d� t�|� 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 r 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, �AttributeErrorr2 �sys�stdoutr5 �strr �waitpidr' r8 )r% �pidr9 r Zstatus�resr r r � test_fork� sV
�
zPtyTest.test_forkN) �__name__�
__module__� __qualname__r& r! � staticmethodr# r= rP r r r r r C s
3r 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 r ) 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 sz | j t_| jt_| jt_| jD ]&}z |�� W q t k
rB Y q X q | j
D ](}zt �|� W qL t k
rr Y qLX qLd S r ) rW r+ rV rY rX r[ rZ r] r8 r, r\ r. )r% �filer r r r �tearDown� s
zSmallPtyTests.tearDownc C s t �� }| j�|� |S r )r. �piper\ �extend)r% Zpipe_fdsr r r �_pipe� s zSmallPtyTests._pipec C s t �� }| j�|� |S r )�socket�
socketpairr] rc )r% rf r r r � _socketpair s zSmallPtyTests._socketpairc C s* | � | j�d�t|�� | j�d�g g fS )Nr )r3 r^ �pop�lenr_ )r% �rfdsZwfdsZxfdsr r r � _mock_select s zSmallPtyTests._mock_selectc C s | � � \}}|t_| � � \}}|t_| �� }dd� |D �}t�|d d� t�|d� | jt_| j �
d� | j �
||d g� | j �
d� | � t
�� t�|d � W 5 Q R X t�||d gg g d�d }| �||d g|� | �t�|d�d� | �t�|d d�d� d S )
z6Test the normal data case on both master_fd and stdin.c S s g | ] }|� � �qS r ��fileno��.0�sr r r �
<listcomp> s z4SmallPtyTests.test__copy_to_each.<locals>.<listcomp>r> s from masters
from stdinr@ r � N)rd r+ rX rV rg r. r5 rk rZ r^ �appendr_ � assertRaises�
IndexError�_copyr3 r2 )r% �read_from_stdout_fd�mock_stdout_fd�
mock_stdin_fd�write_to_stdin_fdrf �mastersrj r r r �test__copy_to_each
s$ z SmallPtyTests.test__copy_to_eachc C s� | � � \}}|t_| � � \}}|t_| �� }dd� |D �}|d �� t�|� | jt_| j �
d� | j �
||d g� | j �
d� | � t
�� t�|d � W 5 Q R X dS )z9Test the empty read EOF case on both master_fd and stdin.c S s g | ] }|� � �qS r rl rn r r r rq 0 s z7SmallPtyTests.test__copy_eof_on_all.<locals>.<listcomp>r> r@ r N)rd r+ rX rV rg r8 r. rk rZ r^ rs r_ rt ru rv )r% rw rx ry rz rf r{ r r r �test__copy_eof_on_all) s
z#SmallPtyTests.test__copy_eof_on_allN) rQ rR rS �__doc__r& ra rd rg rk r| r} r r r r rU � s rU c C s
t � d S r )r r r r r �tearDownModuleA s r �__main__)Z test.supportr r r r4 r+ r. rJ rZ r re r r- r6 r7 r
r r ZTestCaser rU r rQ �mainr r r r �<module> s.
!^