U T‚gã@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ú;/tmp/pip-unpacked-wheel-1_7di8ic/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ÚsetUp8s zReadWriteTests.setUpcCsPzt |j¡Wntk r$YnXzt |j¡Wntk rJYnXdS)z Close pipes. N)rr rÚOSErrorrrrrrÚtearDown@szReadWriteTests.tearDowncCst |j|¡S)z) Write data to the pipe. )rZ writeToFDr)rÚdrrrÚwriteMszReadWriteTests.writecCs6g}t |j|j¡}|dkr.|r(|dSdSn|SdS)z* Read data from the pipe. Nró)rÚ readFromFDrÚappend)rÚlÚresrrrÚreadSszReadWriteTests.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_writeAndReadas  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))rÚorigZwrittenÚresultZ resultlengthÚirrrÚtest_writeAndReadLargels   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&r3rrrÚ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(rr9rrrrÚ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. r8N)rr rr)r"rr9rrrrÚ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ÚerrnoÚEAGAIN©ÚfdÚdataÚerrrrrÚ eagainWrite°sz4ReadWriteTests.test_writeErrors..eagainWriter8rcSstƒ}tj|_|‚dSr=)rr>ZEINTRr@rrrÚ eintrWrite»sz3ReadWriteTests.test_writeErrors..eintrWriteN)rr"r))rZ oldOsWriterDrErrrÚtest_writeErrorsªszReadWriteTests.test_writeErrorsN)rrrrrr r"r(r-r5r6r7r:r;r<rFrrrrr3s   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 BaseException: os._exit(10) else: os._exit(20) cCs~t ¡}|dkrhz&t tjtjd|j| ¡fg¡Wqztk rdddl}|  ¡t  d¡YqzXnt tj |dƒdSdS)Nrz-cér0) rÚforkÚexecvÚsysÚ executableÚprogramÚfilenoÚ BaseExceptionÚ tracebackÚ print_excÚ_exitrÚwaitpid)rÚfObjÚpidrPrrrÚ_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Ú_setCloseOnExecrNrVrrÚ WIFEXITEDr)Ú WEXITSTATUS©rrTÚstatusrrrÚ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. rWéN) rXrYrrZrNZ_unsetCloseOnExecrVrrr[r)r\r]rrrÚtest_unsetCloseOnExecøs  z&CloseOnExecTests.test_unsetCloseOnExecN)rrrrrMrVr_rarrrrrGÇs  rG)rr>rrKr Ú ImportErrorÚskipZtwisted.internetrZtwisted.python.utilrZ twisted.trialrZSynchronousTestCaserrrGrrrrÚs