Your IP : 216.73.216.213
3
� f� � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ejdkrLej d��ddl
m Z ej Z
G dd� dej�Zed kr�ej� dS )
zTests for WindowsConsoleIO
� N)�supportZwin32ztest only relevant on win32)� write_inputc @ s� e Zd Zdd� Zdd� Zdd� Zejej � dd� d kd �d d
� �Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� ZdS ) �WindowsConsoleIOTestsc C s: | j tttj�� | jtttj�� | jtttj�� d S )N)�
assertTrue�
issubclass�ConIO�io� RawIOBase� assertFalse�BufferedIOBase�
TextIOBase)�self� r �;/opt/alt/python36/lib64/python3.6/test/test_winconsoleio.py�test_abc s z WindowsConsoleIOTests.test_abcc # C sb | j tdtd� tj� \}}z| j tdt|� W d tj|� X y td�}W n tk
r^ Y n>X | j|j� � | j |j
� � | j d|j � � |j� |j� ytdd�}W n tk
r� Y n>X | j |j� � | j|j
� � | j d|j � � |j� |j� ytdd�}W n tk
�r Y n>X | j |j� � | j|j
� � | j d|j � � |j� |j� 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� | j tttj� td�}| j|j� � | j|j� � | j |j
� � |j � |j � td�}| j|j� � | j|j� � | j |j
� � |j � |j � tdd�}| j|j� � | j|j� � | j |j
� � |j � |j � t dddd�}| j
|t� |j � 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�}| j|t� |j� t dddd�}| j|t� |j� 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 j� }| jtj|� tjj|d�}t|ddd��4}t j
� d d� dkrV| j |t � n | j
|t � W d Q R X d S ) NzCONOUT$r) r )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��}| j|jd�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 jdd�� t� }W d |t_X | j||� 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, | j d� | j d� | j d� | j d� d S )NZabc123u ϼўТλФЙu A͏B ﬖ̳AA̝u )r; )r
r r r �
test_input� s
z WindowsConsoleIOTests.test_inputc
C s� dj d�}dj d�}xptdd�D ]b}tdddd ��J}t||� d
}x |jd �sb|j|�}||7 }qDW | j||d j|�� 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� dj d�}dj d�}xptdd�D ]b}tdddd ��J}t||� d
}x |jd �sb|j|�}||7 }qDW | j||d j|�� 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��`}djd�}djd�}t||� |jd �|j� }}| j|dd � |� | j|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)r( r )�__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 Z_WindowsConsoleIOr ZTestCaser rM �mainr r r r �<module> s
1