Sindbad~EG File Manager
a
��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_childrenZtermiosN� I wish to buy a fish license.
� For my pet fish, Eric.
c C s t | � d S �N)�print��msg� r �)/usr/local/lib/python3.9/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% �sig�framer r r r! P s zPtyTest.handle_sigc C s d S r r )�signumr) 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 tyh t�d��Y n0 | �t� |�d� t�
|�}zlt� |d� z t� |d�}| �
d |� W n2 ty� } z|jtjkr W Y d }~n
d }~0 0 W t� ||� nt� ||� 0 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�errno�EAGAIN�write�
TEST_STRING_1r r �
TEST_STRING_2�close)r% � master_fdZ
slave_nameZslave_fd�blocking�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 nN tyh t d� Y nH t y� t d� Y n0 t d� t�d� Y n0 t d � t�d
� t�d � n�t d | � zt�
|d
�}W n ty� Y �qY n0 |s��qt j �
t|�dd�dd�� q�t�|d�\}}t�|�}t d|||f � |dk�rZ| �d� n@|d
k�rp| �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 code %d (status %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- �forkZCHILDr0 r1 �_exit�setsidr. �AttributeErrorr4 �sys�stdoutr8 �strr �waitpid�waitstatus_to_exitcoder' r; )r% �pidr< r �status�resr r r � test_fork� sV
�
zPtyTest.test_forkN) �__name__�
__module__� __qualname__r& r! � staticmethodr# rA rU 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 sv | j t_| jt_| jt_| jD ]$}z |�� W q t y@ Y q 0 q | j
D ]&}zt �|� W qJ t yn Y qJ0 qJd S r ) r\ r- r[ r^ r] r` r_ rb r; r. ra r0 )r% �filer r r r �tearDown� s
zSmallPtyTests.tearDownc C s t �� }| j�|� |S r )r0 �pipera �extend)r% Zpipe_fdsr r r �_pipe� s zSmallPtyTests._pipec C s t �� }| j�|� |S r )�socket�
socketpairrb rh )r% rk r r r � _socketpair s zSmallPtyTests._socketpairc C s* | � | j�d�t|�� | j�d�g g fS )Nr )r5 rc �pop�lenrd )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 d � n1 s�0 Y 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> r+ z4SmallPtyTests.test__copy_to_each.<locals>.<listcomp>rB s from masters
from stdinrD r N� )ri r- r] r[ rl r0 r8 rp r_ rc �appendrd � assertRaises�
IndexError�_copyr5 r4 )r% �read_from_stdout_fd�mock_stdout_fd�
mock_stdin_fd�write_to_stdin_fdrk �mastersro 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 d � n1 s�0 Y dS )z9Test the empty read EOF case on both master_fd and stdin.c S s g | ] }|� � �qS r rq rs r r r rv 0 r+ z7SmallPtyTests.test__copy_eof_on_all.<locals>.<listcomp>rB rD r N)ri r- r] r[ rl r; r0 rp r_ rc rx rd ry rz r{ )r% r| r} r~ r rk r� r r r �test__copy_eof_on_all) s
z#SmallPtyTests.test__copy_eof_on_allN) rV rW rX �__doc__r& rf ri rl rp r� r� r r r r rZ � s rZ c C s
t � d S r )r r r r r �tearDownModuleA s r� �__main__)Z test.supportr r r r6 r- r0 rM r_ r rj r r/ r9 r: r
r r ZTestCaser rZ r� rV �mainr r r r �<module> s.
!^
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists