o ¯bÃ+ã@s¶UdZddlZddlZddlZddlmZeeed<zddlZWne y,dZ Yn wddlmZdZ ddl mZGdd „d ƒZGd d„dƒZGdd „d eƒZGdd„deƒZdS)z; Tests for POSIX-based L{IReactorProcess} implementations. éN)ÚOptionalÚplatformSkipznon-POSIX platform)Úprocess)ÚTestCasec@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S)ÚFakeFilez> A dummy file object which records when it is closed. cCs||_||_dS©N)ÚtestcaseÚfd)Úselfrr ©rúI/usr/lib/python3/dist-packages/twisted/internet/test/test_posixprocess.pyÚ__init__ s zFakeFile.__init__cCs|jj |j¡dSr)rÚ_filesÚremover ©r rrrÚclose$szFakeFile.closecCs|SrrrrrrÚ __enter__'szFakeFile.__enter__cCs| ¡dSr)r)r Úexc_typeÚ exc_valueÚ tracebackrrrÚ__exit__*szFakeFile.__exit__N)Ú__name__Ú __module__Ú__qualname__Ú__doc__r rrrrrrrrsrc@s$eZdZdZdZdd„Zdd„ZdS)ÚFakeResourceModulez¹ Fake version of L{resource} which hard-codes a particular rlimit for maximum open files. @ivar _limit: The value to return for the hard limit of number of open files. écCs ||_dSr)Ú_limit©r Úlimitrrrr 8s zFakeResourceModule.__init__cCs||jkr d|jgSddgS)zX A fake of L{resource.getrlimit} which returns a pre-determined result. ré{iÈ)Ú RLIMIT_NOFILEr)r ÚnorrrÚ getrlimit;s zFakeResourceModule.getrlimitN)rrrrr!r r#rrrrr.s rc@seZdZdZeZdZdZdZdd„Z dd„Z dd„Zd d „Zdd„Z d d„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„ZdS) ÚFDDetectorTestsaë Tests for _FDDetector class in twisted.internet.process, which detects which function to drop in place for the _listOpenFDs method. @ivar devfs: A flag indicating whether the filesystem fake will indicate that /dev/fd exists. @ivar accurateDevFDResults: A flag indicating whether the /dev/fd fake returns accurate open file information. @ivar procfs: A flag indicating whether the filesystem fake will indicate that /proc//fd exists. FcCsdS)z> Fake os.getpid, always return the same thing r rrrrrÚgetpidZszFDDetectorTests.getpidcCsNtt|jƒ}|jr|d| ¡fkr|S|jr$|dkr$|jr |Sgd¢Stƒ‚)z„ Fake os.listdir, depending on what mode we're in to simulate behaviour. @param arg: the directory to list z/proc/%d/fdz/dev/fd)Ú0Ú1Ú2)ÚmapÚstrrÚprocfsr%ÚdevfsÚaccurateDevFDResultsÚOSError)r ÚargÚaccuraterrrÚlistdir`szFDDetectorTests.listdircCs2t|tttdƒƒt|jƒƒƒ}|j |j¡|S)aO This is a mock for L{open}. It keeps track of opened files so extra descriptors can be returned from the mock for L{os.listdir} when used on one of the list-of-filedescriptors directories. A L{FakeFile} is returned which can be closed to remove the new descriptor from the open list. é)rÚminÚsetÚrangerÚappendr )r ÚfnameÚmodeÚfrrrÚopenfileos zFDDetectorTests.openfilecCsdtjd<dS)zp Make the L{resource} module unimportable for the remainder of the current test method. NÚresource)ÚsysÚmodulesrrrrÚhideResourceModule}sz"FDDetectorTests.hideResourceModulecCst|ƒtjd<dS)a Make a L{FakeResourceModule} instance importable at the L{resource} name. @param limit: The value which will be returned for the hard limit of number of open files by the fake resource module's C{getrlimit} function. r;N)rr<r=rrrrÚrevealResourceModule„s z$FDDetectorTests.revealResourceModulecCs:|durztjd=WdStyYdSw|tjd<dS)zI Restore the original resource module to L{sys.modules}. Nr;)r<r=ÚKeyError)r ÚvaluerrrÚreplaceResourceModulesÿz%FDDetectorTests.replaceResourceModulecCsLt ¡|_|j|j_|j|j_|j|j_gd¢|_| |jt j d¡¡dS)zž Set up the tests, giving ourselves a detector object to play with and setting up its testable knobs to refer to our mocked versions. ©rrér;N)rÚ_FDDetectorÚdetectorr1r%r:rÚ addCleanuprBr<r=ÚgetrrrrÚsetUp›s zFDDetectorTests.setUpcs@dd„}dd„}‡fdd„}|||gˆj_ˆ |ˆj ¡¡dS)zÁ L{FDDetector._getImplementation} returns the first method from its C{_implementations} list which returns results which reflect a newly opened file descriptor. cSstdƒ‚)NzThis does not work)Ú ValueErrorrrrrÚfailWithException®ózBFDDetectorTests.test_selectFirstWorking..failWithExceptioncSógd¢S©NrCrrrrrÚfailWithWrongResults±rLzEFDDetectorTests.test_selectFirstWorking..failWithWrongResultscsˆjdd…Sr)rrrrrÚcorrect´sz8FDDetectorTests.test_selectFirstWorking..correctN©rFÚ_implementationsÚassertIsÚ_getImplementation)r rKrOrPrrrÚtest_selectFirstWorking§sýz'FDDetectorTests.test_selectFirstWorkingcCs2dd„}dd„}||g|j_| ||j ¡¡dS)zâ L{FDDetector._getImplementation} returns the last method from its C{_implementations} list if none of the implementations manage to return results which reflect a newly opened file descriptor. cSrM)N)ééé rrrrrrOÆrLz=FDDetectorTests.test_selectLast..failWithWrongResultscSrMrNrrrrrÚfailWithOtherWrongResultsÉrLzBFDDetectorTests.test_selectLast..failWithOtherWrongResultsNrQ)r rOrYrrrÚtest_selectLast¿sþzFDDetectorTests.test_selectLastcCsLt ¡}|jj}| ¡|jj}| ¡|jj}| ||¡| ||¡dS)zä Check that the identity of _listOpenFDs changes after running _listOpenFDs the first time, but not after the second time it's run. In other words, check that the monkey patching actually works. N)rrEÚ_listOpenFDsrÚassertNotEqualÚassertEqual)r rFÚfirstÚsecondÚthirdrrrÚ!test_identityOfListOpenFDsChangesÓsz1FDDetectorTests.test_identityOfListOpenFDsChangescCs<d|_| t|jj¡d|_d|_| gd¢|j ¡¡dS)zÈ L{_FDDetector._devFDImplementation} raises L{OSError} if there is no I{/dev/fd} directory, otherwise it returns the basenames of its children interpreted as integers. FTrCN)r,ÚassertRaisesr.rFÚ_devFDImplementationr-r]rrrrÚtest_devFDImplementationæs z(FDDetectorTests.test_devFDImplementationcCs6d|_| t|jj¡d|_| gd¢|j ¡¡dS)zÐ L{_FDDetector._procFDImplementation} raises L{OSError} if there is no I{/proc//fd} directory, otherwise it returns the basenames of its children interpreted as integers. FTrCN)r+rbr.rFÚ_procFDImplementationr]rrrrÚtest_procFDImplementationòsz)FDDetectorTests.test_procFDImplementationcCsT| d¡| ttdƒƒt|j ¡ƒ¡| d¡| ttdƒƒt|j ¡ƒ¡dS)z× L{_FDDetector._fallbackFDImplementation} uses the L{resource} module if it is available, returning a range of integers from 0 to the minimum of C{1024} and the hard I{NOFILE} limit. iir2N)r?r]Úlistr5rFÚ_fallbackFDImplementationrrrrÚtest_resourceFDImplementationýs ÿ ÿz-FDDetectorTests.test_resourceFDImplementationcCs*| ¡| ttdƒƒt|j ¡ƒ¡dS)zÔ L{_FDDetector._fallbackFDImplementation}, the implementation of last resort, succeeds with a fixed range of integers from 0 to 1024 when the L{resource} module is not importable. r2N)r>r]rgr5rFrhrrrrÚtest_fallbackFDImplementationsÿz-FDDetectorTests.test_fallbackFDImplementationN)rrrrrÚskipr,r-r+r%r1r:r>r?rBrIrUrZrardrfrirjrrrrr$Ds(r$c@s$eZdZdZeZdd„Zdd„ZdS)ÚFileDescriptorTestsz< Tests for L{twisted.internet.process._listOpenFDs} cCsdt ¡D]+}z t |tj¡Wqty/}z| tj|jd||jf¡WYd}~qd}~wwdS)zµ File descriptors returned by L{_listOpenFDs} are mostly open. This test assumes that zero-legth writes fail with EBADF on closed file descriptors. z2fcntl(%d, F_GETFL) failed with unexpected errno %dN)rr[ÚfcntlÚF_GETFLr.r]ÚerrnoÚEBADF)r r ÚerrrrrÚtest_openFDs"sÿý€ÿýz FileDescriptorTests.test_openFDsc CsŒttjƒ}t ¡}| | ¡|¡t | ¡¡}| || ¡kd¡z| ¡| |t ¡¡Wt |¡nt |¡w| |t ¡¡dS)zB L{_listOpenFDs} lists expected file descriptors. z>Expected duplicate file descriptor to be greater than originalN)ÚopenÚosÚdevnullrr[ÚassertInÚfilenoÚdupÚ assertTruerÚassertNotIn)r r9Úopenfdsr rrrÚtest_expectedFDs4s þz$FileDescriptorTests.test_expectedFDsN)rrrrrrkrrr|rrrrrls rl)rrortr<Útypingrr*Ú__annotations__rmÚImportErrorrÚtwisted.internetrÚtwisted.trial.unittestrrrr$rlrrrrÚs&ÿX