← Back
Editing: ops_test.cpython-38.pyc
U �]�T � @ sn d Z ddlZddlZddlZddlZddlZddlm Z ddl m m Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lm Z ej�e �d ��ZG d d � d ej�Z G d d� de j �Z G dd� dej�Z G dd� dej�Z G dd� dej�Z!G dd� dej�Z"G dd� dej�Z#G dd� dej�Z$G dd � d ej�Z%G d d � d ej�Z&e'd k�rje�(� dS ) zTest certbot.display.ops.� N)�messages)�account)�errors)�os)� filesystem)�ops)�utilzrsa512_key.pemc @ s� e Zd ZdZedd� �Ze�d�dd� �Ze�d�dd� �Z e�d�d d � �Z e�d�d d � �Z e�d�d d� �Z e�d�dd� �Z dS )� GetEmailTestz(Tests for certbot.display.ops.get_email.c K s ddl m} |f |�S )Nr )� get_email)�certbot.display.opsr )�cls�kwargsr � r �@/usr/lib/python3/dist-packages/certbot/tests/display/ops_test.py�_call s zGetEmailTest._call�certbot.display.ops.z_utilc C s< |� j }tjdf|_| �tj| j� | jtj| jdd� d S )N� foo@bar.bazF�Zoptional)�input� display_util�CANCEL� return_value� assertRaisesr �Errorr )�self�mock_get_utility� mock_inputr r r �test_cancel_none s z GetEmailTest.test_cancel_nonec C sF |� j }tjdf|_t�d�� }d|_| �| �� dk� W 5 Q R X d S )Nr �#certbot.display.ops.util.safe_emailT)r r �OKr �mock�patch� assertTruer �r r r �mock_safe_emailr r r � test_ok_safe& s zGetEmailTest.test_ok_safec C sJ |� j }tjdf|_t�d��"}ddg|_| �| �� dk� W 5 Q R X d S )Nr r FT) r r r r r r! � side_effectr" r r# r r r �test_ok_not_safe. s z GetEmailTest.test_ok_not_safec C s| d}|� j }tjdf|_t�d��P}d|_| �� | �||jd d k� | jdd� | �||jd d k� W 5 Q R X d S )N�There seem to be problemsr r Tr �Zinvalid) r r r r r r! r r" � call_args�r r Z invalid_txtr r$ r r r �test_invalid_flag6 s z GetEmailTest.test_invalid_flagc C sf |� j }tjdf|_t�d��>}ddg|_| jdd� |jD ]}| � d|d d k� q<W 5 Q R X d S )Nr r FTr z!--register-unsafely-without-emailr ) r r r r r r! r& r �call_args_listr" )r r r r$ �callr r r �test_optional_flagB s �z GetEmailTest.test_optional_flagc C s` d}|� j }tjdf|_t�d��4}ddg|_| jdd� | �||j d d k� W 5 Q R X d S )Nr( r r FTr) r ) r r r r r r! r& r r" r* r+ r r r � test_optional_invalid_unsafeM s z)GetEmailTest.test_optional_invalid_unsafeN)�__name__� __module__� __qualname__�__doc__� classmethodr � test_util�patch_get_utilityr r% r' r, r/ r0 r r r r r s r c sb e Zd ZdZ� fdd�Zedd� �Ze�d�dd� �Z e�d�d d � �Z e�d�d d � �Z � Z S ) �ChooseAccountTestz-Tests for certbot.display.ops.choose_account.c s� t t| ��� tj�t�tj d�� t j � | j d�| _t�| jd� tj| j | jdd�| _t| _t�tjd tjjdd�d�| j�| _ t�tjd tjjd d d �d�| j�| _ d S ) NF�keysi� zcertbot-demo.org)Z accounts_dir�account_keys_dirZserverz email1@g.com)�email)ZuriZbodyz email2@g.com�phone)r; r< ) �superr8 �setUp�zope� component�provideUtilityr � FileDisplay�sys�stdoutr �path�joinZtempdirr: r �makedirsr � MagicMockZconfig�KEY�keyr ZAccountr ZRegistrationResourceZ RegistrationZ from_data�acc1�acc2�r �� __class__r r r> Z s: �� ��� ���zChooseAccountTest.setUpc C s t �|�S �N)r Zchoose_account)r Zaccountsr r r r p s zChooseAccountTest._callr c C s, t jdf|� j_| �| �| jg�| j� d S )Nr )r r �menur � assertEqualr rK �r � mock_utilr r r �test_onet s zChooseAccountTest.test_onec C s0 t jdf|� j_| �| �| j| jg�| j� d S �N� )r r rQ r rR r rK rL rS r r r �test_twoy s zChooseAccountTest.test_twoc C s0 t jdf|� j_| �| �| j| jg�d k� d S rV )r r rQ r r" r rK rL rS r r r � test_cancel~ s z ChooseAccountTest.test_cancel) r1 r2 r3 r4 r> r5 r r6 r7 rU rX rY � __classcell__r r rN r r8 X s r8 c @ s4 e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd S ) � GenSSLLabURLszELoose test of _gen_ssl_lab_urls. URL can change easily in the future.c C s t j�t�tjd�� d S �NF�r? r@ rA r rB rC rD rM r r r r> � s �zGenSSLLabURLs.setUpc C s ddl m} ||�S )Nr )�_gen_ssl_lab_urls)r r^ )r �domainsr^ r r r r � s zGenSSLLabURLs._callc C s | � | �g �g � d S rP �rR r rM r r r � test_zero� s zGenSSLLabURLs.test_zeroc C s6 | � ddg�}| �d|d k� | �d|d k� d S )Nzeff.orgz umich.edur rW )r r" )r Zurlsr r r rX � s zGenSSLLabURLs.test_twoN) r1 r2 r3 r4 r> r5 r ra rX r r r r r[ � s r[ c @ sL e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd d � Z d d � Z dd� Z dS )�GenHttpsNamesTestzTest _gen_https_names.c C s t j�t�tjd�� d S r\ r] rM r r r r> � s �zGenHttpsNamesTest.setUpc C s ddl m} ||�S )Nr )�_gen_https_names)r rc )r r_ rc r r r r � s zGenHttpsNamesTest._callc C s | � | �g �d� d S �N� r` rM r r r ra � s zGenHttpsNamesTest.test_zeroc C s. ddg}|D ] }| � | �|g�d| � q d S )N� example.comzasllkjsadfljasdf.cz https://%sr` )r �doms�domr r r rU � s �zGenHttpsNamesTest.test_onec C s8 ddgddgg}|D ] }| � | �|�dj|d�� qd S )Nz foo.bar.orgzbar.orgz paypal.google.facebook.live.comz*.zombo.example.comz%https://{dom[0]} and https://{dom[1]}�rh �rR r �format)r Z domains_listrg r r r rX � s � �zGenHttpsNamesTest.test_twoc C s( dddg}| � | �|�dj|d�� d S )N�a.org�b.org�c.orgz8https://{dom[0]}, https://{dom[1]}, and https://{dom[2]}ri rj )r rg r r r � test_three� s ��z GenHttpsNamesTest.test_threec C s. ddddg}dj |d�}| �| �|�|� d S )Nrl rm rn zd.orgzJhttps://{dom[0]}, https://{dom[1]}, https://{dom[2]}, and https://{dom[3]}ri )rk rR r )r rg Zexpr r r � test_four� s �zGenHttpsNamesTest.test_fourN) r1 r2 r3 r4 r> r5 r ra rU rX ro rp r r r r rb � s rb c @ s� e Zd ZdZdd� Zed!dd��Ze�d�dd � �Z e � d �d d � �Z e � d �d d� �Z dd� Zdd� Ze � d �dd� �Ze � d �dd� �Ze � d �dd� �Ze � d �dd� �Zdd � Ze � d �d d � �Ze � d �d d � �ZdS )"�ChooseNamesTestzTest choose names.c C s$ t j�t�tjd�� t�� | _ d S r\ ) r? r@ rA r rB rC rD r rH � mock_installrM r r r r> � s �zChooseNamesTest.setUpNc C s ddl m} |||�S )Nr )� choose_names)r rs )r Z installer�questionrs r r r r � s zChooseNamesTest._callz*certbot.display.ops._choose_names_manuallyc C s | � d � | �|jd� d S rV )r rR � call_count)r Z mock_manualr r r �test_no_installer� s z!ChooseNamesTest.test_no_installerr c C s& t jg f|� j_| �| �d �g � d S rP )r r r r rR r rS r r r �test_no_installer_cancel� s z(ChooseNamesTest.test_no_installer_cancelc C sn t � | �� j_d}tj|f|� j_| �| j�}| �|� jj d� | �||g� | � d|� jj d d k� d S )Nrf rW zconfiguration filesr ) �setrr � get_all_namesr r r r r rR ru r" r* )r rT �domainZ actual_domsr r r �test_no_names_choose� s �z$ChooseNamesTest.test_no_names_choosec s� ddl m} | �|g �g � dddg}| �||�dddg� d� ddd d g}� fd d �|D �}t|�}� fd d �|D �}| �||�|� d S )Nr �� _sort_nameszex.comzzx.comzax.comz.ex.com�e�a�z�yc s g | ] }|� �qS r r ��.0�us�rz r r � <listcomp>� s z;ChooseNamesTest.test_sort_names_trivial.<locals>.<listcomp>c s g | ] }|� �qS r r r� r� r r r� � s )r r} rR �sorted)r r} Z some_domains�unsorted_shortZ unsorted_long� sorted_shortZ sorted_longr r� r �test_sort_names_trivial� s z'ChooseNamesTest.test_sort_names_trivialc C s� ddl m} ddddg}ddd d d d g}d d d ddd g}g }|D ] }|D ]}|�|| � qHq@g }t|�D ] }|D ]}|�|| � qrqj| �||�|� d S ) Nr r| z.cx.comz.bx.comz.ax.comz.dx.comZwwwzbnother.long.subdomainr za.long.subdomainr� �b)r r} �appendr� rR ) r r} Zunsorted_domainsr� r� Zto_sortZshortrz Zsortdr r r �test_sort_names_many� s z$ChooseNamesTest.test_sort_names_manyc C sR t dg�| jj_tjdgf|� j_| �| j�}| �|dg� | �|� jj d� d S )Nrf rW ) rx rr ry r r r � checklistr rR ru �r rT �namesr r r � test_filter_names_valid_return s z.ChooseNamesTest.test_filter_names_valid_returnc C sn t dg�| jj_tjdgf|� j_| �| jd�}| �|dg� | �|� jj d� | �|� jj d d d� d S )Nrf ZCustomrW r ) rx rr ry r r r r� r rR ru r* r� r r r �$test_filter_namees_override_question s z4ChooseNamesTest.test_filter_namees_override_questionc C s8 t dg�| jj_tjg f|� j_| �| �| j�g � d S �Nrf ) rx rr ry r r r r� rR r rS r r r �"test_filter_names_nothing_selected# s z2ChooseNamesTest.test_filter_names_nothing_selectedc C s: t dg�| jj_tjdgf|� j_| �| �| j�g � d S r� ) rx rr ry r r r r� rR r rS r r r �test_filter_names_cancel* s � z(ChooseNamesTest.test_filter_names_cancelc C sf ddl m} ddddddg}d d g}dd dg}| �||�|� | �||�g � | �t||��d � d S ) Nr )�get_valid_domainsrf zsecond.example.comzalso.example.com�under_score.example.com�justtldz*.wildcard.comu öóòps.net� uniçodé.comu úniçøde.com� )r r� rR �len)r r� Z all_validZ all_invalidZ two_validr r r �test_get_valid_domains2 s � z&ChooseNamesTest.test_get_valid_domainsc C s� ddl m} |� }d|j_tjdf|j_| �|� g � t� d��*}t ddddd�}||_ | �|� g � W 5 Q R X tjd f|j_| �|� d d d d g� d S )Nr ��_choose_names_manuallyFr� z4certbot.display.ops.display_util.separate_list_inputr re rW z=example.com,under_score.example.com,justtld,valid.example.comrf r� r� zvalid.example.com) r r� �yesnor r r r rR r r! �UnicodeEncodeErrorr& )r rT r� � utility_mockZmock_sliZ unicode_errorr r r �test_choose_manually= s. ��� ��z$ChooseNamesTest.test_choose_manuallyc C sH ddl m} |� }tjdf|j_dddg|j_|� | �|jj d� d S )Nr r� r� TF� ) r r� r r r r r� r& rR ru )r rT r� r� r r r �test_choose_manually_retryX s �z*ChooseNamesTest.test_choose_manually_retry)N)r1 r2 r3 r4 r> r5 r r r! rv r6 r7 rw r{ r� r� r� r� r� r� r� r� r� r r r r rq � s2 rq c @ s. e Zd ZdZedd� �Ze�d�dd� �ZdS )�SuccessInstallationTestz&Test the success installation message.c C s ddl m} ||� d S )Nr )�success_installation)r r� )r r� r� r r r r g s z SuccessInstallationTest._callr c C s` d |� j _ddg}| �|� | �|� j jd� |� j jd d d }|D ]}| �||k� qHd S �Nrf zabc.comrW r �� notificationr r rR ru r- r" �r rT r� �arg�namer r r �test_success_installationl s z1SuccessInstallationTest.test_success_installationN) r1 r2 r3 r4 r5 r r6 r7 r� r r r r r� d s r� c @ s. e Zd ZdZedd� �Ze�d�dd� �ZdS )�SuccessRenewalTestz!Test the success renewal message.c C s ddl m} ||� d S )Nr )�success_renewal)r r� )r r� r� r r r r } s zSuccessRenewalTest._callr c C s` d |� j _ddg}| �|� | �|� j jd� |� j jd d d }|D ]}| �||k� qHd S r� r� r� r r r �test_success_renewal� s z'SuccessRenewalTest.test_success_renewalN) r1 r2 r3 r4 r5 r r6 r7 r� r r r r r� z s r� c @ s. e Zd ZdZedd� �Ze�d�dd� �ZdS )�SuccessRevocationTestz$Test the success revocation message.c C s ddl m} ||� d S )Nr )�success_revocation)r r� )r rE r� r r r r � s zSuccessRevocationTest._callr c C sT d |� j _d}| �|� |� j jd�|tj�dd� | �||� j jd d k� d S )Nz/path/to/cert.pemz\Congratulations! You have successfully revoked the certificate that was located at {0}{1}{1}F)�pauser ) r� r r Zassert_called_once_withrk r �linesepr" r* )r rT rE r r r �test_success_revocation� s ��z-SuccessRevocationTest.test_success_revocationN) r1 r2 r3 r4 r5 r r6 r7 r� r r r r r� � s r� c @ s� e Zd ZdZdZdZdZedd� �Ze � � dd� �Z e � � d d � �Z e � � d d � �Z e � � d d� �Ze � � dd� �Ze � � dd� �Ze � � dd� �ZdS )�ValidatorTestsz6Tests for `validated_input` and `validated_directory`.zMust be non-emptyZasdfz /var/www/htmlc C s | dkrt �tj��d S rd )r Z PluginErrorr� �_ValidatorTests__ERROR)�mr r r Z __validator� s zValidatorTests.__validatorc C sr t jdft jdft jdft j| jfg|� j_tj| jddd�}| �t j |� j j d d � | �|t j| jf� d S )Nre �messageT�Zforce_interactiver ) r r � valid_inputr r& r �validated_input�_ValidatorTests__validatorrR r� r� r� r* �r rT Zreturnedr r r � test_input_blank_with_validator� s � z.ValidatorTests.test_input_blank_with_validatorc C s> t j| jfg|� j_tj| jddd�}| �|t j| jf� d S �N�msg�other��default) r r r� r r& r r� r� rR r� r r r �"test_input_validation_with_default� s z1ValidatorTests.test_input_validation_with_defaultc C s0 t j| jfg|� j_| jttj| j ddd� d S �Nr� re r� ) r r r� r r& r �AssertionErrorr r� r� rS r r r �&test_input_validation_with_bad_default� s �z5ValidatorTests.test_input_validation_with_bad_defaultc C s: t jdfg|� j_tj| jddd�\}}| �|t j� d S )Nre r� Tr� )r r r r& r r� r� rR )r rT �codeZ unused_rawr r r � test_input_cancel_with_validator� s z/ValidatorTests.test_input_cancel_with_validatorc C sb t jdft j| jfg|� j_tj| jddd�}| �t j |� j j d d � | �|t j| jf� d S )Nre r� Tr� r ) r r �valid_directory�directory_selectr& r �validated_directoryr� rR r� r� r� r* r� r r r � test_directory_select_validation� s � z/ValidatorTests.test_directory_select_validationc C s> t j| jfg|� j_tj| jddd�}| �|t j| jf� d S r� ) r r r� r� r& r r� r� rR r� r r r �-test_directory_select_validation_with_default� s z<ValidatorTests.test_directory_select_validation_with_defaultc C s0 t j| jfg|� j_| jttj| j ddd� d S r� ) r r r� r� r& r r� r r� r� rS r r r �1test_directory_select_validation_with_bad_default� s �z@ValidatorTests.test_directory_select_validation_with_bad_defaultN)r1 r2 r3 r4 r� r� r� � staticmethodr� r6 r7 r� r� r� r� r� r� r� r r r r r� � s( r� c @ sR e Zd ZdZedd� �Ze�d�dd� �Ze�d�dd� �Z e�d�d d � �Z d S ) �ChooseValuesTestzTest choose_values.c C s ddl m} |||�S )Nr )� choose_values)r r� )r �valuesrt r� r r r r � s zChooseValuesTest._callr c C sl dddg}t j|d gf|� j_| �|d �}| �||d g� | �|� jj� | �|� jjd d d � d S )N�first�second�thirdr� r � r r r� r r rR r" �calledr* )r rT �items�resultr r r �test_choose_names_success� s z*ChooseValuesTest.test_choose_names_successc C sp dddg}d}t j|d gf|� j_| �||�}| �||d g� | �|� jj� | �|� jjd d |� d S )Nr� r� r� z Which one?rW r r� �r rT r� rt r� r r r �"test_choose_names_success_question� s z3ChooseValuesTest.test_choose_names_success_questionc C sd dddg}d}t jg f|� j_| �||�}| �|g � | �|� jj� | �|� jjd d |� d S )Nr� r� r� zWant to cancel?r ) r r r� r r rR r" r� r* r� r r r � test_choose_names_user_cancel s z.ChooseValuesTest.test_choose_names_user_cancelN) r1 r2 r3 r4 r5 r r6 r7 r� r� r� r r r r r� � s r� �__main__))r4 rC ZunittestZjosepyZjoser Zzope.componentr? Zacmer Zcertbot.tests.utilZtestsr r6 Zcertbotr r Zcertbot.compatr r Zcertbot.displayr r ZJWKRSA�loadZ load_vectorrI ZTestCaser ZTempDirTestCaser8 r[ rb rq r� r� r� r� r� r1 �mainr r r r �<module> s6 A,0 G%
Save File
Cancel