Your IP : 216.73.216.60
���]A � @ s` d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m
Z
d d l Z d d l Z d d l
Z
d d l m Z y d d l Z Wn e k
r� d Z Yn Xe e d � r� e j Z n e j e j d d d � Z d d � Z Gd d
� d
e j � Z Gd d � d � Z Gd
d � d e � Z Gd d � d e � Z e j e e d � d � Gd d � d e e � � Z e j e e d � d � Gd d � d e e � � Z e j e e d � d � Gd d � d e e � � Z e j e e d � d � Gd d � d e e � � Z d! d" � Z e! d# k r\e � d S)$� N)�support)�sleep)� monotonic�
socketpairc C s� t j | | | � �� } | j t j d f � | j � t j | | | � } y\ | j | j � � | j � } x6 | j � \ } } | | k r� | | f S| j � qr WWn t k
r� | j � � Yn XWd QRXd S)Nr )
�socket�bindr ZHOST�listen�connect� getsockname�accept�close�OSError)�family�type�proto�l�cZcaddr�a�addr� r �3/opt/alt/python35/lib64/python3.5/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 @ sT e Z d Z d d � Z d d � Z d d � Z d d � Z e j e j
d k d
� d d � � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z e j e j d k d � d! d" � � Z d# d$ � Z e j e e d% � d& � d' d( � � Z e j e e d% � d& � d) d* � � Z d+ S),�BaseSelectorTestCasec C s9 t � \ } } | j | j � | j | j � | | f S)N)r �
addCleanupr )�self�rd�wrr r r �make_socketpair4 s z$BaseSelectorTestCase.make_socketpairc C s% | j � } | j | j � | j � \ } } | j | t j d � } | j | t j � | j | j
| � | j | j | j � � | j | j
t j � | j | j d � | j t | j d d � | j t | j d t j � | j t | j | t j � | j t | j | j � t j � 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 | t j � | j | � | j t | j d � | j t | j | � d S)Ni?B )
r' r r r$ r( r) r* �
unregisterr/ r1 )r! r2 r" r# r r r �test_unregisterT s
z$BaseSelectorTestCase.test_unregisterc C s� | j � } | j | j � | j � \ } } | j � | j � } } | j | t j � | j | t j � | j � | j � | j | � | j | � d S)N)
r' r r r$ r. r( r) r* � EVENT_WRITEr4 )r! r2 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 | t j � | j | t j � | j � \ } } | j � | j � t j
| j � | � t j
| j � | � | j t j | � | j t j | � | j | � | j | � d S)N) r' r r r$ r. r( r) r* r6 �os�dup2r4 )r! r2 r" r# r7 r8 Zrd2Zwr2r r r �(test_unregister_after_fd_close_and_reuseo s
z=BaseSelectorTestCase.test_unregister_after_fd_close_and_reusec C s� | j � } | j | j � | j � \ } } | j | t j � | j | t j � | j � | j � | j | � | j | � d S)N) r' r r r$ r( r) r* r6 r4 )r! r2 r" r# r r r �"test_unregister_after_socket_close� s
z7BaseSelectorTestCase.test_unregister_after_socket_closec C s� | j � } | j | j � | j � \ } } | j | t j � } | j | t j � } | j | j
| j
� | j | | j | � � | j
| � t � } t � } | j | t j | � } | j | t j | � } | j | j
| j
� | j | j | j � | j | | j | � � | j | j | � | j t | j d t j � t � } t j j � | _ t j j � | _
| j | t j | � | j | j j � | j | j
j � d S)Ni?B )r' r r r$ r( r) r* �modifyr6 ZassertNotEqualr r, �get_keyr4 �objectr% r/ r1 �unittestZmockZMock� assertFalseZcalled) r! r2 r" r# r Zkey2Zd1Zd2Zd3r r r � test_modify� s.
z BaseSelectorTestCase.test_modifyc C s� | j � } | j | j � | j � } | j � \ } } | j | t j � | j | t j � | j � | j t
| j | � | j t
| j | � | j t | j
| � | j t | j
| � d S)N)r' r r �get_mapr$ r( r) r* r6 r/ � RuntimeErrorr@ r1 � __getitem__)r! r2 �mappingr" r# r r r �
test_close� s
z BaseSelectorTestCase.test_closec C sy | j � } | j | j � | j � \ } } | j | t j d � } | j | | j | � � | j t
| j d � d S)Nr% i?B ) r' r r r$ r( r) r* r, r@ r/ r1 )r! r2 r" r# r r r r � test_get_key� s z!BaseSelectorTestCase.test_get_keyc C sD | j � } | j | j � | j � \ } } | j � } | j | � | j t | � d � | j t | � g � | j | t
j d � } | j | | � | j | | | � | j t | � d � | j t | � | j
� g � | j t | j � � | g � | j t � �
| d Wd QRX| j t � � | | =Wd QRXd S)Nr r% � i?B )r' r r r$ rE rC r, �len�listr( r) r* ZassertInr. �valuesr/ r1 � TypeError)r! r2 r" r# �keysr r r r � test_get_map� s"
z!BaseSelectorTestCase.test_get_mapc C s� | j � } | j | j � | j � \ } } | j | t j � | j | t j � } | j � } xV | D]N \ } } | j t
| t j � � | j | � | j | t j t j B@� qi W| j
| t j f g | � d S)N)r' r r r$ r( r) r* r6 �select�
assertTrue�
isinstancer+ rC r, )r! r2 r" r# Zwr_key�resultr r r r r � test_select� s
z BaseSelectorTestCase.test_selectc C s� | j � } | j | j � | j � \ } } | �- } | j | t j � | j | t j � Wd QRX| j t | j
| � | j t | j
| � d S)N) r' r r r$ r( r) r* r6 r/ rF r@ )r! r2 r" r# Zselr r r �test_context_manager� s z)BaseSelectorTestCase.test_context_managerc C sa | j � } | j | j � t | d � r] | j � } | j t | t � � | j | d � d S)Nr. r ) r' r r �hasattrr. rS rT �int�assertGreaterEqual)r! r2 r- r r r � test_fileno s z BaseSelectorTestCase.test_filenoc C s | j � } | j | j � d } d } t | � } g } g } i } i } xz t | � D]l } | j � \ }
} | j |
t j � | j | t j � | j
|
� | j
| � | | |
<|
| | <qY Wg } x,| r�| j � }
t |
t j � } | s | j
d � t j | � } | j | � xP t d � D]5 } | j � }
t |
t j � } | r]Pt d � q2W| j
d � | j | | g | � | d }
|
j | � } | j t | � | � | 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 rL �ranger$ r( r) r* r6 r rR r Zfail�random�choice�sendr r, �recvr4 �remove)r! r2 Z NUM_SOCKETSZMSGZMSG_LENZreadersZwritersZr2wZw2r�ir" r# Zbufsr Z
ready_writersZ
ready_readers�bufr r r �
test_selector sT
z"BaseSelectorTestCase.test_selector�win32z1select.select() cannot be used with empty fd setsc C s< | j � } | j | j � | j | j d d � g � d S)N�timeoutr )r' r r r, rR )r! r2 r r r �test_empty_selectD s z&BaseSelectorTestCase.test_empty_selectc C sg | j � } | j | j � | j � \ } } | j | t j � t � } | j d t | j
d � � � | j d t | j
d � � � | j t � | d � | j | � | j | t j
� t � } | j | j
d � � | j | j
d � � | j t � | d � t � } | j | j
d � � t � } | | } | j d | k oZd k n | � d S)NrK r g �?g�������?g @���ri )r' r r r$ r( r) r6 �timer, rL rR �
assertLessr4 r* rC rS )r! r2 r" r# �t�t0�t1Zdtr r r � test_timeoutM s&
z!BaseSelectorTestCase.test_timeout�alarmz%signal.alarm() required for this testc
s� | j � } | j | j � | j � \ } } Gd d � d t � � � f d d � } t j t j | � } | j t j t j | � | j t j d � t j d � | j | t j
� t � } | j � � � | j
d � Wd QRX| j t � | d � d S) Nc @ s e Z d Z d S)zGBaseSelectorTestCase.test_select_interrupt_exc.<locals>.InterruptSelectN)�__name__�
__module__� __qualname__r r r r �InterruptSelecto s rt c s
� � d S)Nr )�args)rt r r �handlerr s z?BaseSelectorTestCase.test_select_interrupt_exc.<locals>.handlerr rK � g @)r' r r r$ � Exception�signal�SIGALRMrp r( r) r* rj r/ rR rk )r! r2 r" r# rv �orig_alrm_handlerrl r )rt r �test_select_interrupt_excg s
z.BaseSelectorTestCase.test_select_interrupt_excc C s� | j � } | j | j � | j � \ } } t j t j d d � � } | j t j t j | � | j t j d � t j d � | j | t j � t
� } | j | j d � � | j
t
� | d � d S)Nc W s d S)Nr )ru r r r �<lambda>� s zDBaseSelectorTestCase.test_select_interrupt_noraise.<locals>.<lambda>r rK g �?g �?)r' r r r$ ry rz rp r( r) r* rj rC rR rZ )r! r2 r" r# r{ rl r r r � test_select_interrupt_noraise� s
z2BaseSelectorTestCase.test_select_interrupt_noraiseN) rq rr rs r$ r3 r5 r9 rB �
skipUnlessr; �namer= r> rD rI rJ rQ rV rW r[ re ZskipIf�sys�platformrh ro rX ry r| r~ r r r r r 2 s* ' &
5 r c @ s@ e Z d Z e j d d � e j e d � d d � � � Z d S)�ScalableSelectorMixInr& � zTest needs resource modulec ! C s� t j t j � \ } } yK t j t j | | f � | j t j t j | | f � t | d � } Wn t t f k
r� | } Yn X| d 8} | j � } | j | j � x� t
| d � D]� } y | j � \ } } Wn t k
r� | j d � Yn Xy* | j
| t j � | j
| t j � Wq� t k
rl} z' | j t j k rW| j d � � WYd d } ~ Xq� Xq� W| j | d t | j � � � d S)N� � � zFD limit reachedi )�resourceZ getrlimitZ
RLIMIT_NOFILEZ setrlimitr �minr
r0 r' r r] r$ ZskipTestr( r) r* r6 �errnoZENOSPCr, rL rR ) r! ZsoftZhardZNUM_FDSr2 rc r" r# �er r r �test_above_fd_setsize� s0
z+ScalableSelectorMixIn.test_above_fd_setsizeN) rq rr rs r Zrequires_mac_verrB r r� r� r r r r r� � s r� c @ s e Z d Z e j Z d S)�DefaultSelectorTestCaseN)rq rr rs r) �DefaultSelectorr' r r r r r� � s r� c @ s e Z d Z e j Z d S)�SelectSelectorTestCaseN)rq rr rs r) �SelectSelectorr' r r r r r� � s r� � PollSelectorz!Test needs selectors.PollSelectorc @ s" e Z d Z e e d d � Z d S)�PollSelectorTestCaser� N)rq rr rs �getattrr) r' r r r r r� � s r� �
EpollSelectorz"Test needs selectors.EpollSelectorc @ s. e Z d Z e e d d � Z d d � Z d S)�EpollSelectorTestCaser� Nc C st | j � } t j � �U } | j t � � | j | t j � Wd QRX| j t � � | j | � Wd QRXWd QRXd S)N)
r' �tempfile�NamedTemporaryFiler/ �IOErrorr( r) r* r1 r@ )r! r2 �fr r r �test_register_file� s z(EpollSelectorTestCase.test_register_file)rq rr rs r� r) r' r� r r r r r� � s r� �KqueueSelectorz$Test needs selectors.KqueueSelector)c @ s. e Z d Z e e d d � Z d d � Z d S)�KqueueSelectorTestCaser� Nc C s� | j � } t j � } | j t � � } | j | t j � Wd QRX| j | j j
t
j � | j t � � | j
| � Wd QRXd S)N)r' r Z make_bad_fdr/ r
r( r) r* r, � exceptionr� �EBADFr1 r@ )r! r2 Zbad_f�cmr r r �test_register_bad_fd� s z+KqueueSelectorTestCase.test_register_bad_fd)rq rr rs r� r) r' r� r r r r r� � s r� �DevpollSelectorz$Test needs selectors.DevpollSelectorc @ s" e Z d Z e e d d � Z d S)�DevpollSelectorTestCaser� N)rq rr rs r� r) r' r r r r r� � s r� c C s3 t t t t t t g } t j | � t j � d S)N) r� r� r� r� r� r� r Z run_unittestZ
reap_children)Ztestsr r r � test_main s
r� �__main__)"r� r; r^ r) ry r r� Ztestr rj r rB Z
unittest.mockr� r r� � ImportErrorrX r �AF_INET� SOCK_STREAMr ZTestCaser r� r� r� r r� r� r� r� r� rq r r r r �<module> sP
� i.