U ¹êW[Ëã@s dZddlZddlZddlZz ddlZWnek r@dZYnXddlmZddl m Z ddl m Z Gdd„de j ƒZGd d „d e j ƒZGd d „d e j ƒZdS) z& Tests for L{twisted.internet.fdesc}. éNznot supported on this platform)Úfdesc)ÚuntilConcludes)Úunittestc@s eZdZdZdd„Zdd„ZdS)ÚNonBlockingTestszE Tests for L{fdesc.setNonBlocking} and L{fdesc.setBlocking}. cCsjt ¡\}}| tj|¡| tj|¡| t |tj¡tj@¡t  |¡|  t |tj¡tj@¡dS)zR L{fdesc.setNonBlocking} sets a file description to non-blocking. N) ÚosÚpipeÚ addCleanupÚcloseÚ assertFalseÚfcntlÚF_GETFLÚ O_NONBLOCKrÚsetNonBlockingÚ assertTrue©ÚselfÚrÚw©rú9/usr/lib/python3/dist-packages/twisted/test/test_fdesc.pyÚtest_setNonBlockings   z$NonBlockingTests.test_setNonBlockingcCsZt ¡\}}| tj|¡| tj|¡t |¡t |¡| t |tj ¡tj @¡dS)zK L{fdesc.setBlocking} sets a file description to blocking. N) rrrr rrZ setBlockingr r r r rrrrÚtest_setBlocking(s    z!NonBlockingTests.test_setBlockingN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrrrrs rc@speZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„ZdS)ÚReadWriteTestsz< Tests for L{fdesc.readFromFD}, L{fdesc.writeToFD}. cCs,t ¡\|_|_t |j¡t |j¡dS)zG Create a non-blocking pipe that can be used in tests. N)rrrrrr©rrrrÚsetUp:s zReadWriteTests.setUpcCsPzt |j¡Wntk r$YnXzt |j¡Wntk rJYnXdS)z Close pipes. N)rr rÚOSErrorrrrrrÚtearDownCszReadWriteTests.tearDowncCst |j|¡S)z) Write data to the pipe. )rZ writeToFDr)rÚdrrrÚwriteQszReadWriteTests.writecCs6g}t |j|j¡}|dkr.|r(|dSdSn|SdS)z* Read data from the pipe. Nró)rÚ readFromFDrÚappend)rÚlÚresrrrÚreadXszReadWriteTests.readcCsH| d¡}| |dk¡| ¡}| t|ƒ|¡| dd|…|¡dS)z” Test that the number of bytes L{fdesc.writeToFD} reports as written with its return value are seen by L{fdesc.readFromFD}. shellorN)r"rr(Ú assertEqualÚlen)rÚnÚsrrrÚtest_writeAndReadgs  z ReadWriteTests.test_writeAndReadcCs–dd}| |¡}| |dk¡g}d}d}||ks<|dkrd| | ¡¡|t|dƒ7}|d7}q,d |¡}| t|ƒ|¡| |d|…|¡dS) z} Similar to L{test_writeAndRead}, but use a much larger string to verify the behavior for that case. s 0123456879i'ré2éÿÿÿÿér#N)r"rr%r(r*Újoinr))rZorigZwrittenÚresultZ resultlengthÚirrrÚtest_writeAndReadLargess   z%ReadWriteTests.test_writeAndReadLargecCs.g}t |j|j¡}| |g¡| |¡dS)z¨ Verify that reading from a file descriptor with no data does not raise an exception and does not result in the callback function being called. N)rr$rr%r)Z assertIsNone)rr&r2rrrÚtest_readFromEmptyˆs z!ReadWriteTests.test_readFromEmptycCs"t |j¡| | ¡tj¡dS)z† Test that using L{fdesc.readFromFD} on a cleanly closed file descriptor returns a connection done indicator. N)rr rr)r(rZCONNECTION_DONErrrrÚtest_readFromCleanClose“s z&ReadWriteTests.test_readFromCleanClosecCs$t |j¡| | d¡tj¡dS)z‰ Verify that writing with L{fdesc.writeToFD} when the read end is closed results in a connection lost indicator. ósN)rr rr)r"rÚCONNECTION_LOSTrrrrÚtest_writeToClosedœs z!ReadWriteTests.test_writeToClosedcCs"t |j¡| | ¡tj¡dS)zŠ Verify that reading with L{fdesc.readFromFD} when the read end is closed results in a connection lost indicator. N)rr rr)r(rr8rrrrÚtest_readFromInvalid¥s z#ReadWriteTests.test_readFromInvalidcCs$t |j¡| | d¡tj¡dS)zŠ Verify that writing with L{fdesc.writeToFD} when the write end is closed results in a connection lost indicator. r7N)rr rr)r"rr8rrrrÚtest_writeToInvalid®s z"ReadWriteTests.test_writeToInvalidcCsftj}dd„}|t_z| | d¡d¡W5|t_Xdd„}|t_z| | d¡d¡W5|t_XdS)z8 Test error path for L{fdesc.writeTod}. cSstƒ}tj|_|‚dS©N)rÚerrnoZEAGAIN©ÚfdÚdataÚerrrrrÚ eagainWrite¼sz4ReadWriteTests.test_writeErrors..eagainWriter7rcSstƒ}tj|_|‚dSr<)rr=ZEINTRr>rrrÚ eintrWriteÆsz3ReadWriteTests.test_writeErrors..eintrWriteN)rr"r))rZ oldOsWriterBrCrrrÚtest_writeErrors·szReadWriteTests.test_writeErrorsN)rrrrrr r"r(r-r4r5r6r9r:r;rDrrrrr5s       rc@s,eZdZdZdZdd„Zdd„Zdd„Zd S) ÚCloseOnExecTestszL Tests for L{fdesc._setCloseOnExec} and L{fdesc._unsetCloseOnExec}. z¸ import os, errno try: os.write(%d, b'lul') except OSError as e: if e.errno == errno.EBADF: os._exit(0) os._exit(5) except: os._exit(10) else: os._exit(20) cCsvt ¡}|dkr`z&t tjtjd|j| ¡fg¡Wqrddl}| ¡t  d¡YqrXnt tj |dƒdSdS)Nrz-cér0) rÚforkÚexecvÚsysÚ executableÚprogramÚfilenoÚ tracebackÚ print_excÚ_exitrÚwaitpid)rÚfObjÚpidrMrrrÚ_execWithFileDescriptoräs&z(CloseOnExecTests._execWithFileDescriptorc CsXt| ¡dƒ@}t | ¡¡| |¡}| t |¡¡|  t  |¡d¡W5QRXdS)z° A file descriptor passed to L{fdesc._setCloseOnExec} is not inherited by a new process image created with one of the exec family of functions. ÚwbrN) ÚopenÚmktemprÚ_setCloseOnExecrLrSrrÚ WIFEXITEDr)Ú WEXITSTATUS©rrQZstatusrrrÚtest_setCloseOnExecós  z$CloseOnExecTests.test_setCloseOnExecc Csft| ¡dƒN}t | ¡¡t | ¡¡| |¡}| t  |¡¡|  t  |¡d¡W5QRXdS)z¦ A file descriptor passed to L{fdesc._unsetCloseOnExec} is inherited by a new process image created with one of the exec family of functions. rTéN) rUrVrrWrLZ_unsetCloseOnExecrSrrrXr)rYrZrrrÚtest_unsetCloseOnExecs  z&CloseOnExecTests.test_unsetCloseOnExecN)rrrrrKrSr[r]rrrrrEÒs  rE)rrrIr=r Ú ImportErrorÚskipZtwisted.internetrZtwisted.python.utilrZ twisted.trialrZSynchronousTestCaserrrErrrrÚs