U Jë]jÃ+ã@s¸UdZddlZddlZddlZddlmZeeed<zddlZWne k rZdZ YnXddlmZdZ 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úpc:\users\psant\downloads\hotstarbackend\school\venv\Lib\site-packages\twisted/internet/test/test_posixprocess.pyÚ__init__ szFakeFile.__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 rrrrrrrrs rc@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 8szFakeResourceModule.__init__cCs||jkrd|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.src@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.getpidcCsTtt|jƒ}|jr(|d| ¡fkr(|S|jrJ|dkrJ|jr@|SdddgStƒ‚dS)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Ú2N)ÚmapÚstrrÚprocfsr%ÚdevfsÚaccurateDevFDResultsÚOSError)r ÚargZaccuraterrrÚlistdir`s zFDDetectorTests.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:|dkr,ztjd=Wq6tk r(Yq6Xn |tjd<dS)zI Restore the original resource module to L{sys.modules}. Nr:)r;r<ÚKeyError)r ÚvaluerrrÚreplaceResourceModulesz%FDDetectorTests.replaceResourceModulecCsNt ¡|_|j|j_|j|j_|j|j_dddg|_| |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Údetectorr0r%r9rZ addCleanuprAr;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ƒ‚dS)NzThis does not work)Ú ValueErrorrrrrÚfailWithException®szBFDDetectorTests.test_selectFirstWorking..failWithExceptioncSs dddgS©NrrrBrrrrrÚfailWithWrongResults±szEFDDetectorTests.test_selectFirstWorking..failWithWrongResultscsˆjdd…Sr)rrrrrÚcorrect´sz8FDDetectorTests.test_selectFirstWorking..correctN©rDZ_implementationsZassertIsZ_getImplementation)r rHrJrKrrrÚ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. cSs dddgS)Nééé rrrrrrJÆsz=FDDetectorTests.test_selectLast..failWithWrongResultscSs dddgSrIrrrrrÚfailWithOtherWrongResultsÉszBFDDetectorTests.test_selectLast..failWithOtherWrongResultsNrL)r rJrQrrrÚ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)rrCÚ_listOpenFDsrZassertNotEqualÚassertEqual)r rDÚfirstÚsecondÚthirdrrrÚ!test_identityOfListOpenFDsChangesÓsz1FDDetectorTests.test_identityOfListOpenFDsChangescCs>d|_| t|jj¡d|_d|_| dddg|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. FTrrrBN)r,ÚassertRaisesr.rDZ_devFDImplementationr-rTrrrrÚtest_devFDImplementationæs z(FDDetectorTests.test_devFDImplementationcCs8d|_| t|jj¡d|_| dddg|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. FTrrrBN)r+rYr.rDZ_procFDImplementationrTrrrrÚ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. iir1N)r>rTÚlistr4rDÚ_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. r1N)r=rTr\r4rDr]rrrrÚtest_fallbackFDImplementations ÿz-FDDetectorTests.test_fallbackFDImplementationN)rrrrrÚskipr,r-r+r%r0r9r=r>rArFrMrRrXrZr[r^r_rrrrr$Ds&r$c@s$eZdZdZeZdd„Zdd„ZdS)ÚFileDescriptorTestsz< Tests for L{twisted.internet.process._listOpenFDs} cCsdt ¡D]V}zt |tj¡Wqtk r\}z | tj|jd||jf¡W5d}~XYqXqdS)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)rrSÚfcntlZF_GETFLr.rTÚerrnoÚEBADF)r r ÚerrrrrÚtest_openFDs"sÿýz FileDescriptorTests.test_openFDsc Cs‚ttjƒ}t ¡}| | ¡|¡t | ¡¡}| || ¡kd¡z| ¡| |t ¡¡W5t |¡X| |t ¡¡dS)zB L{_listOpenFDs} lists expected file descriptors. z>Expected duplicate file descriptor to be greater than originalN)ÚopenÚosÚdevnullrrSZassertInÚfilenoÚdupÚ assertTruerZassertNotIn)r r8Zopenfdsr rrrÚtest_expectedFDs4s þz$FileDescriptorTests.test_expectedFDsN)rrrrrr`rfrmrrrrrasra)rrcrhr;Útypingrr*Ú__annotations__rbÚImportErrorrZtwisted.internetrZtwisted.trial.unittestrrrr$rarrrrÚs" X