U  W[2@sdZddlmZddlZddlmZmZddlm Z ddl m Z ddl m Z ddlmZdd lmZd d lmZd d lmZmZmZmZmZmZmZddlZddlmZddZGdddejj j!Z"Gdddejj j!Z#e eGddde$Z%dS)z3 Tests for L{twisted.application.runner._pidfile}. wrapsN)getpidname)BytesIO) implementer) verifyObject) IFilePath)platform)_pidfile)IPIDFilePIDFile NonePIDFileAlreadyRunningErrorInvalidPIDFileErrorStalePIDFileError NoPIDFound)SkipTestcstfdd}|S)aQ Decorator for tests that are not expected to work on all platforms. Calling L{PIDFile.isRunning} currently raises L{NotImplementedError} on non-POSIX platforms. On an unsupported platform, we expect to see any test that calls L{PIDFile.isRunning} to raise either L{NotImplementedError}, L{SkipTest}, or C{self.failureException}. (C{self.failureException} may occur in a test that checks for a specific exception but it gets NotImplementedError instead.) @param f: The test method to decorate. @type f: method @return: The wrapped callable. cs`tdk}|r |f||S|jtt|jf|f||}t|tr\|t| ddS)Nposixz isRunning is not implemented on ) r ZgetType assertRaisesNotImplementedErrorrZfailureException isinstance assertTruestr startswith)selfargskwargsZ supportedefN/usr/lib/python3/dist-packages/twisted/application/runner/test/test_pidfile.pywrapper0s     z$ifPlatformSupported..wrapperr)r"r%r#r!r$ifPlatformSupportedsr&c@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZeddZeddZeddZeddZed d!Zed"d#Zd$d%Zd&d'Zed(d)Zed*d+Zd,S)- PIDFileTestsz Tests for L{PIDFile}. cCstt}tt|dS)z5 L{PIDFile} conforms to L{IPIDFile}. N)r DummyFilePathrrrpidFiler#r#r$test_interfaceIs zPIDFileTests.test_interfacecCs|tjddddS)zR L{PIDFile._format} returns the expected format when given a PID. 9pids1337 N) assertEqualr_formatrr#r#r$test_formatWithPIDQszPIDFileTests.test_formatWithPIDcCs,d}tttj|d}|||dS)zK L{PIDFile.read} returns the PID from the given file path. r,r-N)rr(r0r/readrr.r*r#r#r$test_readWithPIDXszPIDFileTests.test_readWithPIDcCs8d}ttd}|t|j}|t|d|dS)f L{PIDFile.read} raises L{InvalidPIDFileError} when given an empty file path. 'non-integer PID value in PID file: {!r}Nrr(rrr3r/rformatrZpidValuer*r r#r#r$test_readEmptyPIDcs zPIDFileTests.test_readEmptyPIDcCs8d}tt|}|t|j}|t|d|dS)r6s$foo!r8Nr9r;r#r#r$test_readWithBogusPIDrs z"PIDFileTests.test_readWithBogusPIDcCs,tt}|t|j}|t|ddS)zc L{PIDFile.read} raises L{NoPIDFound} when given a non-existing file path. PID file does not existN)rr(rrr3r/rrr*r r#r#r$test_readDoesntExists z!PIDFileTests.test_readDoesntExistcCsDddd}|td|tt}|t|j}||jtjdS)z L{PIDFile.read} re-raises L{OSError} if the associated C{errno} is anything other than L{errno.ENOENT}. rcSsttjddS)Nz I/O error)OSErrorerrnoEIO)moder#r#r$oopssz>PIDFileTests.test_readOpenRaisesOSErrorNotENOENT..oopsopenN)rA) patchr(rrrBr3r/rCrD)rrFr*errorr#r#r$#test_readOpenRaisesOSErrorNotENOENTs   z0PIDFileTests.test_readOpenRaisesOSErrorNotENOENTcCs,d}tt}|||||dS)z9 L{PIDFile._write} stores the given PID. iN)rr(_writer/r3r4r#r#r$ test_writePIDs  zPIDFileTests.test_writePIDcCstt}|t|jddS)zS L{PIDFile._write} raises L{ValueError} when given an invalid PID. ZburpN)rr(r ValueErrorrKr)r#r#r$test_writePIDInvalids z!PIDFileTests.test_writePIDInvalidcCs(tt}|||tdS)zT L{PIDFile.writeRunningPID} stores the PID for the current process. N)rr(writeRunningPIDr/r3rr)r#r#r$test_writeRunningPIDs z!PIDFileTests.test_writeRunningPIDcCs8ttd}||j|||jdS)z9 L{PIDFile.remove} removes the PID file. r7N)rr(rfilePathexistsremove assertFalser)r#r#r$ test_removes zPIDFileTests.test_removecCs<tt}|ddd}|td|||dS)zR L{PIDFile.isRunning} returns true for a process that does exist. r,cSsdSNr#r.signalr#r#r$killsz2PIDFileTests.test_isRunningDoesExist..killrYNrr(rKrHr r isRunningrr*rYr#r#r$test_isRunningDoesExists   z$PIDFileTests.test_isRunningDoesExistcCs$tt}|||dS)a@ L{PIDFile.isRunning} returns true for this process (which is running). @note: This differs from L{PIDFileTests.test_isRunningDoesExist} in that it actually invokes the C{kill} system call, which is useful for testing of our chosen method for probing the existence of a process. N)rr(rOrr[r)r#r#r$test_isRunningThiss zPIDFileTests.test_isRunningThiscCs<tt}|ddd}|td||t|jdS)z{ L{PIDFile.isRunning} raises L{StalePIDFileError} for a process that does not exist (errno=ESRCH). r,cSsttjddSNzNo such processrBrCZESRCHrWr#r#r$rYsz5PIDFileTests.test_isRunningDoesNotExist..killrYN)rr(rKrHr rrr[r\r#r#r$test_isRunningDoesNotExists   z'PIDFileTests.test_isRunningDoesNotExistcCs<tt}|ddd}|td|||dS)zx L{PIDFile.isRunning} returns true for a process that we are not allowed to kill (errno=EPERM). r,cSsttjddS)NzOperation not permitted)rBrCEPERMrWr#r#r$rYsz3PIDFileTests.test_isRunningNotAllowed..killrYNrZr\r#r#r$test_isRunningNotAlloweds   z%PIDFileTests.test_isRunningNotAllowedcCs6tdkrtdtt}|d||dS)ac L{PIDFile.isRunning} returns true for a process that we are not allowed to kill (errno=EPERM). @note: This differs from L{PIDFileTests.test_isRunningNotAllowed} in that it actually invokes the C{kill} system call, which is useful for testing of our chosen method for probing the existence of a process that we are not allowed to kill. @note: In this case, we try killing C{init}, which is process #1 on POSIX systems, so this test is not portable. C{init} should always be running and should not be killable by non-root users. rzThis test assumes POSIXN) SYSTEM_NAMErrr(rKrr[r)r#r#r$test_isRunningInits   zPIDFileTests.test_isRunningInitcCs:tt}|dd}|td||t|jdS)z L{PIDFile.isRunning} re-raises L{OSError} if the attached C{errno} value from L{os.kill} is not an expected one. cSsttjddS)Nz File exists)rBrCZEEXISTrWr#r#r$rY(sz5PIDFileTests.test_isRunningUnknownErrno..killrYN)rr(rOrHr rrBr[r\r#r#r$test_isRunningUnknownErrnos  z'PIDFileTests.test_isRunningUnknownErrnocCstt}||dS)zS L{PIDFile.isRunning} returns false if the PID file doesn't exist. N)rr(rTr[r)r#r#r$test_isRunningNoPIDFile0s z$PIDFileTests.test_isRunningNoPIDFilec Cs`tt}||j|(||j||tW5QRX||jdS)z When used as a context manager, a L{PIDFile} will store the current pid on entry, then removes the PID file on exit. N) rr(rTrQrRrr/r3rr)r#r#r$test_contextManager9s  z PIDFileTests.test_contextManagerc Csntt}|ddd}|td||t|j}|t |d||| t W5QRXdS)z When used as a context manager, a L{PIDFile} will replace the underlying PIDFile rather than raising L{AlreadyRunningError} if the contained PID file exists but refers to a non-running PID. r,cSsttjddSr_r`rWr#r#r$rYRsz9PIDFileTests.test_contextManagerDoesntExist..killrYz'PID file refers to non-existing processN) rr(rKrHr rrr[r/rr3r)rr*rYr r#r#r$test_contextManagerDoesntExistHs  z+PIDFileTests.test_contextManagerDoesntExistcCsJtt}|ddd}|td||||t|j dS)z When used as a context manager, a L{PIDFile} will raise L{AlreadyRunningError} if the there is already a running process with the contained PID. r,cSsdSrVr#rWr#r#r$rYhsz.killrYN) rr(rKrHr rr[rr __enter__r\r#r#r$!test_contextManagerAlreadyRunning^s   z.PIDFileTests.test_contextManagerAlreadyRunningN)__name__ __module__ __qualname____doc__r+r2r5r<r=r@rJrLrNrPrUr&r]r^rarcrfrgrhrirjrlr#r#r#r$r'Ds<              r'c@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS)NonePIDFileTestsz# Tests for L{NonePIDFile}. cCst}tt|dS)z9 L{NonePIDFile} conforms to L{IPIDFile}. N)rrrr)r#r#r$r+xszNonePIDFileTests.test_interfacecCs(t}|t|j}|t|ddS)z; L{NonePIDFile.read} raises L{NoPIDFound}. r>N)rrrr3r/rr?r#r#r$ test_readszNonePIDFileTests.test_readcCs*t}|t|jd}||jtjdS)zZ L{NonePIDFile._write} raises L{OSError} with an errno of L{errno.EPERM}. rN)rrrBrKr/rCrbrr*rIr#r#r$ test_writeszNonePIDFileTests.test_writecCs(t}|t|j}||jtjdS)zk L{NonePIDFile.writeRunningPID} raises L{OSError} with an errno of L{errno.EPERM}. N)rrrBrOr/rCrbrsr#r#r$rPsz%NonePIDFileTests.test_writeRunningPIDcCs(t}|t|j}||jtjdS)zZ L{NonePIDFile.remove} raises L{OSError} with an errno of L{errno.EPERM}. N)rrrBrSr/rCENOENTrsr#r#r$rUszNonePIDFileTests.test_removecCst}||ddS)z< L{NonePIDFile.isRunning} returns L{False}. FN)rr/r[r)r#r#r$test_isRunningszNonePIDFileTests.test_isRunningc Cst}|W5QRXdS)zo When used as a context manager, a L{NonePIDFile} doesn't raise, despite not existing. N)rr)r#r#r$risz$NonePIDFileTests.test_contextManagerN) rmrnrorpr+rrrtrPrUrvrir#r#r#r$rqss     rqc@sDeZdZdZdddZdddZdd Zd d Zd d ZddZ dS)r(z! In-memory L{IFilePath}. NcCs||dSrV setContentrZcontentr#r#r$__init__szDummyFilePath.__init__rAcCs|jsttjdt|S)NzNo such file or directory)_existsrBrCrur getContent)rrEr#r#r$rGs zDummyFilePath.opencCs|dk |_||_dSrV)r{_contentryr#r#r$rxs zDummyFilePath.setContentcCs|jSrV)r}r1r#r#r$r|szDummyFilePath.getContentcCs|ddSrVrwr1r#r#r$rSszDummyFilePath.removecCs|jSrV)r{r1r#r#r$rRszDummyFilePath.exists)N)rA) rmrnrorprzrGrxr|rSrRr#r#r#r$r(s  r()&rp functoolsrrCosrrreiorZzope.interfacerZzope.interface.verifyrZtwisted.python.filepathr Ztwisted.python.runtimer Zrunnerr rrrrrrrZtwisted.trial.unittestZtwistedrr&ZtrialZunittestZTestCaser'rqobjectr(r#r#r#r$s&       $ &1K