Your IP : 216.73.216.213
��YfA � @ 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 )
�socketZbindr ZHOSTZlistenZconnectZ getsocknameZaccept�close�OSError)Zfamily�type�proto�l�cZcaddr�aZaddr� r �8/opt/alt/python35/lib64/python3.5/test/test_selectors.pyr s
c C s; g } x. | D]& \ } } | | @r
| j | j � q
W| S)N)�append�fileobj)�readyZflag�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' r) )r r* 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_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 | 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" r. �os�dup2r, )r r* r r r/ r0 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" r. r, )r r* 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" 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 | 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" r. r' � RuntimeErrorr7 r) � __getitem__)r r* �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$ r7 r' r) )r r* 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 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 | 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" 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 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" r. r' r= r7 )r r* 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& rJ rK �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 } 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 rC �ranger r r! r" r. r rI 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_selector�win32z1select.select() cannot be used with empty fd setsc C s< | j � } | j | j � | j | j d d � g � d S)NZtimeoutr )r r r r$ rI )r r* 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)NrB r g �?g�������?g @���r\ )r r r r r r! r. �timer$ rC rI �
assertLessr, r" r: rJ )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 � \ } } 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 re c s
� � d S)Nr )�args)re r r �handlerr s z?BaseSelectorTestCase.test_select_interrupt_exc.<locals>.handlerr rB � g @)r r r r � Exception�signal�SIGALRMra r r! r" r] r' rI 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 � \ } } 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 )rf r r r �<lambda>� s zDBaseSelectorTestCase.test_select_interrupt_noraise.<locals>.<lambda>r rB g �?g �?)r r r r rj rk ra r r! r" r] r: rI rQ )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- r1 r9 �
skipUnlessr3 �namer5 r6 r; r@ rA rH rM rN rR rY ZskipIf�sys�platformr[ r` rO rj rm ro 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 r( r r rT r ZskipTestr r! r" r. �errnoZENOSPCr$ rC rI ) r ZsoftZhardZNUM_FDSr* rX r r �er r r �test_above_fd_setsize� s0
z+ScalableSelectorMixIn.test_above_fd_setsizeN) rb rc rd r Zrequires_mac_verr9 rp ry r} r r r r rt � s rt c @ s e Z d Z e j Z d S)�DefaultSelectorTestCaseN)rb rc rd r! ZDefaultSelectorr r r r r r~ � s r~ c @ s e Z d Z e j Z d S)�SelectSelectorTestCaseN)rb rc rd r! ZSelectSelectorr 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)rb rc rd �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 �tempfileZNamedTemporaryFiler' �IOErrorr r! r" r) r7 )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 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$ Z exceptionr{ ZEBADFr) r7 )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 Z d Z e e d d � Z d S)�DevpollSelectorTestCaser� N)rb rc rd 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{ r3 rU r! rj r rr Ztestr r] r r9 Z
unittest.mockr� r ry � ImportErrorrO r ZAF_INETZ SOCK_STREAMr ZTestCaser rt r~ r rp r� r� r� r� r� rb r r r r �<module> sP
� i.