o ¯b€ã@szdZddlZddlmZddlmZGdd„dƒZGdd„dƒZGd d „d ƒZGd d „d eeeƒZ Gd d„deeeƒZ dS)z& Tests for L{twisted.python.systemd}. éN)Ú ListenFDs)ÚTestCasec@ó 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¡}| gd¢| ¡¡dS)z‹ C{inheritedDescriptors} returns a list of integers giving the file descriptors which were inherited from systemd. éé)réé 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 /szMemoryOnlyMixin.getDaemonN)rrrrr rrrrr(s rc@r)Ú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ÚinitializeEnvironmentBs   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 LszEnvironmentMixin.getDaemonN)rrrrr&r rrrrr9s rc@seZdZdZdS)ÚMemoryOnlyTestszb Apply tests to L{ListenFDs}, explicitly constructed with some fake file descriptors. N)rrrrrrrrr*Wsr*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_secondEnvironmentcs   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_mismatchedPIDms z#EnvironmentTests.test_mismatchedPIDcCó6| dt ¡¡}|d=tj|d}| g| ¡¡dS)zz If the I{LISTEN_PID} environment variable is not present, no inherited descriptors are reported. rrr,Nr3r4rrrÚtest_missingPIDVariablevó 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 r r4rrrÚtest_nonIntegerPIDVariable€s  z+EnvironmentTests.test_nonIntegerPIDVariablecCr6)zz If the I{LISTEN_FDS} environment variable is not present, no inherited descriptors are reported. rrr,Nr3r4rrrÚtest_missingFDSVariable‰r8z(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. r9r,Nr3r4rrrÚ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)rrrrN) Úpatchr r#r'rr(r r.rr rrrrÚtest_defaultEnvironœsz$EnvironmentTests.test_defaultEnvironN) rrrrr1r5r7r:r;r<r?rrrrr+^s     r+) rr Útwisted.python.systemdrÚtwisted.trial.unittestrrrrr*r+rrrrÚs