File: //usr/lib/python3/dist-packages/certbot/plugins/__pycache__/disco_test.cpython-38.pyc
U
�]\- � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddl m
Z
ddl m Z ddl m
Z
ddlmZ ddlmZ ejdd d
ejd d �d
�Zejddd
ejd d �d
�ZG dd� dej�ZG dd� dej�Zedkr�e�� dS )z Tests for certbot.plugins.disco.� N)�List)�errors)�
interfaces)�
standalone)�webroot�sazcertbot.plugins.standalone)�
Authenticator�certbot��key)Zattrs�dist�wrzcertbot.plugins.webrootc @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd S )!�PluginEntryPointTestz1Tests for certbot.plugins.disco.PluginEntryPoint.c C s� t jddtjdd�d�| _t jddtjdd�d�| _t jdd tjdd�d�| _t jd
d tjd d�d�| _d
dlm } |t
�| _ d S )N�ep1zp1.ep1Zp1r
)r zp2.ep2Zp2�ep2z
p2.foo.ep2�ep3za.ep3Zp3r ��PluginEntryPoint) �
pkg_resources�
EntryPoint�mock� MagicMockr �ep1primr r �certbot.plugins.discor �EP_SA� plugin_ep)�selfr � r �</usr/lib/python3/dist-packages/certbot/plugins/disco_test.py�setUp s,
�
�
�
� zPluginEntryPointTest.setUpc
C sV ddl m} | jd| jd| jd| jdtdi}t�|�D ]\}}| � ||�
|�� q6d S )Nr r zp1:ep1zp2:ep1zp2:ep2zp3:ep3r ) r r r r r r r �sixZ iteritems� assertEqualZentry_point_to_plugin_name)r r �names� entry_point�namer r r � test_entry_point_to_plugin_name. s" � �z4PluginEntryPointTest.test_entry_point_to_plugin_namec C s | � d| jjk� d S )Nztemporary webserver)�
assertTruer � description�r r r r �test_description= s z%PluginEntryPointTest.test_descriptionc C s$ t jdd�| j_| �d| jj� d S )NZDesc)r' z Desc (sa))r r r �
plugin_clsr! Zdescription_with_namer( r r r �test_description_with_name@ s
�z/PluginEntryPointTest.test_description_with_namec C s$ t jdd�| j_| �d| jj� d S )Nz Long desc)�long_description�r r r r* r! r, r( r r r �test_long_descriptionE s �
�z*PluginEntryPointTest.test_long_descriptionc C s( t jddgd�| j_| �d| jj� d S )NzLong desc not foundr' )r' �specr- r( r r r �!test_long_description_nonexistentK s �
�z6PluginEntryPointTest.test_long_description_nonexistentc C sJ | � | j�tjf�� | �| j�tjf�� | �| j�tjtjf�� d S �N)r& r �ifacesr ZIAuthenticator� assertFalseZ
IInstallerr( r r r � test_ifacesQ s
�z PluginEntryPointTest.test_ifacesc C s� | � | jj� | � | jj� | � | jj� | � | jj� | �| jjd k� | �| jjt k� | �
d| jj � | �| jj t
jk� d S )Nr )r3 r � initialized�prepared�
misconfigured� availabler& �problemr# r r! r$ r* r r r( r r r � test__init__W s z!PluginEntryPointTest.test__init__c C s� t �� }| jj|d�}| �| jj� | �|j|k� | �| j�� |k� | �|j|k� | �| j�d�|k� | �|j|k� | �| jj� | �| jj � | �| jj
� d S )N��config�{ ) r r r �initr& r5 r< r3 r6 r7 r8 )r r< �pluginr r r � test_initb s z PluginEntryPointTest.test_initc s� t jdd�� t jdd��t jdd��t �� | j_�tjj}t �d��~�|�_� ����fdd�}|�jj _
| � | j�� f�� | � | j�� �f�� | � | j��f�� | � | j�� �f�� W 5 Q R X d S )N�iface1)�__name__�iface2�iface3z$certbot.plugins.disco.zope.interfacec sB |�ks t �| � ks(| �ks(| �ks(t �| �kr>�j�d d ��d S r1 )�AssertionError�
exceptionsZBrokenImplementation)Ziface�obj�rA rC rD Z mock_zoper? r r �
verify_object~ s z7PluginEntryPointTest.test_verify.<locals>.verify_object)
r r r � _initialized�zopeZ interfacerF �patch�verifyZ verifyObject� side_effectr& r3 )r rF rI r rH r � test_verifyr s
z PluginEntryPointTest.test_verifyc C sJ t �� }| jj|d� | j�� | �| jj� | �| jj� t | j� d S )Nr; )
r r r r> �preparer& r6 r3 r7 �str)r r< r r r � test_prepare� s
z!PluginEntryPointTest.test_preparec C sv t �� }tj|j_|| j_| �t | j�� tj�� | �| jj
� | �| jj � | �t | jj tj�� | �| jj
� d S r1 )r r r ZMisconfigurationErrorrP rN r rJ r& �
isinstancer6 r7 r9 r8 �r r? r r r �test_prepare_misconfigured� s
� �z/PluginEntryPointTest.test_prepare_misconfiguredc C s` t �� }tj|j_|| j_| �t | j�� tj�� | �| jj
� | � | jj � | � | jj
� d S r1 )r r r ZNoInstallationErrorrP rN r rJ r& rS r6 r3 r7 r8 rT r r r � test_prepare_no_installation� s
�z1PluginEntryPointTest.test_prepare_no_installationc C s` t �� }tj|j_|| j_| �t | j�� tj�� | �| jj
� | � | jj � | � | jj
� d S r1 )r r r Z PluginErrorrP rN r rJ r& rS r6 r3 r7 r8 rT r r r �!test_prepare_generic_plugin_error� s
z6PluginEntryPointTest.test_prepare_generic_plugin_errorc C s | � dt| j�� d S )NzPluginEntryPoint#sa)r! �reprr r( r r r � test_repr� s z PluginEntryPointTest.test_reprN)rB �
__module__� __qualname__�__doc__r r% r) r+ r. r0 r4 r: r@ rO rR rU rV rW rY r r r r r s
r c @ s� e Zd ZdZedd� �Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"S )#�PluginsRegistryTestz0Tests for certbot.plugins.disco.PluginsRegistry.c C s ddl m} ||�S )Nr ��PluginsRegistry)r r_ )�cls�pluginsr_ r r r �_create_new_registry� s z(PluginsRegistryTest._create_new_registryc C s> t �� | _d| j_t| jj_| jj| ji| _| �| j�| _ d S �Nr )
r r r r$ � TypeError�__hash__rN ra rb �regr( r r r r � s
zPluginsRegistryTest.setUpc C s� ddl m} t�d��&}ttg�ttg�g|j_|� � }W 5 Q R X | �
|d j t j
k� | �
|d jtk� | �
|d j tj
k� | �
|d jtk� d S )Nr r^ z#certbot.plugins.disco.pkg_resourcesr r
)r r_ r rL �iterr �EP_WRZiter_entry_pointsrN Zfind_allr& r* r r r# r )r r_ Zmock_pkgra r r r �
test_find_all� s �z!PluginsRegistryTest.test_find_allc C s | � | j| jd � d S rc )r! r rf r( r r r � test_getitem� s z PluginsRegistryTest.test_getitemc C s | � dgt| j�� d S rc )r! �listrf r( r r r � test_iter� s z PluginsRegistryTest.test_iterc C s, | � dt| �i ��� | � dt| j�� d S )Nr � )r! �lenrb rf r( r r r �test_len� s z PluginsRegistryTest.test_lenc C s2 d| j j_| �dg| j�d�� | j j�d� d S )N�bazZbar)r r> � return_valuer! rf �assert_called_once_withr( r r r r@ � s
z PluginsRegistryTest.test_initc C s6 | � | j| j�dd� �� | � i | j�dd� �� d S )Nc S s | j �d�S )N�m�r$ �
startswith�Zp_epr r r �<lambda>� � z1PluginsRegistryTest.test_filter.<locals>.<lambda>c S s | j �d�S )N�brt rv r r r rw � rx )r! ra rf �filterr( r r r � test_filter� s � �z PluginsRegistryTest.test_filterc C sB d| j j_| �| j| j�� j� d| j j_| �i | j�� j� d S �NTF)r r2 rq r! ra rf �_pluginsr( r r r r4 � s
z PluginsRegistryTest.test_ifacesc C sN d| j j_| �| j| j�t�� �j� d| j j_| �i | j�t�� �j� d S r| ) r rM rq r! ra rf r r r} r( r r r rO � s
�
z PluginsRegistryTest.test_verifyc C s. d| j j_| �dg| j�� � | j j�� d S )Nrp )r rP rq r! rf rr r( r r r rR � s
z PluginsRegistryTest.test_preparec sD g � t � fdd�tjD ��}| �|�}|�� | �� ttj�� d S )Nc 3 s( | ] }|t jt�� j|�d �fV qdS ))rP N)r r � functools�partial�append)�.0�c��orderr r � <genexpr>� s �z9PluginsRegistryTest.test_prepare_order.<locals>.<genexpr>)�dict�stringZ
ascii_lettersrb rP r! �sorted�r ra rf r r� r �test_prepare_order� s �
z&PluginsRegistryTest.test_prepare_orderc C s> d| j _| �| j| j�� j� d| j _| �i | j�� j� d S r| )r r8 r! ra rf r} r( r r r �test_available s z"PluginsRegistryTest.test_availablec C sD | � | j�t�� �d k� d| j_| � | j�| j�� �| jk� d S )NT)r& rf Z find_initr �Mockr r5 r> r( r r r �test_find_init s
�z"PluginsRegistryTest.test_find_initc C s" dd� | j _| �dt| j�� d S )Nc S s dS )NzPluginEntryPoint#mockr ��_r r r rw rx z/PluginsRegistryTest.test_repr.<locals>.<lambda>z&PluginsRegistry(PluginEntryPoint#mock))r �__repr__r! rX rf r( r r r rY s �z PluginsRegistryTest.test_reprc C sd | � dt| �i ��� dd� | j_| � dt| j�� | jj| jddi}| �|�}| � dt|�� d S )Nz
No pluginsc S s dS )Nr� r r� r r r rw rx z.PluginsRegistryTest.test_str.<locals>.<lambda>r� ZfooZBarz Bar
Mock)r! rQ rb r �__str__rf r$ r� r r r �test_str s
z PluginsRegistryTest.test_strN)rB rZ r[ r\ � classmethodrb r ri rj rl ro r@ r{ r4 rO rR r� r� r� rY r� r r r r r] � s$
r] �__main__)r\ r~ r� Zunittestr r r Zzope.interfacerK Zacme.magic_typingr r r r Zcertbot.pluginsr r r r r rh ZTestCaser r] rB �mainr r r r �<module> s:
�
� j