CasperSecurity
o
�b
� @ sJ d Z ddlZddlmZ ddlmZmZmZ ddlm Z G dd� d�Z
dS )z\
Facilities for helping test code which interacts with Python's module system
to load code.
� N)�
ModuleType)�Iterable�List�Tuple)�FilePathc @ sX e Zd ZdZdee ddfdd�Zdeeee f ddfdd �Z
dd ede fd d
�Z dS )�TwistedModulesMixinz�
A mixin for C{twisted.trial.unittest.SynchronousTestCase} providing useful
methods for manipulating Python's module system.
�sysPath�returnNc s8 t jdd� � d� fdd� }| �|� |t jdd�< dS )zW
Replace sys.path, for the duration of the test, with the given value.
Nr c s � t jd d �< d S �N)�sys�path� ��originalSysPathr
�E/usr/lib/python3/dist-packages/twisted/python/test/modules_helpers.py�cleanUpSysPath s z:TwistedModulesMixin.replaceSysPath.<locals>.cleanUpSysPath�r N)r r �
addCleanup)�selfr r r
r r �replaceSysPath s
z"TwistedModulesMixin.replaceSysPath�
sysModulesc s< t j�� � d� fdd� }| �|� t j�� t j�|� dS )zZ
Replace sys.modules, for the duration of the test, with the given value.
r Nc s t j�� t j�� � d S r
)r �modules�clear�updater
��originalSysModulesr
r �cleanUpSysModules( s
z@TwistedModulesMixin.replaceSysModules.<locals>.cleanUpSysModulesr )r r �copyr r r )r r r r
r r �replaceSysModules" s
z%TwistedModulesMixin.replaceSysModules� test_package�pkgnamec C s2 t | �� �}|�d�}|�� |�d��d� |S )z�
Generate a L{FilePath} with one package, named C{pkgname}, on it, and
return the L{FilePath} of the path entry.
r z __init__.py� )r �mktemp�child�makedirs�
setContent)r r �entry�pkgr
r
r �pathEntryWithOnePackage0 s
z+TwistedModulesMixin.pathEntryWithOnePackage)r )
�__name__�
__module__� __qualname__�__doc__r �strr r r r r r r( r
r
r
r r s
r ) r, r �typesr �typingr r r �twisted.python.filepathr r r
r
r
r �<module> s