CasperSecurity
o
�b� � @ sz d Z ddlZddlmZ ddlmZ G dd� d�ZG dd� d�ZG d d
� d
�ZG d d � d eee�Z G d
d� deee�Z
dS )z&
Tests for L{twisted.python.systemd}.
� N)� ListenFDs)�TestCasec @ � e Zd ZdZdd� 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.
c C s$ | � dd�}| �g d�|�� � 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_inheritedDescriptors s z3InheritedDescriptorsMixin.test_inheritedDescriptorsc C s$ | � dd�}| �|�� |�� � dS )zW
Any subsequent calls to C{inheritedDescriptors} return the same list.
r r Nr
r r r r �
test_repeated s �z'InheritedDescriptorsMixin.test_repeatedN)�__name__�
__module__� __qualname__�__doc__r r r r r r r s r c @ s e Zd ZdZdd� 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.
c C s t t||| ��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�countr r r r / s zMemoryOnlyMixin.getDaemonN)r r r r r r r r r r ( s r c @ 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.
c C s& t j�� }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)r r �pid�resultr r r �initializeEnvironmentB s
z&EnvironmentMixin.initializeEnvironmentc C s | � |t�� �}tj||d�S )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)r r r �fakeEnvironmentr r r r L s zEnvironmentMixin.getDaemonN)r r r r r& r r r r r r 9 s
r c @ s e Zd ZdZdS )�MemoryOnlyTestszb
Apply tests to L{ListenFDs}, explicitly constructed with some fake file
descriptors.
N)r r r r r r r r r* W s r* c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�EnvironmentTestszV
Apply tests to L{ListenFDs}, constructed based on an environment dictionary.
c C sV | � 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' r r( r �listr r
)r r) �first�secondr r r �test_secondEnvironmentc s
z'EnvironmentTests.test_secondEnvironmentc C s4 | � 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' r r( r r
�r r) r r r r �test_mismatchedPIDm s z#EnvironmentTests.test_mismatchedPIDc C �6 | � dt�� �}|d= tj|d�}| �g |�� � dS )zz
If the I{LISTEN_PID} environment variable is not present, no inherited
descriptors are reported.
r r r, Nr3 r4 r r r �test_missingPIDVariablev � z(EnvironmentTests.test_missingPIDVariablec C s, | � 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& r r( r r
r4 r r r �test_nonIntegerPIDVariable� s z+EnvironmentTests.test_nonIntegerPIDVariablec C r6 )zz
If the I{LISTEN_FDS} environment variable is not present, no inherited
descriptors are reported.
r r r, Nr3 r4 r r r �test_missingFDSVariable� r8 z(EnvironmentTests.test_missingFDSVariablec C s0 | � 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.
r9 r, Nr3 r4 r r r �test_nonIntegerFDSVariable� s z+EnvironmentTests.test_nonIntegerFDSVariablec C sB | � 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 r r r N)
�patchr r# r' r r( r r. r r
r r r r �test_defaultEnviron� s z$EnvironmentTests.test_defaultEnvironN) r r r r r1 r5 r7 r: r; r< r? r r r r r+ ^ s
r+ ) r r �twisted.python.systemdr �twisted.trial.unittestr r r r r* r+ r r r r �<module> s