U  W[ã@sdZddlmZmZddlZddlmZddlmZGdd„de ƒZ Gdd „d e ƒZ Gd d „d e ƒZ Gd d „d e e eƒZ Gdd„de e eƒZdS)z& Tests for L{twisted.python.systemd}. é)ÚdivisionÚabsolute_importN)ÚTestCase)Ú ListenFDsc@s eZdZdZdd„Zdd„ZdS)ÚInheritedDescriptorsMixinz¾ Mixin for a L{TestCase} subclass which defines test methods for some kind of systemd sd-daemon class. In particular, it defines tests for a C{inheritedDescriptors} method. cCs&| dd¡}| dddg| ¡¡dS)z‹ C{inheritedDescriptors} returns a list of integers giving the file descriptors which were inherited from systemd. éééé N©Ú getDaemonÚ assertEqualÚinheritedDescriptors©ÚselfÚsddaemon©rúB/usr/lib/python3/dist-packages/twisted/python/test/test_systemd.pyÚtest_inheritedDescriptorss z3InheritedDescriptorsMixin.test_inheritedDescriptorscCs$| dd¡}| | ¡| ¡¡dS)zW Any subsequent calls to C{inheritedDescriptors} return the same list. rrNr rrrrÚ test_repeateds  þz'InheritedDescriptorsMixin.test_repeatedN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrrrrs rc@seZdZdZdd„ZdS)ÚMemoryOnlyMixinzê Mixin for a L{TestCase} subclass which creates creating a fake, in-memory implementation of C{inheritedDescriptors}. This provides verification that the fake behaves in a compatible way with the real implementation. cCstt|||ƒƒS)a Invent C{count} new I{file descriptors} (actually integers, attached to no real file description), starting at C{start}. Construct and return a new L{ListenFDs} which will claim those integers represent inherited file descriptors. )rÚrange)rÚstartÚcountrrrr 0szMemoryOnlyMixin.getDaemonN)rrrrr rrrrr*src@s eZdZdZdd„Zdd„ZdS)ÚEnvironmentMixinaV Mixin for a L{TestCase} subclass which creates a real implementation of C{inheritedDescriptors} which is based on the environment variables set by systemd. To facilitate testing, this mixin will also create a fake environment dictionary and add keys to it to make it look as if some descriptors have been inherited. cCs&tj ¡}t|ƒ|d<t|ƒ|d<|S)zœ Create a copy of the process environment and add I{LISTEN_FDS} and I{LISTEN_PID} (the environment variables set by systemd) to it. Ú LISTEN_FDSÚ LISTEN_PID)ÚosÚenvironÚcopyÚstr)rrÚpidÚresultrrrÚinitializeEnvironmentCs   z&EnvironmentMixin.initializeEnvironmentcCs| |t ¡¡}tj||dS)a  Create a new L{ListenFDs} instance, initialized with a fake environment dictionary which will be set up as systemd would have set it up if C{count} descriptors were being inherited. The descriptors will also start at C{start}. )r"r)r'r!ÚgetpidrÚfromEnvironment)rrrÚfakeEnvironmentrrrr NszEnvironmentMixin.getDaemonN)rrrrr'r rrrrr;s rc@seZdZdZdS)ÚMemoryOnlyTestszb Apply tests to L{ListenFDs}, explicitly constructed with some fake file descriptors. N)rrrrrrrrr+Zsr+c@sHeZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dS)ÚEnvironmentTestszV Apply tests to L{ListenFDs}, constructed based on an environment dictionary. cCsV| dt ¡¡}tj|d}tj|d}| ttddƒƒ| ¡¡| g| ¡¡dS)zV Only a single L{Environment} can extract inherited file descriptors. r©r"éN) r'r!r(rr)r Úlistrr)rr*ÚfirstÚsecondrrrÚtest_secondEnvironmentfs   z'EnvironmentTests.test_secondEnvironmentcCs4| dt ¡d¡}tj|d}| g| ¡¡dS)z† If the current process PID does not match the PID in the environment, no inherited descriptors are reported. rér-N©r'r!r(rr)r r©rr*rrrrÚtest_mismatchedPIDqs z#EnvironmentTests.test_mismatchedPIDcCs6| dt ¡¡}|d=tj|d}| g| ¡¡dS)zz If the I{LISTEN_PID} environment variable is not present, no inherited descriptors are reported. rr r-Nr4r5rrrÚtest_missingPIDVariable{s z(EnvironmentTests.test_missingPIDVariablecCs,| dd¡}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. rú hello, worldr-N)r'rr)r rr5rrrÚtest_nonIntegerPIDVariable†s  z+EnvironmentTests.test_nonIntegerPIDVariablecCs6| dt ¡¡}|d=tj|d}| g| ¡¡dS)zz If the I{LISTEN_FDS} environment variable is not present, no inherited descriptors are reported. rrr-Nr4r5rrrÚtest_missingFDSVariables z(EnvironmentTests.test_missingFDSVariablecCs0| dt ¡¡}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. r8r-Nr4r5rrrÚtest_nonIntegerFDSVariable›s z+EnvironmentTests.test_nonIntegerFDSVariablecCsB| tdtt ¡ƒddœ¡t ¡}| ttddƒƒ|  ¡¡dS)z¢ If the process environment is not explicitly passed to L{Environment.__init__}, the real process environment dictionary is used. r"Ú5)r rrr N) Zpatchr!r$r(rr)r r/rrrrrrÚtest_defaultEnviron¥s þÿz$EnvironmentTests.test_defaultEnvironN) rrrrr2r6r7r9r:r;r=rrrrr,bs      r,)rZ __future__rrr!Ztwisted.trial.unittestrZtwisted.python.systemdrÚobjectrrrr+r,rrrrÚs