U n5{ih2@sdZddlZddlmZddlmZmZddlm Z m Z m Z ddl m Z ddlmZddlZddlmZdd lmZdd lmZd d lmZd dlmZmZmZmZmZmZmZe de fe de fdddZ Gdddej!j"j#Z$Gdddej!j"j#Z%dS)z3 Tests for L{twisted.application.runner._pidfile}. N)wraps)getpidname)AnyCallableOptional) verifyObject)NoReturn)FilePath)platform)SkipTest)_pidfile)AlreadyRunningErrorInvalidPIDFileErrorIPIDFile NonePIDFile NoPIDFoundPIDFileStalePIDFileError.)freturncs$tttttdfdd }|S)a= 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. @return: The wrapped callable. )selfargskwargsrcs`tdk}|r |f||S|jtt|jf|f||}t|tr\|t| ddS)Nposixz isRunning is not implemented on ) r getType assertRaisesNotImplementedErrorr ZfailureException isinstance assertTruestr startswith)rrr supportederP/tmp/pip-unpacked-wheel-mnhg7r0q/twisted/application/runner/test/test_pidfile.pywrapper3s   z$ifPlatformSupported..wrapper)rr)rr)r'r&r(ifPlatformSupported!sr*c@speZdZdZd1eeeedddZddddZ ddd d Z ddd d Z ddd dZ ddddZ ddddZddddZddddZddddZddddZddddZeddddZedddd Zeddd!d"Zeddd#d$Zeddd%d&Zeddd'd(Zddd)d*Zddd+d,Zeddd-d.Zeddd/d0ZdS)2 PIDFileTestsz Tests for L{PIDFile}. N)contentrcCs"t|}|dk r|||SN)r mktempZ setContent)rr,filePathr'r'r(r/Ls  zPIDFileTests.filePathrcCst|}tt|dS)z5 L{PIDFile} conforms to L{IPIDFile}. N)rr/rrrpidFiler'r'r(test_interfaceRs zPIDFileTests.test_interfacecCs|tjddddS)zR L{PIDFile._format} returns the expected format when given a PID. 9pids1337 N) assertEqualr_format)rr'r'r(test_formatWithPIDYszPIDFileTests.test_formatWithPIDcCs.d}t|tj|d}|||dS)zK L{PIDFile.read} returns the PID from the given file path. r4r5N)rr/r8r7readrr6r2r'r'r(test_readWithPID_szPIDFileTests.test_readWithPIDcCs:d}t|d}|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: Nrr/rrr:r7r"rZpidValuer2r%r'r'r(test_readEmptyPIDiszPIDFileTests.test_readEmptyPIDcCs:d}t||}|t|j}|t|d|dS)r=s$foo!r?Nr@rAr'r'r(test_readWithBogusPIDtsz"PIDFileTests.test_readWithBogusPIDcCs.t|}|t|j}|t|ddS)zc L{PIDFile.read} raises L{NoPIDFound} when given a non-existing file path. PID file does not existN)rr/rrr:r7r"rr2r%r'r'r(test_readDoesntExists z!PIDFileTests.test_readDoesntExistcCsNdttddd}|td|t|}|t|j}| |j t j dS)z L{PIDFile.read} re-raises L{OSError} if the associated C{errno} is anything other than L{errno.ENOENT}. r)modercSsttjddS)Nz I/O error)OSErrorerrnoEIO)rHr'r'r(oopssz>PIDFileTests.test_readOpenRaisesOSErrorNotENOENT..oopsopenN)rG) r"r patchr rr/rrIr:r7rJrK)rrLr2errorr'r'r(#test_readOpenRaisesOSErrorNotENOENTs  z0PIDFileTests.test_readOpenRaisesOSErrorNotENOENTcCs.d}t|}|||||dS)z9 L{PIDFile._write} stores the given PID. iN)rr/_writer7r:r;r'r'r( test_writePIDs  zPIDFileTests.test_writePIDcCs t|}|t|jddS)zS L{PIDFile._write} raises L{ValueError} when given an invalid PID. ZburpN)rr/r ValueErrorrQr1r'r'r(test_writePIDInvalids z!PIDFileTests.test_writePIDInvalidcCs*t|}|||tdS)zT L{PIDFile.writeRunningPID} stores the PID for the current process. N)rr/writeRunningPIDr7r:rr1r'r'r(test_writeRunningPIDs z!PIDFileTests.test_writeRunningPIDcCs:t|d}||j|||jdS)z9 L{PIDFile.remove} removes the PID file. r>N)rr/r!existsremove assertFalser1r'r'r( test_removeszPIDFileTests.test_removecCsHt|}|dttdddd}|td|||dS)zR L{PIDFile.isRunning} returns true for a process that does exist. r4Nr6signalrcSsdSr-r'r6r\r'r'r(killsz2PIDFileTests.test_isRunningDoesExist..killr^rr/rQintrNrr! isRunningrr2r^r'r'r(test_isRunningDoesExists   z$PIDFileTests.test_isRunningDoesExistcCs&t|}|||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/rUr!rar1r'r'r(test_isRunningThiss zPIDFileTests.test_isRunningThiscCsHt|}|dttdddd}|td||t|jdS)z{ L{PIDFile.isRunning} raises L{StalePIDFileError} for a process that does not exist (errno=ESRCH). r4Nr[cSsttjddSNzNo such processrIrJZESRCHr]r'r'r(r^sz5PIDFileTests.test_isRunningDoesNotExist..killr^) rr/rQr`rNrrrrarbr'r'r(test_isRunningDoesNotExists   z'PIDFileTests.test_isRunningDoesNotExistcCsHt|}|dttdddd}|td|||dS)zx L{PIDFile.isRunning} returns true for a process that we are not allowed to kill (errno=EPERM). r4Nr[cSsttjddS)NzOperation not permitted)rIrJEPERMr]r'r'r(r^sz3PIDFileTests.test_isRunningNotAllowed..killr^r_rbr'r'r(test_isRunningNotAlloweds   z%PIDFileTests.test_isRunningNotAllowedcCs8tdkrtdt|}|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_NAMEr rr/rQr!rar1r'r'r(test_isRunningInits   zPIDFileTests.test_isRunningInitcCsFt|}|ttdddd}|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. Nr[cSsttjddS)Nz File exists)rIrJEEXISTr]r'r'r(r^sz5PIDFileTests.test_isRunningUnknownErrno..killr^) rr/rUr`rNrrrIrarbr'r'r(test_isRunningUnknownErrnos  z'PIDFileTests.test_isRunningUnknownErrnocCst|}||dS)zS L{PIDFile.isRunning} returns false if the PID file doesn't exist. N)rr/rYrar1r'r'r(test_isRunningNoPIDFile#s z$PIDFileTests.test_isRunningNoPIDFilec Csbt|}||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/rYrWr!r7r:rr1r'r'r(test_contextManager+s  z PIDFileTests.test_contextManagerc Cszt|}|dttdddd}|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. r4Nr[cSsttjddSrerfr]r'r'r(r^Csz9PIDFileTests.test_contextManagerDoesntExist..killr^z'PID file refers to non-existing process) rr/rQr`rNrrrrar7r"r:r)rr2r^r%r'r'r(test_contextManagerDoesntExist9s  z+PIDFileTests.test_contextManagerDoesntExistcCsVt|}|dttdddd}|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. r4Nr[cSsdSr-r'r]r'r'r(r^Xsz.killr^) rr/rQr`rNrr!rarr __enter__rbr'r'r(!test_contextManagerAlreadyRunningNs   z.PIDFileTests.test_contextManagerAlreadyRunning)N)__name__ __module__ __qualname____doc__rbytesr r"r/r3r9r<rBrCrFrPrRrTrVrZr*rcrdrgrirlrnrorprqrsr'r'r'r(r+Gs>        r+c@sreZdZdZddddZddddZdddd Zddd d Zddd d ZddddZ ddddZ dS)NonePIDFileTestsz# Tests for L{NonePIDFile}. Nr0cCst}tt|dS)z9 L{NonePIDFile} conforms to L{IPIDFile}. N)rrrr1r'r'r(r3gszNonePIDFileTests.test_interfacecCs(t}|t|j}|t|ddS)z; L{NonePIDFile.read} raises L{NoPIDFound}. rDN)rrrr:r7r"rEr'r'r( test_readnszNonePIDFileTests.test_readcCs*t}|t|jd}||jtjdS)zZ L{NonePIDFile._write} raises L{OSError} with an errno of L{errno.EPERM}. rN)rrrIrQr7rJrhrr2rOr'r'r( test_writewszNonePIDFileTests.test_writecCs(t}|t|j}||jtjdS)zk L{NonePIDFile.writeRunningPID} raises L{OSError} with an errno of L{errno.EPERM}. N)rrrIrUr7rJrhr{r'r'r(rVsz%NonePIDFileTests.test_writeRunningPIDcCs(t}|t|j}||jtjdS)zZ L{NonePIDFile.remove} raises L{OSError} with an errno of L{errno.EPERM}. N)rrrIrXr7rJENOENTr{r'r'r(rZszNonePIDFileTests.test_removecCst}||ddS)z< L{NonePIDFile.isRunning} returns L{False}. FN)rr7rar1r'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)rr1r'r'r(rpsz$NonePIDFileTests.test_contextManager) rtrurvrwr3rzr|rVrZr~rpr'r'r'r(rybs    ry)&rwrJ functoolsrosrrrktypingrrrZzope.interface.verifyrZtyping_extensionsr Ztwisted.trial.unittestZtwistedZtwisted.python.filepathr Ztwisted.python.runtimer r runnerrrrrrrrrr*ZtrialZunittestZTestCaser+ryr'r'r'r(s"       $ &