/usr/lib/python3.9/site-packages/certbot/_internal/tests/display/__pycache__
a }|�g3O � @ s� d Z ddlZddlZddlmZ ddlZddlZddlmZ ddl m Z ddl m Z ddl mZ ddlmZ dd lmZ dd lmZ dd lmZ ddlm mZ ej�e� d ��Z G d d� dej �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 �e�)d �G d!d"� d"ej ���Z*e+d#k�r�e�,e�-ej.d$d� e/g �� dS )%zTest certbot.display.ops.� N)�mock)�messages)�errors)�account)�obj)� filesystem)�os)�ops)�utilzrsa512_key.pemc @ s| e Zd ZdZedd� �Ze�� dd� �Ze�� dd� �Z e�� dd � �Z e�� d d � �Z e�� d d � �Z e�� dd� �Z dS )� GetEmailTestz(Tests for certbot.display.ops.get_email.c K s ddl m} |f i |��S )Nr )� get_email)�certbot.display.opsr )�cls�kwargsr � r �L/usr/lib/python3.9/site-packages/certbot/_internal/tests/display/ops_test.py�_call s zGetEmailTest._callc C s� |� j }tjdf|_t�tj�� | �� W d � n1 s>0 Y t�tj�� | jdd� W d � n1 sv0 Y d S )N� foo@bar.bazF�Zoptional) �input� display_util�CANCEL� return_value�pytest�raisesr �Errorr )�self�mock_get_utility� mock_inputr r r �test_cancel_none s &z GetEmailTest.test_cancel_nonec C sX |� j }tjdf|_t�d��&}d|_| �� dks6J �W d � n1 sJ0 Y d S )Nr �#certbot.display.ops.util.safe_emailT)r r �OKr r �patchr �r r r �mock_safe_emailr r r � test_ok_safe( s zGetEmailTest.test_ok_safec C s\ |� j }tjdf|_t�d��*}ddg|_| �� dks:J �W d � n1 sN0 Y d S )Nr r FT)r r r! r r r" � side_effectr r# r r r �test_ok_not_safe0 s z GetEmailTest.test_ok_not_safec C s� d}|� j }tjdf|_t�d��V}d|_| �� ||jd d vsHJ �| jdd� ||jd d v sjJ �W d � n1 s~0 Y d S )N�There seem to be problemsr r Tr �Zinvalid)r r r! r r r" r � call_args�r r Z invalid_txtr r$ r r r �test_invalid_flag8 s z GetEmailTest.test_invalid_flagc C sx |� j }tjdf|_t�d��F}ddg|_| jdd� |jD ]}d|d d vs<J �q<W d � n1 sj0 Y d S )Nr r FTr z!--register-unsafely-without-emailr ) r r r! r r r" r& r �call_args_list)r r r r$ �callr r r �test_optional_flagD s z GetEmailTest.test_optional_flagc C sr d}|� j }tjdf|_t�d��<}ddg|_| jdd� ||jd d v sPJ �W d � n1 sd0 Y d S )Nr( r r FTr) r ) r r r! r r r" r& r r* r+ r r r � test_optional_invalid_unsafeN s z)GetEmailTest.test_optional_invalid_unsafeN)�__name__� __module__� __qualname__�__doc__� classmethodr � test_util�patch_display_utilr r% r' r, r/ r0 r r r r r s r c s\ e Zd ZdZ� fdd�Zedd� �Ze�� dd� �Z e�� dd � �Z e�� d d � �Z � Z S ) �ChooseAccountTestz-Tests for certbot.display.ops.choose_account.c s� t � �� t�t�tjd�� tj� | j d�| _ t � | j d� tj| j | j dd�| _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)Zuri�bodyz email2@g.com�phone)r; r= )�super�setUp� display_obj� set_display� FileDisplay�sys�stdoutr �path�joinZtempdirr: r �makedirsr � MagicMockZconfig�KEY�keyr ZAccountr ZRegistrationResourceZ RegistrationZ from_data�acc1�acc2�r �� __class__r r r? [ s0 �������zChooseAccountTest.setUpc C s t �|�S �N)r Zchoose_account)r Zaccountsr r r r p s zChooseAccountTest._callc C s, t jdf|� j_| �| jg�| jks(J �d S )Nr )r r! �menur r rK �r � mock_utilr r r �test_onet s zChooseAccountTest.test_onec C s0 t jdf|� j_| �| j| jg�| jks,J �d S �N� )r r! rQ r r rK rL rR r r r �test_twoy s zChooseAccountTest.test_twoc C s. t jdf|� j_| �| j| jg�d u s*J �d S rU )r r rQ r r rK rL rR r r r � test_cancel~ s z ChooseAccountTest.test_cancel) r1 r2 r3 r4 r? r5 r r6 r7 rT rW rX � __classcell__r r rN r r8 Y s r8 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 �t �tjd�� d S �NF)r@ rA rB rC rD rM r r r r? � s zGenHttpsNamesTest.setUpc C s ddl m} ||�S )Nr )�_gen_https_names)r r\ )r Zdomainsr\ r r r r � s zGenHttpsNamesTest._callc C s | � g �dksJ �d S �N� �r rM r r r � test_zero� s zGenHttpsNamesTest.test_zeroc C s. ddg}|D ] }| � |g�d| ks J �q d S )N� example.comzasllkjsadfljasdf.cz https://%sr_ )r �doms�domr r r rT � s �zGenHttpsNamesTest.test_onec C s8 ddgddgg}|D ] }| � |�dj|d�ksJ �qd S )Nz foo.bar.orgzbar.orgz paypal.google.facebook.live.comz*.zombo.example.comz%https://{dom[0]} and https://{dom[1]}�rc �r �format)r Z domains_listrb r r r rW � s � �zGenHttpsNamesTest.test_twoc C s& g d�}| � |�dj|d�ks"J �d S )N)�a.org�b.org�c.orgz8https://{dom[0]}, https://{dom[1]}, and https://{dom[2]}rd re )r rb r r r � test_three� s ��z GenHttpsNamesTest.test_threec C s* g d�}dj |d�}| �|�|ks&J �d S )N)rg rh ri zd.orgzJhttps://{dom[0]}, https://{dom[1]}, https://{dom[2]}, and https://{dom[3]}rd )rf r )r rb Zexpr r r � test_four� s �zGenHttpsNamesTest.test_fourN) r1 r2 r3 r4 r? r5 r r` rT rW rj rk r r r r rZ � s rZ c @ s� e Zd ZdZdd� Zed dd��Ze�d�dd � �Z e � � d d � �Z e � � d d � �Z dd� Zdd� Ze � � dd� �Ze � � dd� �Ze � � dd� �Ze � � dd� �Zdd� Ze � � d d � �Ze � � d d � �ZdS )!�ChooseNamesTestzTest choose names.c C s" t �t �tjd�� t�� | _d S r[ )r@ rA rB rC rD r rH � mock_installrM r r r r? � s zChooseNamesTest.setUpNc C s ddl m} |||�S )Nr )� choose_names)r rn )r Z installer�questionrn r r r r � s zChooseNamesTest._callz*certbot.display.ops._choose_names_manuallyc C s | � d � |jdksJ �d S rU )r � call_count)r Z mock_manualr r r �test_no_installer� s z!ChooseNamesTest.test_no_installerc C s& t jg f|� j_| �d �g ks"J �d S rP )r r r r r rR r r r �test_no_installer_cancel� s z(ChooseNamesTest.test_no_installer_cancelc C sR t � | �� j_d}tj|f|� j_| �| j�}|� jjdks@J �||gksNJ �d S �Nra rV ) �setrm � get_all_namesr r r! r r rp )r rS �domainZ actual_domsr r r �test_no_names_choose� s z$ChooseNamesTest.test_no_names_choosec s� ddl m} |g �g ks J �g d�}||�g d�ks8J �d� g d�}� fdd�|D �}t|�}� fd d�|D �}||�|ks�J �d S ) Nr �� _sort_names)�ex.com�zx.com�ax.com)r| rz r{ z.ex.com)�e�a�z�yc s g | ] }|� �qS r r ��.0�us�rv r r � <listcomp>� � z;ChooseNamesTest.test_sort_names_trivial.<locals>.<listcomp>c s g | ] }|� �qS r r r� r� r r r� � r� )r ry �sorted)r ry 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} g d�}g d�}g d�}g }|D ] }|D ]}|�|| � q4q,g }t|�D ] }|D ]}|�|| � q^qV||�|ks�J �d S )Nr rx )z.cx.comz.bx.comz.ax.comz.dx.com)�www�bnother.long.subdomainr~ �a.long.subdomainr �b)r~ r� r� r� r� r )r ry �appendr� ) r ry Zunsorted_domainsr� r� Zto_sortZshortrv Zsortdr r r �test_sort_names_many� s z$ChooseNamesTest.test_sort_names_manyc C sN dh| j j_tjdgf|� j_| �| j �}|dgks8J �|� jjdksJJ �d S rs )rm ru r r r! � checklistr rp �r rS �namesr r r � test_filter_names_valid_return� s z.ChooseNamesTest.test_filter_names_valid_returnc C sj dh| j j_tjdgf|� j_| �| j d�}|dgks:J �|� jjdksLJ �|� jjd d dksfJ �d S )Nra ZCustomrV r ) rm ru r r r! r� r rp r* r� r r r �$test_filter_namees_override_question s z4ChooseNamesTest.test_filter_namees_override_questionc C s4 dh| j j_tjg f|� j_| �| j �g ks0J �d S �Nra )rm ru r r r! r� r rR r r r �"test_filter_names_nothing_selected s z2ChooseNamesTest.test_filter_names_nothing_selectedc C s6 dh| j j_tjdgf|� j_| �| j �g ks2J �d S r� )rm ru r r r r� r rR r r r �test_filter_names_cancel s � z(ChooseNamesTest.test_filter_names_cancelc C s\ ddl m} g d�}ddg}g d�}||�|ks4J �||�g ksDJ �t||��dksXJ �d S )Nr )�get_valid_domains)ra zsecond.example.com�also.example.com�under_score.example.com�justtldz*.wildcard.comu öóòps.net� uniçodé.com)ra u úniçøde.comr� � )r r� �len)r r� Z all_validZ all_invalidZ two_validr r r �test_get_valid_domains s z&ChooseNamesTest.test_get_valid_domainsc C s� ddl m} |� }d|j_tjdf|j_|� g ks6J �t�d��4}t ddddd�}||_ |� g ksfJ �W d � n1 sz0 Y tjd f|j_|� g d �ks�J �d S ) Nr ��_choose_names_manuallyFr� z=certbot.display.ops.internal_display_util.separate_list_inputr r^ rV z=example.com,under_score.example.com,justtld,valid.example.com)ra r� r� zvalid.example.com) r r� �yesnor r r! r r r" �UnicodeEncodeErrorr& )r rS r� � utility_mockZmock_sliZ unicode_errorr r r �test_choose_manually# s( ��,��z$ChooseNamesTest.test_choose_manuallyc C sF ddl m} |� }tjdf|j_g d�|j_|� |jjdksBJ �d S )Nr r� r� )TTF� ) r r� r r! r r r� r& rp )r rS r� r� r r r �test_choose_manually_retry> s � z*ChooseNamesTest.test_choose_manually_retry)N)r1 r2 r3 r4 r? r5 r r r" rq r6 r7 rr rw r� r� r� r� r� r� r� r� r� r r r r rl � s2 rl c @ s6 e Zd ZdZedd� �Ze�� e� d�dd� ��Z dS )�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 L s z SuccessInstallationTest._call�certbot.display.util.notifyc C sV d |� j _ddg}| �|� |jdks*J �|jd d d }|D ]}||v s@J �q@d S )Nra �abc.comrV r )� notificationr r rp r- )r � mock_notify� mock_displayr� �arg�namer r r �test_success_installationQ s z1SuccessInstallationTest.test_success_installationN) r1 r2 r3 r4 r5 r r6 r7 r r" r� r r r r r� J s r� c @ s6 e Zd ZdZedd� �Ze�� e� d�dd� ��Z dS )�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 b s zSuccessRenewalTest._callr� c C s. d |� j _ddg}| �|� |jdks*J �d S )Nra r� rV )r� r r rp )r r� r� r� r r r �test_success_renewalg s z'SuccessRenewalTest.test_success_renewalN) r1 r2 r3 r4 r5 r r6 r7 r r" r� r r r r r� ` s r� c @ s6 e Zd ZdZedd� �Ze�� e� d�dd� ��Z dS )�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 t s zSuccessRevocationTest._callr� c C s" d}| � |� |�d�|�� d S )Nz/path/to/cert.pemzWCongratulations! You have successfully revoked the certificate that was located at {0}.)r Zassert_called_once_withrf )r r� Zunused_mock_displayrE r r r �test_success_revocationy s ��z-SuccessRevocationTest.test_success_revocationN) r1 r2 r3 r4 r5 r r6 r7 r r" r� 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 r] )r Z PluginErrorr� �_ValidatorTests__ERROR)�mr r r Z __validator� s zValidatorTests.__validatorc C sr t jdft jdft jdft j| jfg|� j_tj| jddd�}tj |� j j d d ksZJ �|t j| jfksnJ �d S )Nr^ �messageT�Zforce_interactiver ) r r! � valid_inputr r& r �validated_input�_ValidatorTests__validatorr� r� r� r* �r rS 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| jfks:J �d S �N�msg�other��default)r r! r� r r& r r� r� r� r r r �"test_input_validation_with_default� s z1ValidatorTests.test_input_validation_with_defaultc C sT t j| jfg|� j_t�t��" tj | j ddd� W d � n1 sF0 Y d S �Nr� r^ r� ) r r! r� r r& r r �AssertionErrorr r� r� rR 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 jks6J �d S )Nr^ r� Tr� )r r r r& r r� r� )r rS �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�}tj |� j j d d ksJJ �|t j| jfks^J �d S )Nr^ r� Tr� r ) r r! �valid_directory�directory_selectr& r �validated_directoryr� 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| jfks:J �d S r� )r r! r� r� r& r r� r� r� r r r �-test_directory_select_validation_with_default� s z<ValidatorTests.test_directory_select_validation_with_defaultc C sT t j| jfg|� j_t�t��" tj | j ddd� W d � n1 sF0 Y d S r� ) r r! r� r� r& r r r� r r� r� rR 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 @ sL e Zd ZdZedd� �Ze�� dd� �Ze�� dd� �Z e�� dd � �Z d S ) �ChooseValuesTestzTest choose_values.c C s ddl m} |||�S )Nr )� choose_values)r r� )r �valuesro r� r r r r � s zChooseValuesTest._callc C sl g d�}t j|d gf|� j_| �|d �}||d gks<J �|� jjdu sNJ �|� jjd d dkshJ �d S )N��first�second�thirdr� Tr r^ �r r! r� r r �calledr* )r rS �items�resultr r r �test_choose_names_success� s z*ChooseValuesTest.test_choose_names_successc C sp g d�}d}t j|d gf|� j_| �||�}||d gks@J �|� jjdu sRJ �|� jjd d |kslJ �d S )Nr� z Which one?rV Tr r� �r rS r� ro r� r r r �"test_choose_names_success_question� s z3ChooseValuesTest.test_choose_names_success_questionc C sd g d�}d}t jg f|� j_| �||�}|g ks4J �|� jjdu sFJ �|� jjd d |ks`J �d S )Nr� zWant to cancel?Tr )r r r� r 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� zcertbot.display.ops.loggerr� c @ s@ e Zd ZdZeeeeed�dd��Zdd� Zdd� Z d d � Z d S ) �ReportExecutedCommandz Test report_executed_command)�cmd_name�rc�out�errc C s ddl m} |||||� d S )Nr )�report_executed_command)r r� )r r� r� r� r� r� r r r r � s zReportExecutedCommand._callc C s2 | � dddd� |jjdks J �|jdks.J �d S )N� some-hookr � Did a thing� Some warningrV �r Zwarningrp �r r� Z mock_loggerr r r �test_mixed_success� s z(ReportExecutedCommand.test_mixed_successc C s2 | � dddd� |jjdks J �|jdks.J �d S )Nr� i����r� r� r� rV r� r� r r r �test_mixed_error� s z&ReportExecutedCommand.test_mixed_errorc C s2 | � dddd� |jjdks J �|jdks.J �d S )Nr� r � � r� r� r r r �test_empty_success s z(ReportExecutedCommand.test_empty_successN) r1 r2 r3 r4 r5 �str�intr r� r� r� r r r r r� � s r� �__main__rV )0r4 rC Zunittestr ZjosepyZjoser Zacmer Zcertbotr Zcertbot._internalr Zcertbot._internal.displayr r@ Zcertbot.compatr r Zcertbot.displayr r r Zcertbot.tests.utilZtestsr6 ZJWKRSA�loadZ load_vectorrI ZTestCaser ZTempDirTestCaser8 rZ rl r� r� r� r� r� r" r� r1 �exit�main�argv�__file__r r r r �<module> s< B+- E%
.
Edit
..
Edit
__init__.cpython-39.opt-1.pyc
Edit
__init__.cpython-39.pyc
Edit
completer_test.cpython-39.opt-1.pyc
Edit
completer_test.cpython-39.pyc
Edit
internal_util_test.cpython-39.opt-1.pyc
Edit
internal_util_test.cpython-39.pyc
Edit
obj_test.cpython-39.opt-1.pyc
Edit
obj_test.cpython-39.pyc
Edit
ops_test.cpython-39.opt-1.pyc
Edit
ops_test.cpython-39.pyc
Edit
util_test.cpython-39.opt-1.pyc
Edit
util_test.cpython-39.pyc
Edit