File: //usr/local/lib/python3.7/test/__pycache__/test_pty.cpython-37.pyc
B
��g�/ � @ 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_childrenZtermiosNs I wish to buy a fish license.
s For my pet fish, Eric.
c C s t | � d S )N)�print)�msg� r �)/usr/local/lib/python3.7/test/test_pty.py�debug s r c C s d S )Nr )r r r r r s c C s0 | � d�r| �dd�S | � d�r,| �dd�S | S )Ns
�
s
)�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 )Nr )Zsignumr# r r r r S s zPtyTest.handle_sighupc
C s� yLt 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� y t� |d�}| �
d |� W n0 tk
r� } z|jtjkr҂ W d d }~X Y nX W d 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_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_basicZ sB
zPtyTest.test_basicc C s� t d� t�� \}}|tjkr�t�d�s:t d� t�d� t d� y 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 | � xNyt�
|d
�}W n tk
r� P Y nX |s�P t j �
t|�dd�dd�� q�W t�|d�\}}|d? }t d|||f � |dk�r\| �d� n@|d
k�rr| �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 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�stdoutr0 �strr �waitpidr" r3 )r �pidr4 r
�statusZresr r r � test_fork� sV
zPtyTest.test_forkN) �__name__�
__module__� __qualname__r! r � staticmethodr r8 rK 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 )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 |�� W q t k
rD Y q X q W x2| j
D ](}yt �|� W qR t k
rx Y qRX qRW d S )N) rR r& rQ rT rS rV rU rX r3 r' rW r) )r �filer r r r �tearDown� s
zSmallPtyTests.tearDownc C s t �� }| j�|� |S )N)r) �piperW �extend)r Zpipe_fdsr r r �_pipe� s zSmallPtyTests._pipec C s t �� }| j�|� |S )N)�socket�
socketpairrX r^ )r ra r r r � _socketpair s zSmallPtyTests._socketpairc C s* | � | j�d�t|�� | j�d�g g fS )Nr )r. rY �pop�lenrZ )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 dQ R X t�||d gg g d�d }| �||d g|� | �t�|d �d� | �t�|d d �d� dS )
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>r9 s from masters
from stdinr; r N� )r_ r& rS rQ rb r) r0 rf rU rY �appendrZ � assertRaises�
IndexError�_copyr. r- )r �read_from_stdout_fd�mock_stdout_fd�
mock_stdin_fd�write_to_stdin_fdra �mastersre 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 dQ R X dS )z9Test the empty read EOF case on both master_fd and stdin.c S s g | ] }|� � �qS r )rg )rh ri r r r rj 0 s z7SmallPtyTests.test__copy_eof_on_all.<locals>.<listcomp>r9 r; r N)r_ r& rS rQ rb r3 r) rf rU rY rl rZ rm rn ro )r rp rq rr rs ra rt r r r �test__copy_eof_on_all) s
z#SmallPtyTests.test__copy_eof_on_allN) rL rM rN �__doc__r! r\ r_ rb rf ru rv r r r r rP � s rP c C s
t � d S )N)r r r r r �tearDownModuleA s rx �__main__)Z test.supportr r r r/ r& r) rE rU r r` r r( r1 r2 r r r ZTestCaser rP rx rL �mainr r r r �<module> s.
!^