o ¯bã@sždZddlZddlZddlZzddlZWn eydZYnwddlmZddl 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 rrÚ 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.setUpcCsLzt |j¡Wn tyYnwz t |j¡WdSty%YdSw)z Close pipes. N)rr rÚOSErrorrrrrrÚtearDown@s ÿ ÿzReadWriteTests.tearDowncCst |j|¡S)z) Write data to the pipe. )rÚ writeToFDr)rÚdrrrÚwriteMszReadWriteTests.writecCs0g}t |j|j¡}|dur|r|dSdS|S)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|dkr9| | ¡¡|t|dƒ7}|d7}||ks|dksd |¡}| 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ÚorigÚwrittenÚresultÚ 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+Ú assertIsNone)rr(r6rrrÚtest_readFromEmpty€s z!ReadWriteTests.test_readFromEmptycCó"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*rÚCONNECTION_DONErrrrÚtest_readFromCleanCloseŠó z&ReadWriteTests.test_readFromCleanClosecCó$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’ó z!ReadWriteTests.test_writeToClosedcCr<)zŠ Verify that reading with L{fdesc.readFromFD} when the read end is closed results in a connection lost indicator. N)rr rr+r*rrBrrrrÚtest_readFromInvalidšr?z#ReadWriteTests.test_readFromInvalidcCr@)zŠ Verify that writing with L{fdesc.writeToFD} when the write end is closed results in a connection lost indicator. rAN)rr rr+r$rrBrrrrÚtest_writeToInvalid¢rDz"ReadWriteTests.test_writeToInvalidcCsptj}dd„}|t_z| | d¡d¡W|t_n|t_wdd„}|t_z| | d¡d¡W|t_dS|t_w)z8 Test error path for L{fdesc.writeTod}. cSótƒ}tj|_|‚©N)r ÚerrnoÚEAGAIN©ÚfdÚdataÚerrrrrÚ eagainWrite°óz4ReadWriteTests.test_writeErrors..eagainWriterArcSrGrH)r rIÚEINTRrKrrrÚ eintrWrite»rPz3ReadWriteTests.test_writeErrors..eintrWriteN)rr$r+)rÚ oldOsWriterOrRrrrÚtest_writeErrorsªszReadWriteTests.test_writeErrorsN)rrrrrr!r$r*r/r9r;r>rCrErFrTrrrrr3s    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) cCszt ¡}|dkr4zt tjtjd|j| ¡fg¡WdSty3ddl}|  ¡t  d¡YdSwt tj |dƒdS)Nrz-cér2) rÚforkÚexecvÚsysÚ executableÚprogramÚfilenoÚ BaseExceptionÚ tracebackÚ print_excÚ_exitrÚwaitpid)rÚfObjÚpidr^rrrÚ_execWithFileDescriptorÚs þ üz(CloseOnExecTests._execWithFileDescriptorcCsnt| ¡dƒ&}t | ¡¡| |¡}| t |¡¡|  t  |¡d¡WdƒdS1s0wYdS)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Ú_setCloseOnExecr\rdrrÚ WIFEXITEDr+Ú WEXITSTATUS©rrbÚstatusrrrÚtest_setCloseOnExecìs  "üz$CloseOnExecTests.test_setCloseOnExeccCs|t| ¡dƒ-}t | ¡¡t | ¡¡| |¡}| t  |¡¡|  t  |¡d¡WdƒdS1s7wYdS)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. reéN) rfrgrrhr\Ú_unsetCloseOnExecrdrrrir+rjrkrrrÚtest_unsetCloseOnExecøs "ûz&CloseOnExecTests.test_unsetCloseOnExecN)rrrrr[rdrmrprrrrrUÇs  rU)rrIrrYr Ú ImportErrorÚskipÚtwisted.internetrÚtwisted.python.utilrÚ twisted.trialrÚSynchronousTestCaserrrUrrrrÚs   ÿ