U 2ܟj@sdZddlZddlmZmZmZddlmZmZm Z ddl m Z ddl m Z mZmZddlmZddlmZd d lmZeeeeefd d d ZGdddeZdS)z& Tests for L{twisted.python.systemd}. N)DictMappingSequence) assert_thatequal_tonot_)given) dictionariesintegerslists) ListenFDs)SynchronousTestCase)systemdDescriptorNames)countpidreturncCsBtj}t||d<dddt|D|d<t||d<|S)a+ @param count: The number of file descriptors to indicate as inherited. @param pid: The pid of the inheriting process to indicate. @return: A copy of the current process environment with the I{systemd} file descriptor inheritance-related environment variables added to it. LISTEN_FDS:cSsg|]}|dqS)z.socket).0nrrD/tmp/pip-unpacked-wheel-4b529toj/twisted/python/test/test_systemd.py sz$buildEnvironment..LISTEN_FDNAMES LISTEN_PID)osenvironcopystrjoinrange)rrresultrrrbuildEnvironments   r#c@s$eZdZdZeeedddeeddddZ dd d d Z dd d d Z dd ddZ dd ddZ dd ddZdd ddZdd ddZeeedddddeeddddZeeeeddddeeefdd d!d"Zeeedddddeeddd#d$ZdS)%ListenFDsTestszV Apply tests to L{ListenFDs}, constructed based on an environment dictionary. r )Zmin_sizemax_sizeN)namesrcCs^t|}tttjtj|}tttt|d |d}t |t t|t |dS)a The L{ListenFDs} and L{ListenFDs.fromEnvironment} constructors are equivalent for their respective representations of the same information. @param names: The names of the file descriptors to represent as inherited in the test environment given to the parser. The number of descriptors represented will equal the length of this list. r)rrrN) lenlistr!r Z_STARTfromEnvironmentrrgetpidr rrtuple)selfr'ZnumFDs descriptorsfdsrrrtest_fromEnvironmentEquivalence*s  z.ListenFDsTests.test_fromEnvironmentEquivalence)rcCs>|tdtdtt}|ttdd| dS)z If the process environment is not explicitly passed to L{ListenFDs.fromEnvironment}, the real process environment dictionary is used. rN) patchrr#r+r r* assertEqualr)r!inheritedDescriptors)r-sddaemonrrrtest_defaultEnviron@sz"ListenFDsTests.test_defaultEnvironcCsTtdt}tj|d}tj|d}|ttdd||g|dS)a  L{ListenFDs.fromEnvironment} removes information about the inherited file descriptors from the environment mapping so that the same inherited file descriptors cannot be handled repeatedly from multiple L{ListenFDs} instances. r2rN) r#rr+r r*r5r)r!r6)r-envfirstsecondrrrtest_secondEnvironmentJs   z%ListenFDsTests.test_secondEnvironmentcCs2tdtd}tj|d}|g|dS)aV If the current process PID does not match the PID in the environment then the systemd variables in the environment were set for a different process (perhaps our parent) and the inherited descriptors are not intended for this process so L{ListenFDs.inheritedDescriptors} returns an empty list. r2rr9Nr#rr+r r*r5r6r-r;r7rrrtest_mismatchedPIDWs z!ListenFDsTests.test_mismatchedPIDcCs4tdt}|d=tj|d}|g|dS)z If the I{LISTEN_PID} environment variable is not present then there is no clear indication that any file descriptors were inherited by this process so L{ListenFDs.inheritedDescriptors} returns an empty list. r2rr9Nr?r@rrrtest_missingPIDVariablecs z&ListenFDsTests.test_missingPIDVariablecCs*tdd}tj|d}|g|dS)z If the I{LISTEN_PID} environment variable is set to a string that cannot be parsed as an integer, no inherited descriptors are reported. r2 hello, worldr9N)r#r r*r5r6r@rrrtest_nonIntegerPIDVariableos  z)ListenFDsTests.test_nonIntegerPIDVariablecCs:tdt}|d=|d=tj|d}|g|dS)z If the I{LISTEN_FDS} and I{LISTEN_FDNAMES} environment variables are not present, no inherited descriptors are reported. r2rrr9Nr?r@rrrtest_missingFDSVariablexs  z&ListenFDsTests.test_missingFDSVariablecCs6tdt}d|d<tj|d}|g|dS)z If the I{LISTEN_FDS} environment variable is set to a string that cannot be parsed as an integer, no inherited descriptors are reported. r2rCrr9Nr?r@rrrtest_nonIntegerFDSVariables z)ListenFDsTests.test_nonIntegerFDSVariable) min_valueZ max_valueT)unique)r.rcCsNttt|}t||}|}t|t||dt|tt|dS)zm L{ListenFDs.inheritedDescriptors} returns a copy of the inherited descriptors list. rN) r,maprr r6rrappendr)r-r.r'r/ZfdsCopyrrrtest_inheritedDescriptorss   z(ListenFDsTests.test_inheritedDescriptors)rG)r&)expectedrcCsHt|}dd|D}dd|D}t||}t|t|dS)z L{ListenFDs.inheritedNamedDescriptors} returns a mapping from the descriptor names to their integer values, with items formed by pairwise combination of the input descriptors and names. cSsg|] \}}|qSrr)rname_rrrrszAListenFDsTests.test_inheritedNamedDescriptors..cSsg|] \}}|qSrr)rrNfdrrrrsN)r)itemsr rinheritedNamedDescriptorsr)r-rLrPr'r.r/rrrtest_inheritedNamedDescriptorss   z-ListenFDsTests.test_inheritedNamedDescriptorscCsDttt|}t||}||||||dS)z Any subsequent calls to C{inheritedDescriptors} and C{inheritedNamedDescriptors} return the same list. N)r,rIrr r5r6rQ)r-r.r'r7rrr test_repeateds zListenFDsTests.test_repeated)__name__ __module__ __qualname____doc__rr rrrr0r8r>rArBrDrErFr intrKr rrRrSrrrrr$%s          r$)rWrtypingrrrZhamcrestrrrZ hypothesisrZhypothesis.strategiesr r r Ztwisted.python.systemdr Ztwisted.trial.unittestr Z strategiesrrXobjectrr#r$rrrrs