Your IP : 216.73.216.172
U
���]�) � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZ ddl
m Z ddl m
Z ddlmZ ddlm
Z ddlmZ ddlmZ ddlmZ G d d
� d
ej�ZG d d � d e�Zddd�Zdd� ZdS )z$Common utilities for certbot_apache.� N)�os)�util)�common�� configurator)�
entrypoint)�objc s& e Zd Zd� fdd� Zdd� Z� ZS ) �
ApacheTest�!debian_apache_2_4/multiple_vhosts�)debian_apache_2_4/multiple_vhosts/apache2�9debian_apache_2_4/multiple_vhosts/apache2/sites-availablec s� t t| ��� tj|dd�\| _| _| _tj �
| j|�| _ tj �
| j|�| _ t
j�t�d��| _t| j || j| j�| _tj �
| j d�}tj �|�s�d S t�|�D ]R}|dkr�q�tj �
||�}tj �|�s�t�|� tj �
tj jd|�}t�||� q�d S )Nzcertbot_apache.tests)�test_dir�pkgzrsa512_key.pemz
sites-enabled�non-symlink.confzsites-available)�superr �setUpr Z dir_setup�temp_dir�
config_dir�work_dirr �path�join� config_path�
vhost_path�joseZJWKRSA�load� test_utilZ load_vectorZ rsa512jwk�get_apache_configurator�config�exists�listdir�islink�remove�pardir�symlink)�selfr
� config_root�
vhost_rootZ
sites_enabledZvhost_basenameZvhost�target�� __class__� �;/usr/lib/python3/dist-packages/certbot_apache/tests/util.pyr s: �
� �
�zApacheTest.setUpc C s( t �| j� t �| j� t �| j� d S )N)�shutilZrmtreer r r )r$ r* r* r+ �tearDown; s zApacheTest.tearDown)r
r r )�__name__�
__module__� __qualname__r r- �
__classcell__r* r* r( r+ r s
�$r c s e Zd Zd� fdd� Z� ZS )�
ParserTestr
r r c s� t t| ��|||� tj�t�tj d�� ddl
m } t j
t j
jt j
jB d�| _t�d�� || j| j| jd�| _W 5 Q R X d S )NFr )� ApacheParser)�flags�;certbot_apache.parser.ApacheParser.update_runtime_variablesr )r r2 r �zopeZ componentZprovideUtility� display_utilZ FileDisplay�sys�stdoutZcertbot_apache.parserr3 �augeasZAugeasZNONEZNO_MODL_AUTOLOADZaug�mock�patchr r r �parser)r$ r
r% r& r3 r( r* r+ r C s � � �zParserTest.setUp)r
r r )r. r/ r0 r r1 r* r* r( r+ r2 A s �r2 �� � � �genericc C s� t j�|d�}tj| dd| d||dt j�|d�t j�|d�|d� }t�d��� t�d ���} d
| _t�d ��l ztj| }
W n t k
r� t
j }
Y nX |
|d |d
�} |s�||
j d< n|| j
_|
j d | j
_| �� W 5 Q R X W 5 Q R X W 5 Q R X | S )z�Create an Apache Configurator with the specified options.
:param conf: Function that returns binary paths. self.conf in Configurator
�backupsNz -le-ssl.conf�P Ztemp_checkpointsZ IN_PROGRESS) Zapache_server_root�apache_vhost_rootZapache_le_vhost_extZapache_challenge_locationZ apache_enmodZ
backup_dirr Z http01_portZtemp_checkpoint_dirZin_progress_dirr z+certbot_apache.configurator.util.run_scriptz+certbot_apache.configurator.util.exe_existsTr5 Zapache)r �name�versionr& Zctl)r r r r; Z MagicMockr< Z return_valuer ZOVERRIDE_CLASSES�KeyErrorr ZApacheConfiguratorZ OS_DEFAULTSr rE Z
apache_ctlZprepare) r r r r rG Zos_infoZconf_vhost_pathrC Zmock_le_configZmock_exe_existsZ config_classr r* r* r+ r T s@
�
� &r c C s |dk�r�t j�| |d�}d| }t�t j�|d�t j�|d�ttj�d�g�ddd �t�t j�|d
�t j�|d �ttj�d �g�dd�t�t j�|d
�t j�|d�ttj�d�tj�d�g�ddd�tjt j�|d�t j�|d�ttj�d�g�ddddgd�tjt j�|d�t j�|d�ttj�d�g�dddd�t�t j�|d�t j�|d�ttj�d �g�dd�tjt j�|d�t j�|d �ttj�d�g�dddd gd�t�t j�|d �t j�|d �ttj�d �g�ddd!�t�t j�|d"�t j�|d#�ttj�d�g�ddd$�t�t j�|d�t j�|d%�ttj�d�g�ddd&�t�t j�| |d'�dt j�| |d(� ttj�d�g�ddd)�t�t j�|d*�t j�|d+�ttj�d,�g�ddd-�t�t j�|d.�t j�|d/�ttj�d �g�ddd-�g
}|S |d0k�rt j�| |d1�}d| }t�t j�|d2�t j�|d3�ttj�d�g�ddd�t�t j�|d2�t j�|d4�ttj�d�g�ddd5�t�t j�|d6�t j�|d7�ttj�d�g�ddd8�t�t j�|d6�t j�|d9�ttj�d�g�ddd:�t�t j�|d6�t j�|d;�ttj�d�g�ddd<�g}|S d=S )>z4Return the ground truth for the specified directory.r
zapache2/sites-enabledz/fileszencryption-example.confz#encryption-example.conf/Virtualhostz*:80FTzencryption-example.demozdefault-ssl.confz%default-ssl.conf/IfModule/VirtualHostz
_default_:443z000-default.confz 000-default.conf/VirtualHostz[::]:80zip-172-30-0-17z certbot.confzcertbot.conf/VirtualHostz certbot.demozwww.certbot.demo)�aliaseszmod_macro-example.confz(mod_macro-example.conf/Macro/VirtualHost)Zmodmacrozdefault-ssl-port-only.confz/default-ssl-port-only.conf/IfModule/VirtualHostz
wildcard.confzwildcard.conf/VirtualHostz*.blue.purple.comz
ocsp-ssl.confz"ocsp-ssl.conf/IfModule/VirtualHostz 10.2.3.4:443z
ocspvhost.comr z non-symlink.conf/VirtualHostz nonsym.linkz&default-ssl-port-only.conf/VirtualHost� zapache2/apache2.confz apache2/apache2.conf/VirtualHostzvhost.in.rootconfzduplicatehttp.confz duplicatehttp.conf/VirtualHostz 10.2.3.4:80zduplicate.example.comzduplicatehttps.confz(duplicatehttps.conf/IfModule/VirtualHostz debian_apache_2_4/multi_vhostszapache2/sites-availablez default.confzdefault.conf/VirtualHost[1]zdefault.conf/VirtualHost[2]zbanana.vomit.comzmulti-vhost.confz multi-vhost.conf/VirtualHost[1]z1.multi.vhost.tldz%multi-vhost.conf/IfModule/VirtualHostz2.multi.vhost.tldz multi-vhost.conf/VirtualHost[2]z3.multi.vhost.tldN)r r r r Z VirtualHost�setZAddrZ
fromstring)r Z config_name�prefixZaug_preZvh_truthr* r* r+ � get_vh_truth� s^
� � � �
� � � � � � � � � � �
� � � � ��F
� � � � � ��rM )r> rB N) �__doc__r, r8 Zunittestr: Zjosepyr r; Zzope.componentr6 Zcertbot.compatr Zcertbot.displayr r7 Zcertbot.pluginsr Z
certbot.testsr Zcertbot_apacher r r ZTestCaser r2 r rM r* r* r* r+ �<module> s* , �
.