Your IP : 216.73.216.213
3
� f� � @ s� d Z ddlZddlZddlZddlZddlmZ ejdkrDejd��ddlZddl m
Z
ddl m Z yddl m
Z
W n ek
r� dd l mZ
Y nX G d
d � d ej�ZG d d
� d
ej�ZG dd� dej�Zedkr�ej� dS )zTests for window_utils� N)�mockZwin32z Windows only)� _overlapped)�
windows_utils)�support)� test_supportc @ s� e Zd Zdd� Zdd� Zejejd�dd� �Z ej
e e d�d �e
jd
�d d � ��Zd
d� Zej
e e d�d �e
jd
�dd� ��ZdS )�WinsocketpairTestsc C s0 |j d� | jd|jd�� |j� |j� d S )Ns xxxi )�send� assertEqualZrecv�close)�self�ssock�csock� r �I/opt/alt/python36/lib64/python3.6/test/test_asyncio/test_windows_utils.py�check_winsocketpair s
z&WinsocketpairTests.check_winsocketpairc C s t j� \}}| j||� d S )N)r �
socketpairr )r r r
r r r �test_winsocketpair s z%WinsocketpairTests.test_winsocketpairz IPv6 not supported or enabledc C s" t jtjd�\}}| j||� d S )N)�family)r r �socketZAF_INET6r )r r r
r r r �test_winsocketpair_ipv6" s z*WinsocketpairTests.test_winsocketpair_ipv6r z socket.socketpair is availablez asyncio.windows_utils.socketc C sP t j|_t j|_d|j jj_t� t� f|j jj_t� |j jj_ | j
tt j � d S )N� �90 )r r )
r �AF_INET� SOCK_STREAM� return_valueZ getsockname�objectZaccept�OSErrorZconnect� side_effect� assertRaisesr r )r �m_socketr r r �test_winsocketpair_exc' s z)WinsocketpairTests.test_winsocketpair_excc C s> | j ttjtjd� | j ttjtjd� | j ttjdd� d S )N)r )�type� )�proto)r �
ValueErrorr r r Z AF_UNSPECZ
SOCK_DGRAM)r r r r � test_winsocketpair_invalid_args3 s z2WinsocketpairTests.test_winsocketpair_invalid_argsc C sH t j|_t j|_tj� }||j _t|j_| j tt
j � | j |j
j� d S )N)r r r r ZMockr r Zbindr r r r �
assertTruer
Zcalled)r r Zsockr r r �test_winsocketpair_close; s z+WinsocketpairTests.test_winsocketpair_closeN)�__name__�
__module__� __qualname__r r �unittestZ
skipUnlessr Z IPV6_ENABLEDr ZskipIf�hasattrr r Zpatchr r% r' r r r r r s
r c @ s e Zd Zdd� Zdd� ZdS )� PipeTestsc C sr t jd d�\}}�zDtj� }| j|j� | j|jd� |j|d� | j |j� | j|jt
j � d}y |j � W n0 t
k
r� } z| j|j|� W Y d d }~X n
X td��tj� }| j|j� | j|jd� |j|d� | j|jdt
j h� t
j|jgdd�}| j|t
j� | j|j� | j|j|� | j|j� | j|jdt
j h� | j|j � d� W d t
j|� t
j|� X d S )
NT)�
overlappedr �d i� z expected ERROR_IO_INCOMPLETEs helloF)TT)r �piper �
Overlapped� assertFalse�pendingr �error�ReadFiler& �_winapiZERROR_IO_PENDING� getresultr �winerror� RuntimeError� WriteFileZassertIn�WaitForMultipleObjects�event�
WAIT_OBJECT_0� CloseHandle)r Zh1Zh2Zov1ZERROR_IO_INCOMPLETE�eZov2�resr r r �test_pipe_overlappedJ s8
z PipeTests.test_pipe_overlappedc C s� t jdd�\}}tj|� t j|�}| j|j� |� | j|j|� tj � � tj
ddt � ~t j
� W d Q R X ytj|� W n0 tk
r� } z| j|jd� W Y d d }~X n
X td��d S )NT)r. �ignorer � z expected ERROR_INVALID_HANDLE)TT)r r0 r6 r> �
PipeHandler �fileno�handle�warnings�catch_warnings�filterwarnings�ResourceWarningr Z
gc_collectr r8 r9 )r �h�_�pr? r r r �test_pipe_handleo s
zPipeTests.test_pipe_handleN)r( r) r* rA rN r r r r r- H s %r- c @ s e Zd Zdd� ZdS )�
PopenTestsc C sp d}d}t jtjd|gt jt jt jd�}x&|j|j|jgD ]}| j|t j � q:W t
j � }t
j � }t
j � }|j |jj
|� |j|jj
d� |j|jj
d� |j|j|jg}tj|dd�} | j| tj� | j|j� | j|j� | j|j� | j|j� t|�� |j� j� }
|j� j� } | jt|
�d� | jt| �d� | j|j� j� j |
�� | jd j | �� |� W d Q R X d S )
Nz�if 1:
import sys
s = sys.stdin.readline()
sys.stdout.write(s.upper())
sys.stderr.write('stderr')
s blah
z-c)�stdin�stdout�stderrr/ Ti' r s stderr) r �Popen�sys�
executable�PIPErP rQ rR ZassertIsInstancerD r r1 r: rF r5 r< r6 r; r r= r2 r3 r7 �len�rstripZ
assertGreaterr& �upper�
startswith) r Zcommand�msgrM �fZovinZovoutZoverrZeventsr@ �out�errr r r �
test_popen� s:
zPopenTests.test_popenN)r( r) r* r_ r r r r rO � s rO �__main__)�__doc__r rT r+ rG r �platformZSkipTestr6 Zasyncior r Ztestr � ImportErrorr ZTestCaser r- rO r( �mainr r r r �<module> s&
2;2