Your IP : 216.73.216.196
B
� f� � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ejdkrLe� d��ddl
m Z ej Z
G dd� dej�Zed kr�e�� dS )
zTests for WindowsConsoleIO
� N)�supportZwin32ztest only relevant on win32)� write_inputc @ s� e Zd Zdd� Zdd� Zdd� Ze�e� � dd� d kd
�d d � �Z
d
d� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� ZdS )�WindowsConsoleIOTestsc C s: | � tttj�� | �tttj�� | �tttj�� d S )N)�
assertTrue�
issubclass�ConIO�io� RawIOBase� assertFalse�BufferedIOBase�
TextIOBase)�self� r �;/opt/alt/python37/lib64/python3.7/test/test_winconsoleio.py�test_abc s z WindowsConsoleIOTests.test_abcc C sb | � tdtd� t�� \}}z| � tdt|� W d t�|� X y td�}W n tk
r^ Y n>X | �|�� � | � |�
� � | � d|� � � |�� |�� ytdd�}W n tk
r� Y n>X | � |�� � | �|�
� � | � d|� � � |�� |�� ytdd�}W n tk
�r Y n>X | � |�� � | �|�
� � | � d|� � � |�� |�� d S )Nznegative file descriptor���z&Cannot open (console|non-console file)r � �w� )
ZassertRaisesRegex�
ValueErrorr �tempfileZmkstemp�os�closer �readabler
�writable� assertEqual�fileno)r
�fd�_�fr r r � test_open_fd sD
z"WindowsConsoleIOTests.test_open_fdc C s� | � tttj� td�}| �|�� � | �|�� � | � |�
� � |� � |� � td�}| �|�� � | �|�� � | � |�
� � |� � |� � tdd�}| �|�� � | �|�� � | � |�
� � |� � |� � t dddd�}| �
|t� |� � d S ) NZCONzCONIN$zCONOUT$r zC:/con�rbr )� buffering)Z assertRaisesr r �sys�
executabler r r
r ZassertIsNotNoner r �open�assertIsInstance)r
r r r r �test_open_nameI s,
z$WindowsConsoleIOTests.test_open_nameNr )� r z+test does not work on Windows 7 and earlierc C sH t dddd�}| �|t� |�� t dddd�}| �|t� |�� d S )Nz
\\.\conin$r! r )r" z //?/conout$�wb)r% r& r r )r
r r r r �test_conin_conout_namese s z-WindowsConsoleIOTests.test_conin_conout_namesc C sp t �� }| �tj|� tj�|d�}t|ddd��4}t �
� d d� dkrV| � |t � n | �
|t � W d Q R X d S )NzCONOUT$r) r )r" r )r( r )r ZmkdtempZ
addCleanupr Zrmtreer �path�joinr% r# �getwindowsversionr& r ZassertNotIsInstance)r
Z temp_pathZ conout_pathr r r r �test_conout_pathp s z&WindowsConsoleIOTests.test_conout_pathc C s, t dd��}| �|�d�d� W d Q R X d S )NzCONOUT$r � r )r r �write)r
r r r r �test_write_empty_data| s z+WindowsConsoleIOTests.test_write_empty_datac C sT t dd�}tj}z*|t_t|jj|d �dd�� t� }W d |t_X | �||� d S )NzCONIN$�rz
z utf-16-le�
surrogatepass) r% r# �stdinr �buffer�raw�encode�inputr )r
�textr4 Z old_stdin�actualr r r �assertStdinRoundTrip� s
z*WindowsConsoleIOTests.assertStdinRoundTripc C s, | � d� | � d� | � d� | � d� d S )NZabc123u ϼўТλФЙu A͏B ﬖ̳AA̝u )r; )r
r r r �
test_input� s
z WindowsConsoleIOTests.test_inputc
C s� d� d�}d� d�}xptdd�D ]b}tdddd ��J}t||� d
}x |�d �sb|�|�}||7 }qDW | �||d �|�� W d Q R X q W d S )
Nu ϼўТλФЙ
z utf-16-lezutf-8r � zCONIN$r! r )r" r/ �
zstdin.read({}))r7 �ranger% r �endswith�readr �format)r
�source�expected�
read_countr4 r: �br r r �test_partial_reads� s
z(WindowsConsoleIOTests.test_partial_readsc
C s� d� d�}d� d�}xptdd�D ]b}tdddd ��J}t||� d
}x |�d �sb|�|�}||7 }qDW | �||d �|�� W d Q R X q W d S )
Nu
z utf-16-lezutf-8r r= zCONIN$r! r )r" r/ r> zstdin.read({}))r7 r? r% r r@ rA r rB )r
rC rD rE r4 r: rF r r r � test_partial_surrogate_reads� s
z2WindowsConsoleIOTests.test_partial_surrogate_readsc C sx t dddd��`}d�d�}d�d�}t||� |�d �|�� }}| �|dd � |� | �|d d � |� W d Q R X d S )
NzCONIN$r! r )r" u Ä
z utf-16-le� Äzutf-8r )r% r7 r rA �readallr )r
r4 rC rD �arF r r r � test_ctrl_z� s
z!WindowsConsoleIOTests.test_ctrl_z)�__name__�
__module__� __qualname__r r r' �unittestZskipIfr# r- r* r. r1 r; r< rG rH rL r r r r r s 1
r �__main__)�__doc__r r r# r rP Ztestr �platformZSkipTestZ _testconsoler �_WindowsConsoleIOr ZTestCaser rM �mainr r r r �<module> s
1