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