File: //usr/local/lib/python3.7/test/__pycache__/test_winconsoleio.cpython-37.pyc
B
��g� � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ejdkrLe� d��ddl
m Z ej Z
G dd� dej�Zed kr�e�� dS )
zTests for WindowsConsoleIO
� N)�support�win32ztest 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 �2/usr/local/lib/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 �textr5 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# r0 �
zstdin.read({}))r8 �ranger& r �endswith�readr �format)r �source�expected�
read_countr5 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# r0 r? zstdin.read({}))r8 r@ r& r rA rB r rC )r rD rE rF r5 r; rG 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& r8 r rB �readallr )r r5 rD rE �arG r r r � test_ctrl_z� s
z!WindowsConsoleIOTests.test_ctrl_z)�__name__�
__module__� __qualname__r r! r( �unittestZskipIfr$ r. r+ r/ r2 r< r= rH rI rM r r r r r s 1
r �__main__)�__doc__r r r$ r rQ Ztestr �platformZSkipTestZ _testconsoler �_WindowsConsoleIOr ZTestCaser rN �mainr r r r �<module> s
1