← Back
Editing: 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
Save File
Cancel