File: //lib64/python3.6/test/__pycache__/test_selectors.cpython-36.pyc
3
\�B � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dl m
Z
d dl Z d dl Z d dl
Z
d dl mZ y d dlZW n ek
r� dZY nX eed�r�ejZnejejd fdd�Zdd� ZG d d
� d
e j�ZG d d � d �ZG d
d� de�ZG dd� de�Ze jeed�d�G dd� dee��Z e jeed�d�G dd� dee��Z e jeed�d�G dd � d ee��Z e jeed �d �G d d � d ee��Z d!d"� Z e!d#k�r�e � dS )$� N)�support)�sleep)� monotonic�
socketpairc C s� t j | ||���}|jtjdf� |j� t j | ||�}yD|j|j� � |j� }x(|j� \}}||krl||fS |j� qPW W n t k
r� |j� � Y nX W d Q R X d S )Nr )
�socketZbindr ZHOSTZlistenZconnectZ getsocknameZaccept�close�OSError)Zfamily�type�proto�l�cZcaddr�aZaddr� r �&/usr/lib64/python3.6/test_selectors.pyr s c C s. g }x$| D ] \}}||@ r
|j |j� q
W |S )N)�append�fileobj)�ready�flag�match�key�eventsr r r �find_ready_matching* s
r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zeje j
d kd
�d d � �Z d
d� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zejejd kd �d!d"� �Zd#d$� Zejeed%�d&�d'd(� �Z ejeed%�d&�d)d*� �Z d+S ),�BaseSelectorTestCasec C s* t � \}}| j|j� | j|j� ||fS )N)r �
addCleanupr )�self�rd�wrr r r �make_socketpair4 s
z$BaseSelectorTestCase.make_socketpairc C s� | j � }| j|j� | j� \}}|j|tjd�}| j|tj� | j |j
|� | j |j |j � � | j |j
tj� | j |jd� | jt|jdd� | jt|jdtj� | jt|j|tj� | jt|j|j � tj� d S )N�datar i?B �
i����)�SELECTORr r r �register� selectors�
EVENT_READZassertIsInstance� SelectorKey� assertEqualr �fd�filenor r � assertRaises�
ValueError�KeyError)r �sr r r r r r �
test_register: s z"BaseSelectorTestCase.test_registerc C s\ | j � }| j|j� | j� \}}|j|tj� |j|� | jt |jd� | jt |j|� d S )Ni?B )
r r r r r! r" r# �
unregisterr( r* )r r+ r r r r r �test_unregisterT s
z$BaseSelectorTestCase.test_unregisterc C sv | j � }| j|j� | j� \}}|j� |j� }}|j|tj� |j|tj� |j� |j� |j |� |j |� d S )N)
r r r r r' r! r" r# � EVENT_WRITEr- )r r+ r r �r�wr r r � test_unregister_after_fd_closec s
z3BaseSelectorTestCase.test_unregister_after_fd_close�posixzrequires posixc C s� | j � }| j|j� | j� \}}|j� |j� }}|j|tj� |j|tj� | j� \}}|j� |j� t j
|j� |� t j
|j� |� | jt j|� | jt j|� |j |� |j |� d S )N) r r r r r' r! r" r# r/ �os�dup2r- )r r+ r r r0 r1 Zrd2Zwr2r r r �(test_unregister_after_fd_close_and_reuseo s
z=BaseSelectorTestCase.test_unregister_after_fd_close_and_reusec C sd | j � }| j|j� | j� \}}|j|tj� |j|tj� |j� |j� |j|� |j|� d S )N) r r r r r! r" r# r/ r- )r r+ r r r r r �"test_unregister_after_socket_close� s
z7BaseSelectorTestCase.test_unregister_after_socket_closec C s6 | j � }| j|j� | j� \}}|j|tj�}|j|tj�}| j |j
|j
� | j ||j |�� |j
|� t� }t� }|j|tj|�}|j|tj|�}| j |j
|j
� | j |j|j� | j ||j |�� | j |j|� | jt|jdtj� t� }tjj� |_tjj� |_
|j|tj|� | j|jj� | j|j
j� d S )Ni?B )r r r r r! r" r# Zmodifyr/ ZassertNotEqualr r% �get_keyr- �objectr r( r* �unittestZmockZMock� assertFalseZcalled) r r+ r r r Zkey2Zd1Zd2Zd3r r r � test_modify� s.
z BaseSelectorTestCase.test_modifyc C s� | j � }| j|j� |j� }| j� \}}|j|tj� |j|tj� |j� | j t
|j |� | j t
|j |� | j t |j
|� | j t |j
|� d S )N)r r r �get_mapr r! r" r# r/ r( � RuntimeErrorr8 r* � __getitem__)r r+ �mappingr r r r r �
test_close� s z BaseSelectorTestCase.test_closec C sV | j � }| j|j� | j� \}}|j|tjd�}| j||j|�� | j t
|jd� d S )Nr i?B ) r r r r r! r" r# r% r8 r( r* )r r+ r r r r r r � test_get_key� s z!BaseSelectorTestCase.test_get_keyc C s� | j � }| j|j� | j� \}}|j� }| j|� | jt|�d� | jt|�g � |j |t
j d�}| j ||� | j||| � | jt|�d� | jt|�|j
� g� | jt|j� �|g� | jt�� |d W d Q R X | jt�� ||= W d Q R X d S )Nr r � i?B )r r r r r= r; r% �len�listr! r" r# ZassertInr' �valuesr( r* � TypeError)r r+ r r �keysr r r r � test_get_map� s"
z!BaseSelectorTestCase.test_get_mapc C s� | j � }| j|j� | j� \}}|j|tj� |j|tj�}|j� }xD|D ]<\}}| j t
|tj �� | j |� | j |tjtjB @ � qJW | j
|tjfg|� d S )N)r r r r r! r" r# r/ �select�
assertTrue�
isinstancer$ r; r% )r r+ r r Zwr_key�resultr r r r r � test_select� s
z BaseSelectorTestCase.test_selectc C sp | j � }| j|j� | j� \}}|�"}|j|tj� |j|tj� W d Q R X | jt |j
|� | jt |j
|� d S )N) r r r r r! r" r# r/ r( r> r8 )r r+ r r Zselr r r �test_context_manager� s z)BaseSelectorTestCase.test_context_managerc C sF | j � }| j|j� t|d�rB|j� }| jt|t�� | j|d� d S )Nr' r ) r r r �hasattrr' rK rL �int�assertGreaterEqual)r r+ r& r r r � test_fileno s
z BaseSelectorTestCase.test_filenoc C s� | j � }| j|j� d}d}t|�}g }g }i }i }x\t|�D ]P} | j� \}
} |j|
tj� |j| tj � |j
|
� |j
| � | ||
< |
|| < q>W g } x�|�rx|j � }
t |
tj �}|s�| j
d� tj|�} | j|� x<td�D ]&} |j � }
t |
tj�}|r�P td� q�W | j
d� | j|| g|� |d }
|
j|�}| jt|�|� | j
|� |j||
� |j|
� |j||
� q�W | j| |g| � d S )N� s This is a test.z no sockets ready for writingr g�������?z no sockets ready for readingr )r r r rD �ranger r! r" r# r/ r rJ r Zfail�randomZchoice�sendr r% Zrecvr- �remove)r r+ Z NUM_SOCKETSZMSGZMSG_LENZreadersZwritersZr2wZw2r�ir r Zbufsr Z
ready_writersZ
ready_readersZbufr r r �
test_selector sT
z"BaseSelectorTestCase.test_selectorZwin32z1select.select() cannot be used with empty fd setsc C s, | j � }| j|j� | j|jdd�g � d S )Nr )Ztimeout)r r r r% rJ )r r+ r r r �test_empty_selectD s z&BaseSelectorTestCase.test_empty_selectc C s | j � }| j|j� | j� \}}|j|tj� t� }| jdt |j
d��� | jdt |j
d��� | j t� | d� |j |� |j|tj
� t� }| j|j
d�� | j|j
d�� | j t� | d� t� }| j|j
d�� t� }|| }| jd| k� o�dkn |� d S )NrC r g �?g�������?g @���r\ )r r r r r! r" r/ �timer% rD rJ �
assertLessr- r# r; rK )r r+ r r �tZt0Zt1Zdtr r r � test_timeoutM s&
z!BaseSelectorTestCase.test_timeout�alarmz%signal.alarm() required for this testc s� | j � }| j|j� | j� \}}G dd� dt�� � fdd�}tjtj|�}| jtjtj|� zTtjd� |j|t j
� t � }| j � �� |j
d� W d Q R X | jt � | d� W d tjd� X d S ) Nc @ s e Zd ZdS )zGBaseSelectorTestCase.test_select_interrupt_exc.<locals>.InterruptSelectN)�__name__�
__module__� __qualname__r r r r �InterruptSelecto s re c s � �d S )Nr )�args)re r r �handlerr s z?BaseSelectorTestCase.test_select_interrupt_exc.<locals>.handlerrC � g @r )r r r r � Exception�signal�SIGALRMra r! r" r# r] r( rJ r^ )r r+ r r rg �orig_alrm_handlerr_ r )re r �test_select_interrupt_excg s
z.BaseSelectorTestCase.test_select_interrupt_excc C s� | j � }| j|j� | j� \}}tjtjdd� �}| jtjtj|� zDtjd� |j|tj � t
� }| j |j d�� | j
t
� | d� W d tjd� X d S )Nc W s d S )Nr )rf r r r �<lambda>� s zDBaseSelectorTestCase.test_select_interrupt_noraise.<locals>.<lambda>rC g �?g �?r )r r r r rj rk ra r! r" r# r] r; rJ rR )r r+ r r rl r_ r r r � test_select_interrupt_noraise� s
z2BaseSelectorTestCase.test_select_interrupt_noraiseN) rb rc rd r r, r. r2 r: �
skipUnlessr4 �namer6 r7 r< rA rB rI rN rO rS rZ ZskipIf�sys�platformr[ r` rP rj rm ro r r r r r 2 s* &
5 r c @ s, e Zd Zejdd�ejed�dd� ��ZdS )�ScalableSelectorMixInr � zTest needs resource modulec
! C s� t jt j�\}}y6t jt j||f� | jt jt j||f� t|d�}W n ttfk
rb |}Y nX |d8 }| j� }| j|j � x�t
|d �D ]�}y| j � \}}W n tk
r� | j d� Y nX y |j
|tj� |j
|tj� W q� tk
�r } z |jtjk�r| j d� � W Y d d }~X q�X q�W y |j� } W nJ tk
�r| } z,|jtjk�rjtjdk�rj| j d� � W Y d d }~X nX | j|d t| �� d S )N� � � zFD limit reached�darwinz%Invalid argument error calling poll()i )�resourceZ getrlimitZ
RLIMIT_NOFILEZ setrlimitr �minr r) r r rU r ZskipTestr! r" r# r/ �errnoZENOSPCrJ ZEINVALrr rs r% rD )
r ZsoftZhardZNUM_FDSr+ rY r r �eZfdsr r r �test_above_fd_setsize� s<
z+ScalableSelectorMixIn.test_above_fd_setsizeN) rb rc rd r Zrequires_mac_verr: rp rz r~ r r r r rt � s
rt c @ s e Zd ZejZdS )�DefaultSelectorTestCaseN)rb rc rd r" ZDefaultSelectorr r r r r r � s r c @ s e Zd ZejZdS )�SelectSelectorTestCaseN)rb rc rd r" ZSelectSelectorr r r r r r� � s r� � PollSelectorz!Test needs selectors.PollSelectorc @ s e Zd Zeedd�ZdS )�PollSelectorTestCaser� N)rb rc rd �getattrr" r r r r r r� � s r� �
EpollSelectorz"Test needs selectors.EpollSelectorc @ s e Zd Zeedd�Zdd� ZdS )�EpollSelectorTestCaser� Nc C sd | j � }tj� �J}| jt�� |j|tj� W d Q R X | jt�� |j |� W d Q R X W d Q R X d S )N)
r �tempfileZNamedTemporaryFiler( �IOErrorr! r" r# r* r8 )r r+ �fr r r �test_register_file� s
z(EpollSelectorTestCase.test_register_file)rb rc rd r� r" r r� r r r r r� � s r� �KqueueSelectorz$Test needs selectors.KqueueSelector)c @ s e Zd Zeedd�Zdd� ZdS )�KqueueSelectorTestCaser� Nc C sj | j � }tj� }| jt��}|j|tj� W d Q R X | j|j j
t
j � | jt �� |j
|� W d Q R X d S )N)r r Z make_bad_fdr( r r! r" r# r% Z exceptionr| ZEBADFr* r8 )r r+ Zbad_f�cmr r r �test_register_bad_fd� s z+KqueueSelectorTestCase.test_register_bad_fd)rb rc rd r� r" r r� r r r r r� � s r� �DevpollSelectorz$Test needs selectors.DevpollSelectorc @ s e Zd Zeedd�ZdS )�DevpollSelectorTestCaser� N)rb rc rd r� r" r r r r r r� s r� c C s& t tttttg} tj| � tj� d S )N) r r� r� r� r� r� r Z run_unittestZ
reap_children)Ztestsr r r � test_main s
r� �__main__)"r| r4 rV r" rj r rr Ztestr r] r r: Z
unittest.mockr� r rz � ImportErrorrP r ZAF_INETZ SOCK_STREAMr ZTestCaser rt r r� rp r� r� r� r� r� rb r r r r �<module> sR
n5