File: //lib/python3/dist-packages/twisted/python/test/__pycache__/test_systemd.cpython-38.pyc
U
W[ � @ s� d Z ddlmZmZ ddlZddlmZ ddlmZ G dd� de �Z
G dd � d e �Z G d
d � d e �Z G d d
� d
e e
e�Z
G dd� de e
e�ZdS )z&
Tests for L{twisted.python.systemd}.
� )�division�absolute_importN)�TestCase)� ListenFDsc @ s 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�}| �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_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 0 s zMemoryOnlyMixin.getDaemonN)r r r r r r r r r r * s r c @ s e Zd ZdZdd� 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.
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 �initializeEnvironmentC 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 N s zEnvironmentMixin.getDaemonN)r r r r r' r r r r r r ; 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+ Z 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_secondEnvironmentf 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_mismatchedPIDq s z#EnvironmentTests.test_mismatchedPIDc C s6 | � 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- Nr4 r5 r r r �test_missingPIDVariable{ s 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 r5 r r r �test_nonIntegerPIDVariable� s z+EnvironmentTests.test_nonIntegerPIDVariablec C s6 | � dt�� �}|d= tj|d�}| �g |�� � dS )zz
If the I{LISTEN_FDS} environment variable is not present, no inherited
descriptors are reported.
r r r- Nr4 r5 r r r �test_missingFDSVariable� s 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.
r8 r- Nr4 r5 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)
Zpatchr! r$ r( r r) r
r/ r r r r r r �test_defaultEnviron� s
��z$EnvironmentTests.test_defaultEnvironN) r r r r r2 r6 r7 r9 r: r; r= r r r r r, b s
r, )r Z
__future__r r r! Ztwisted.trial.unittestr Ztwisted.python.systemdr �objectr r r r+ r, r r r r �<module> s