File: //lib/python3/dist-packages/twisted/python/test/__pycache__/modules_helpers.cpython-38.pyc
U
W[E � @ s< d Z ddlmZmZ ddlZddlmZ G dd� de�ZdS )z\
Facilities for helping test code which interacts with Python's module system
to load code.
� )�division�absolute_importN)�FilePathc @ s* e Zd ZdZdd� Zdd� Zd
dd�Zd S ) �TwistedModulesMixinz�
A mixin for C{twisted.trial.unittest.SynchronousTestCase} providing useful
methods for manipulating Python's module system.
c s6 t jdd� � � fdd�}| �|� |t jdd�< dS )zW
Replace sys.path, for the duration of the test, with the given value.
Nc s � t jd d �< d S �N)�sys�path� �ZoriginalSysPathr �E/usr/lib/python3/dist-packages/twisted/python/test/modules_helpers.py�cleanUpSysPath s z:TwistedModulesMixin.replaceSysPath.<locals>.cleanUpSysPath)r r �
addCleanup)�selfZsysPathr r r
r �replaceSysPath s
z"TwistedModulesMixin.replaceSysPathc s: t j�� � � fdd�}| �|� t j�� t j�|� dS )zZ
Replace sys.modules, for the duration of the test, with the given value.
c s t j�� t j�� � d S r )r �modules�clear�updater �ZoriginalSysModulesr r �cleanUpSysModules& s
z@TwistedModulesMixin.replaceSysModules.<locals>.cleanUpSysModulesN)r r �copyr
r r )r Z
sysModulesr r r r �replaceSysModules! s
z%TwistedModulesMixin.replaceSysModules� test_packagec 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 ZmktempZchild�makedirsZ
setContent)r Zpkgname�entry�pkgr r r �pathEntryWithOnePackage. s
z+TwistedModulesMixin.pathEntryWithOnePackageN)r )�__name__�
__module__� __qualname__�__doc__r r r r r r r r s
r ) r Z
__future__r r r Ztwisted.python.filepathr �objectr r r r r �<module> s