Your IP : 216.73.216.213
�
f f�; � @ sX 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 Z d d l m Z d d l
m Z d d l Z d d l
Z y d d l
m Z
Wn" e k
r� d d l
m
Z
Yn Xy d d l Z Wn e k
r� d Z Yn Xe e d � re 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 d d � Z e d k rTe � n d S)!� N)�support)�sleep)� monotonic)�time�
socketpairc C s� t j | | | � �� } | j t j d f � | j d � t j | | | � } y\ | j | j � � | j � } x6 | j � \ } } | | k r� | | f S| j � qu WWn t k
r� | j � � Yn XWd QXd S)Nr � )
�socketZbindr ZHOSTZlistenZconnectZ getsocknameZaccept�close�OSError)Zfamily�type�proto�l�cZcaddr�aZaddr� r �8/opt/alt/python34/lib64/python3.4/test/test_selectors.pyr s
c C s> g } x1 | D]) \ } } | | @r
| j | j � q
q
W| S)N)�append�fileobj)�readyZflag�match�key�eventsr r r �find_ready_matching- s
r c @ s- 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 d) S)*�BaseSelectorTestCasec C s9 t � \ } } | j | j � | j | j � | | f S)N)r �
addCleanupr )�self�rd�wrr r r �make_socketpair7 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_unregisterW 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_closef 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$ r0 �os�dup2r. )r r, r r r1 r2 Zrd2Zwr2r r r �(test_unregister_after_fd_close_and_reuser 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$ r0 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$ Zmodifyr0 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$ r0 r) r+ r9 � __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& r9 r) r+ )r r, r r r r r r � test_get_key� s z!BaseSelectorTestCase.test_get_keyc C sB | 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 QX| j t � � | | =Wd QXd 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$ r0 �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 QX| j t | j
| � | j t | j
| � d S)N) r! r r r r" r# r$ r0 r) r+ r9 )r r, r r Zselr r r �test_context_manager� s z)BaseSelectorTestCase.test_context_managerc C sd | j � } | j | j � t | d � r` | j � } | j t | t � � | j | d � n d S)Nr( r ) r! r r �hasattrr( rK rL �intZassertGreaterEqual)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 } x2| r| j � }
t |
t j � } | s | j
d � n t j | � } | j | � xS t d � D]8 } | j � }
t |
t j � } | rcPn t d � q5W| 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 rD �ranger r" r# r$ r0 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_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& rJ )r r, r r r �test_empty_selectG 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)NrC r g �?g�������?g @���r\ )r! r r r r" r# r0 r r& rD rJ �
assertLessr. r$ r< rK )r r, r r �tZt0Zt1Zdtr r r � test_timeoutP s&
z!BaseSelectorTestCase.test_timeout�alarmz%signal.alarm() required for this testc 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 )�argsr r r �<lambda>r s z<BaseSelectorTestCase.test_select_interrupt.<locals>.<lambda>r rC � g @)r! r r r �signal�SIGALRMr` r" r# r$ r r< rJ r] )r r, r r Zorig_alrm_handlerr^ r r r �test_select_interruptj s
z*BaseSelectorTestCase.test_select_interruptN) �__name__�
__module__� __qualname__r r- r/ r3 r; �
skipUnlessr5 �namer7 r8 r= rA rB rI rN rO rR rY ZskipIf�sys�platformr[ r_ rP rd rf r r r r r 5 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
ro} z* | j t j k rZ| j d � n � WYd d } ~ Xq� Xq� W| j | d t | j � � � d S)Nrc � � zFD limit reachedi )�resourceZ getrlimitZ
RLIMIT_NOFILEZ setrlimitr �minr
r* r! r rT r ZskipTestr" r# r$ r0 �errnoZENOSPCr& rD rJ ) r ZsoftZhardZNUM_FDSr, rX r r �er r r �test_above_fd_setsize� s0
z+ScalableSelectorMixIn.test_above_fd_setsizeN) rg rh ri r Zrequires_mac_verr; rj rr rv r r r r rn ~ s rn c @ s e Z d Z e j Z d S)�DefaultSelectorTestCaseN)rg rh ri r# ZDefaultSelectorr! r r r r rw � s rw c @ s e Z d Z e j Z d S)�SelectSelectorTestCaseN)rg rh ri r# ZSelectSelectorr! r r r r rx � s rx � PollSelectorz!Test needs selectors.PollSelectorc @ s" e Z d Z e e d d � Z d S)�PollSelectorTestCasery N)rg rh ri �getattrr# r! r r r r rz � s rz �
EpollSelectorz"Test needs selectors.EpollSelectorc @ s. e Z d Z e e d d � Z d d � Z d S)�EpollSelectorTestCaser| Nc C sq | j � } t j � �S } | j t � � | j | t j � Wd QX| j t � � | j | � Wd QXWd QXd S)N)
r! �tempfileZNamedTemporaryFiler) �IOErrorr" r# r$ r+ r9 )r r, �fr r r �test_register_file� s z(EpollSelectorTestCase.test_register_file)rg rh ri 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 QX| j | j j
t
j � | j t � � | j
| � Wd QXd S)N)r! r Z make_bad_fdr) r
r" r# r$ r& Z exceptionrt ZEBADFr+ r9 )r r, Zbad_f�cmr r r �test_register_bad_fd� s z+KqueueSelectorTestCase.test_register_bad_fd)rg rh ri r{ r# r! r� r r r r r� � s r� c C s0 t t t t t g } t j | � t j � d S)N)rw rx rz r} r� r Z run_unittestZ
reap_children)Ztestsr r r � test_main� s
r� �__main__)!rt r5 rU r# rd r rl r~ Ztestr r r r; Z
unittest.mockr � ImportErrorrr rP r ZAF_INETZ SOCK_STREAMr ZTestCaser rn rw rx rj rz r} r� r� rg r r r r �<module> sP
� J.