a zeH@sddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl Z ddlZ ddlZddl mZ z ddlZWneydZYn0eedrejZnejejdfddZdd ZGd d d ZGd d d ZGdddee jZGdddee jZe eeddGdddeee jZe eeddGdddeee jZe eeddGdddeee jZ e eeddGd d!d!eee jZ!d"d#Z"e#d$kre $dS)%N)support) socket_helper)sleep) monotonic socketpairc Cst|||}|tjdf|t|||}zP|||}|\}}||krz||fWWdS|qNWnt y|Yn0Wdn1s0YdS)Nr) socketbindrZHOSTlistenconnect getsocknameacceptcloseOSError)familytypeprotolcZcaddraaddrrG/opt/bitninja-python-dojo/embedded/lib/python3.9/test/test_selectors.pyrs  cCs*g}|D]\}}||@r||jq|SN)appendfileobj)readyflagmatchkeyeventsrrrfind_ready_matching+s  r c@seZdZddZddZddZddZee j d kd d d Z d dZ ddZ ddZddZddZddZddZddZddZdd Zeejd!kd"d#d$Zd%d&Zeeed'd(d)d*Zeeed'd(d+d,Zd-S).BaseSelectorTestCasecCs*t\}}||j||j||fSr)r addCleanupr )selfrdwrrrrmake_socketpair5s   z$BaseSelectorTestCase.make_socketpaircCs|}||j|\}}||tjd}||tj| |j || |j | | |j tj| |jd|t|jdd|t|jdtj|t|j|tj|t|j| tjdS)Ndatar?Bi)SELECTORr"r r&register selectors EVENT_READZassertIsInstance SelectorKey assertEqualrfdfilenorr' assertRaises ValueErrorKeyErrorr#sr$r%rrrr test_register;s  z"BaseSelectorTestCase.test_registercCs\|}||j|\}}||tj|||t |jd|t |j|dSNr() r)r"r r&r*r+r, unregisterr1r3r#r5r$r%rrrtest_unregisterUs   z$BaseSelectorTestCase.test_unregistercCsv|}||j|\}}||}}||tj||tj||| || |dSr) r)r"r r&r0r*r+r, EVENT_WRITEr8)r#r5r$r%rwrrrtest_unregister_after_fd_closeds   z3BaseSelectorTestCase.test_unregister_after_fd_closeposixzrequires posixcCs|}||j|\}}||}}||tj||tj|\}}||t ||t |||t j||t j|| || |dSr) r)r"r r&r0r*r+r,r;osdup2r8)r#r5r$r%r<r=Zrd2Zwr2rrr(test_unregister_after_fd_close_and_reuseps    z=BaseSelectorTestCase.test_unregister_after_fd_close_and_reusecCsd|}||j|\}}||tj||tj||||||dSr) r)r"r r&r*r+r,r;r8r9rrr"test_unregister_after_socket_closes   z7BaseSelectorTestCase.test_unregister_after_socket_closec Cs6|}||j|\}}||tj}||tj}| |j |j | || || |t}t}||tj|}||tj|}| |j |j | |j|j| || || |j||t|jdtjt}tj|_tj|_ ||tj|||jj||j jdSr7)r)r"r r&r*r+r,modifyr;ZassertNotEqualrr.get_keyr8objectr'r1r3unittestmockMock assertFalseZcalled) r#r5r$r%rZkey2Zd1Zd2Zd3rrr test_modifys.     z BaseSelectorTestCase.test_modifyc Cs|jjdkrtjd}n>|jjdkr4tjd}n$|jjdkrNtjd}n |d|}tjjtd|j_ |}| |j | \}}| |tj|t|jd |t| |tjWdn1s0Y|t|jd Wdn1s0YdS) N EpollSelectorz%selectors.EpollSelector._selector_cls PollSelectorz$selectors.PollSelector._selector_clsDevpollSelectorz'selectors.DevpollSelector._selector_cls)Z side_effectr)r)__name__rGrHpatchskipTestrIZeroDivisionErrorZ return_valuerDr"r r&r*r+r,r.len_mapr1r;)r#rRmr5r$r%rrrtest_modify_unregisters2        ,z+BaseSelectorTestCase.test_modify_unregistercCs|}||j|}|\}}||tj||tj|| t |j || t |j || t |j || t |j |dSr)r)r"r get_mapr&r*r+r,r;r1 RuntimeErrorrEr3 __getitem__)r#r5mappingr$r%rrr test_closes  zBaseSelectorTestCase.test_closecCsV|}||j|\}}||tjd}||||| t |jddS)Nr'r() r)r"r r&r*r+r,r.rEr1r3r4rrr test_get_keys   z!BaseSelectorTestCase.test_get_keycCs"|}||j|\}}|}|||t|d|t|g| |t j d}| |||||||t|d|t|| g|t||g|t|dWdn1s0Y|t||=Wdn1s0YdS)Nrr'rPr()r)r"r r&rYrJr.rUlistr*r+r,ZassertInr0valuesr1r3 TypeError)r#r5r$r%keysrrrr test_get_maps"     & z!BaseSelectorTestCase.test_get_mapcCs|}||j|\}}||tj||tj}|}|D]<\}}| t |tj | || |tjtjB@qH| |tjfg|dSr)r)r"r r&r*r+r,r;select assertTrue isinstancer-rJr.)r#r5r$r%Zwr_keyresultrrrrr test_selects      z BaseSelectorTestCase.test_selectcCs|}||j|\}}|,}||tj||tjWdn1sV0Y|t |j ||t |j |dSr) r)r"r r&r*r+r,r;r1rZrE)r#r5r$r%Zselrrrtest_context_managers  ,z)BaseSelectorTestCase.test_context_managercCsF|}||jt|drB|}|t|t||ddS)Nr0r) r)r"r hasattrr0rerfintassertGreaterEqual)r#r5r/rrr test_fileno"s   z BaseSelectorTestCase.test_filenocCs|}||jd}d}t|}g}g}i}i}t|D]P} |\} } || tj|| tj | | | | | || <| || <q.InterruptSelectN)rQ __module__ __qualname__rrrrInterruptSelectsrcsdSrrargsrrrhandlersz?BaseSelectorTestCase.test_select_interrupt_exc..handlerrPg@r)r)r"r r& ExceptionsignalSIGALRMrr*r+r,rr1rdr)r#r5r$r%rorig_alrm_handlerrrrrtest_select_interrupt_excs     (z.BaseSelectorTestCase.test_select_interrupt_excc Cs|}||j|\}}ttjdd}|tjtj|zNtd||tj t }| | d| t |dWtdn td0dS)NcWsdSrrrrrrzDBaseSelectorTestCase.test_select_interrupt_noraise..rPg?g?r)r)r"r r&rrrr*r+r,rrJrdrl)r#r5r$r%rrrrrtest_select_interrupt_noraises   z2BaseSelectorTestCase.test_select_interrupt_noraiseN)rQrrr&r6r:r>rG skipUnlessr@namerBrCrKrXr]r^rcrhrirmryZskipIfsysplatformr|rrjrrrrrrrr!3s:   &   5     r!c@s,eZdZeddeedddZdS)ScalableSelectorMixInrozTest needs resource modulec Csttj\}}z6ttj||f|tjtj||ft|d}Wnttfy`|}Yn0|d8}|}||j t |dD]}z| \}}Wnty| dYn0z | |tj| |tjWqty}z&|jtjkr| dWYd}~qd}~00qz |} WnLtyx}z2|jtjkrbtjdkrb| dWYd}~n d}~00||dt| dS)Ni zFD limit reacheddarwinz%Invalid argument error calling poll())resourceZ getrlimitZ RLIMIT_NOFILEZ setrlimitr"minrr2r)r rpr&rSr*r+r,r;errnoENOSPCrdEINVALrrr.rU) r#ZsoftZhardZNUM_FDSr5rwr$r%efdsrrrtest_above_fd_setsizes>       z+ScalableSelectorMixIn.test_above_fd_setsizeN) rQrrrZrequires_mac_verrGrrrrrrrrs  rc@seZdZejZdS)DefaultSelectorTestCaseN)rQrrr+DefaultSelectorr)rrrrrsrc@seZdZejZdS)SelectSelectorTestCaseN)rQrrr+SelectSelectorr)rrrrrsrrMz!Test needs selectors.PollSelectorc@seZdZeeddZdS)PollSelectorTestCaserMNrQrrgetattrr+r)rrrrrsrrLz"Test needs selectors.EpollSelectorc@s eZdZeeddZddZdS)EpollSelectorTestCaserLNc Cs|}t|}|t||tjWdn1s@0Y|t| |Wdn1st0YWdn1s0YdSr) r)tempfileNamedTemporaryFiler1IOErrorr*r+r,r3rE)r#r5frrrtest_register_files   , z(EpollSelectorTestCase.test_register_file)rQrrrr+r)rrrrrrs rKqueueSelectorz$Test needs selectors.KqueueSelector)c@s(eZdZeeddZddZddZdS)KqueueSelectorTestCaserNcCs|}t}|t}||tjWdn1s>0Y||j j t j |t | |Wdn1s0YdSr)r)rZ make_bad_fdr1rr*r+r,r. exceptionrEBADFr3rE)r#r5Zbad_fcmrrrtest_register_bad_fds , z+KqueueSelectorTestCase.test_register_bad_fdcCs^|}||jt}||dgt}||}|d|koPdkn|dS)NrPr~r)r)r"r rr.rdre)r#r5rrrrrrtest_empty_select_timeout&s z0KqueueSelectorTestCase.test_empty_select_timeout)rQrrrr+r)rrrrrrrs  rrNz$Test needs selectors.DevpollSelectorc@seZdZeeddZdS)DevpollSelectorTestCaserNNrrrrrr4srcCs tdSr)r reap_childrenrrrrtearDownModule<sr__main__)%rr@rqr+rrrtestrZ test.supportrrrrGZ unittest.mockrrr ImportErrorrjrAF_INET SOCK_STREAMr r!rZTestCaserrrrrrrrrQmainrrrrsn         5