tf+UdZddlZddlZddlZddlmZeeed< ddlZddl m Z dZ n #e $rdZ YnwxYwddl mZGdd ZGd d ZGd d eZGddeZdS)z; Tests for POSIX-based L{IReactorProcess} implementations. N)Optional platformSkip)processznon-POSIX platform)TestCasec*eZdZdZdZdZdZdZdS)FakeFilez> A dummy file object which records when it is closed. c"||_||_dSN)testcasefd)selfr r s f/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/twisted/internet/test/test_posixprocess.py__init__zFakeFile.__init__ s  cN|jj|jdSr )r _filesremover r s rclosezFakeFile.close$s# ##DG,,,,,rc|Sr rs r __enter__zFakeFile.__enter__'s rc.|dSr )r)r exc_type exc_value tracebacks r__exit__zFakeFile.__exit__*s rN)__name__ __module__ __qualname____doc__rrrrrrrrrsZ---rrc"eZdZdZdZdZdZdS)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. c||_dSr )_limitr limits rrzFakeResourceModule.__init__8s  rc2||jkr d|jgSddgS)zX A fake of L{resource.getrlimit} which returns a pre-determined result. r{i) RLIMIT_NOFILEr&)r nos r getrlimitzFakeResourceModule.getrlimit;s) # # #t{# #SzrN)rrr r!r+rr-rrrr#r#.sCMrr#cveZdZdZeZdZdZdZdZ dZ dZ dZ dZ dZd Zd Zd Zd Zd ZdZdZdZdS)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. FcdS)z> Fake os.getpid, always return the same thing r*rrs rgetpidzFDDetectorTests.getpidZs srctt|j}|jr|d|fzkr|S|jr|dkr |jr|SgdSt)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)012)mapstrrprocfsr1devfsaccurateDevFDResultsOSError)r argaccurates rlistdirzFDDetectorTests.listdir`su sDK(( ; 3=DKKMM3C#CDDO : ##**( "?? "iirc t|tttdt|jz }|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. )rminsetrangerappendr )r fnamemodefs ropenfilezFDDetectorTests.openfileosU T3s5;;//#dk2B2BBCC D D 14   rc$dtjd<dS)zp Make the L{resource} module unimportable for the remainder of the current test method. Nresource)sysmodulesrs rhideResourceModulez"FDDetectorTests.hideResourceModule}s #' Jrc>t|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. rJN)r#rKrLr's rrevealResourceModulez$FDDetectorTests.revealResourceModules#5U";"; Jrcj|! tjd=dS#t$rYdSwxYw|tjd<dS)zI Restore the original resource module to L{sys.modules}. NrJ)rKrLKeyError)r values rreplaceResourceModulez%FDDetectorTests.replaceResourceModulesT = K +++    ',CK # # #s  !!ctj|_|j|j_|j|j_|j|j_gd|_||jtj ddS)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$rJN) r _FDDetectordetectorr>r1rHr addCleanuprSrKrLgetrs rsetUpzFDDetectorTests.setUpsn  +-- $  #{ !% ii  2CKOOJ4O4OPPPPPrcd}d}fd}|||gj_|jdS)z L{FDDetector._getImplementation} returns the first method from its C{_implementations} list which returns results which reflect a newly opened file descriptor. c td)NzThis does not work) ValueErrorrrrfailWithExceptionzBFDDetectorTests.test_selectFirstWorking..failWithExceptions122 2rc gdSNrUrrrrfailWithWrongResultszEFDDetectorTests.test_selectFirstWorking..failWithWrongResults 99 rc"jddSr )rrsrcorrectz8FDDetectorTests.test_selectFirstWorking..corrects;qqq> !rNrX_implementationsassertIs_getImplementation)r r_rbres` rtest_selectFirstWorkingz'FDDetectorTests.test_selectFirstWorkings{ 3 3 3    " " " " "  *  & gt}??AABBBBBrcd}d}||g|j_|||jdS)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. c gdS)N) rrrrrbz=FDDetectorTests.test_selectLast..failWithWrongResultsrcrc gdSrarrrrfailWithOtherWrongResultszBFDDetectorTests.test_selectLast..failWithOtherWrongResultsrcrNrf)r rbrqs rtest_selectLastzFDDetectorTests.test_selectLasts_       ! %*  & /1Q1Q1S1STTTTTrctj}|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)rrW _listOpenFDsrassertNotEqual assertEqual)r rXfirstsecondthirds r!test_identityOfListOpenFDsChangesz1FDDetectorTests.test_identityOfListOpenFDsChangess&((%.&/%. E6*** '''''rcd|_|t|jjd|_d|_|gd|jdS)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. FTrUN)r9 assertRaisesr;rX_devFDImplementationr:rvrs rtest_devFDImplementationz(FDDetectorTests.test_devFDImplementationsc   '4=#EFFF $)! DM$F$F$H$HIIIIIrcd|_|t|jjd|_|gd|jdS)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. FTrUN)r8r|r;rX_procFDImplementationrvrs rtest_procFDImplementationz)FDDetectorTests.test_procFDImplementations[   '4=#FGGG  DM$G$G$I$IJJJJJrc|d|ttdt|j|d|ttdt|jdS)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. iir@N)rOrvlistrCrX_fallbackFDImplementationrs rtest_resourceFDImplementationz-FDDetectorTests.test_resourceFDImplementations !!#&&&  s  d4=#J#J#L#LMM   !!$'''  t  tDM$K$K$M$MNN     rc||ttdt|jdS)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. r@N)rMrvrrCrXrrs rtest_fallbackFDImplementationz-FDDetectorTests.test_fallbackFDImplementationsa !!!  t  tDM$K$K$M$MNN     rN)rrr r!rskipr9r:r8r1r>rHrMrOrSr[rjrrrzr~rrrrrrr/r/Ds   D E  F      ''' < < < , , , Q Q QCCC0UUU((((& J J J K K K   $      rr/c"eZdZdZeZdZdZdS)FileDescriptorTestsz< Tests for L{twisted.internet.process._listOpenFDs} c tjD]i} tj|tj##t$r:}|t j|jd||jfzYd}~bd}~wwxYwdS)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)rrtfcntlF_GETFLr;rverrnoEBADF)r r errs r test_openFDsz FileDescriptorTests.test_openFDs"s&((  B  B ....     KIH39o&   s7 A;0A66A;ctttj}tj}|||tj|}|||kd | ||tjtj |n#tj |wxYw| |tjdS)zB L{_listOpenFDs} lists expected file descriptors. z>Expected duplicate file descriptor to be greater than originalN) openosdevnullrrtassertInfilenodup assertTruer assertNotIn)r rGopenfdsr s rtest_expectedFDsz$FileDescriptorTests.test_expectedFDs4s   &(( ahhjj'***VAHHJJ    O L     GGIII MM"g244 5 5 5 HRLLLLBHRLLLL W13344444s (;C88DN)rrr r!rrrrrrrrrsC D$)5)5)5)5)5rr)r!rrrKtypingrr7__annotations__rtwisted.internetrr ImportErrortwisted.trial.unittestrrr#r/rrrrrsw  smLLL)(((((LL ((('LLL(,+++++&,T T T T T hT T T nB5B5B5B5B5(B5B5B5B5B5s /99